75 lines
2.3 KiB
Python
75 lines
2.3 KiB
Python
import logging
|
|
import os
|
|
from pathlib import Path
|
|
|
|
import trafaret
|
|
from flask import Flask, abort, redirect, render_template, request
|
|
|
|
|
|
class Web:
|
|
def __init__(self, handler, debug=False):
|
|
self.logger = logging.getLogger(self.__class__.__name__)
|
|
self.app = None
|
|
self.host = None
|
|
self.port = None
|
|
self.handler = handler
|
|
self.debug = debug
|
|
|
|
def start_up(self):
|
|
self.init()
|
|
self.app.run(host=self.host, port=self.port, debug=self.debug)
|
|
|
|
def init(self):
|
|
self.app = Flask(__name__)
|
|
self.setup_routes()
|
|
|
|
def setup_routes(self):
|
|
self.app.add_url_rule("/", "/", self.handler.index, methods=["GET"])
|
|
self.app.add_url_rule(
|
|
"/shortenit", "/shortenit", self.handler.shortenit, methods=["POST"]
|
|
)
|
|
self.app.add_url_rule(
|
|
"/<identifier>", "/identifier", self.handler.short_redirect, methods=["GET"]
|
|
)
|
|
|
|
|
|
class SiteHandler:
|
|
def __init__(self, configuration, database, shorten_url, lenghten_url):
|
|
self.logger = logging.getLogger(self.__class__.__name__)
|
|
self.configuration = configuration
|
|
self.database = database
|
|
self.shorten_url = shorten_url
|
|
self.lenghten_url = lenghten_url
|
|
self.shortenit_load_format = trafaret.Dict(
|
|
{trafaret.Key("url"): trafaret.URL, trafaret.Key("timestamp"): trafaret.Int}
|
|
)
|
|
|
|
def shortenit(self):
|
|
data = request.get_json()
|
|
try:
|
|
data = self.shortenit_load_format(data)
|
|
except Exception as e:
|
|
self.logger.error(e)
|
|
return {}, 400
|
|
self.logger.error(e)
|
|
abort(400)
|
|
try:
|
|
short_url = self.shorten_url(
|
|
self.configuration, self.database, data["url"], data["timestamp"]
|
|
)
|
|
except KeyError as e:
|
|
self.logger.error(e)
|
|
abort(400)
|
|
self.logger.debug(short_url)
|
|
return {"url": short_url}
|
|
|
|
def short_redirect(self, identifier):
|
|
url = self.lenghten_url(self.database, identifier)
|
|
self.logger.debug("The URL is...")
|
|
self.logger.debug(url)
|
|
if not url:
|
|
abort(404)
|
|
return redirect(url)
|
|
|
|
def index(self):
|
|
return render_template("index.html")
|