app.py

import logging
from flask import Flask, json, request, redirect, render_template, make_response
from url_shortener.shorten import UrlShortener
from urllib.parse import urlparse

app = Flask(__name__)
shrt = UrlShortener()
logger = logging.getLogger()

logger.info("starting up")


# template pushing routes
@app.route("/")
def index():
    return render_template("index.html")


@app.route("/404")
def missing():
    return render_template("missing.html")


@app.route("/400")
def invalid():
    return render_template("invalid")


# short url lookup
@app.route("/<code>")
def lookup(code):
    url = shrt.lookup(code)
    if not url:
        return redirect("/404")
    else:
        return redirect(url)


# short url generation
#
# If JSON is fed, we shorten and reply in JSON as well
# If a Form is posted we reply in HTML
# Otherwise we redirect to a failure page


@app.route("/", methods=["POST"])
def shorten_url():
    if request.json and "url" in request.json:
        u = urlparse(request.json["url"])
        if u.netloc == "":
            url = "http://" + request.json["url"]
        else:
            url = request.json["url"]
        res = shrt.shorten(url)
        logger.debug("shortened %s to %s" % (url, res))
        response = make_response(json.dumps(res))
        response.headers["Content-Type"] = "application/json"
        return response

    elif request.form and "url" in request.form:
        u = urlparse(request.form["url"])
        if u.netloc == "":
            url = "http://" + request.form["url"]
        else:
            url = request.form["url"]
        res = shrt.shorten(url)
        logger.debug("shortened %s to %s" % (url, res))
        return render_template("result.html", result=res)

    else:
        logger.info("invalid shorten request")
        return redirect("/400")