werkzeug

1
from werkzeug.exceptions import NotFound
2
from werkzeug.utils import redirect
3

4
from .models import URL
5
from .utils import expose
6
from .utils import Pagination
7
from .utils import render_template
8
from .utils import url_for
9
from .utils import validate_url
10

11

12
@expose("/")
13
def new(request):
14
    error = url = ""
15
    if request.method == "POST":
16
        url = request.form.get("url")
17
        alias = request.form.get("alias")
18
        if not validate_url(url):
19
            error = "I'm sorry but you cannot shorten this URL."
20
        elif alias:
21
            if len(alias) > 140:
22
                error = "Your alias is too long"
23
            elif "/" in alias:
24
                error = "Your alias might not include a slash"
25
            elif URL.load(alias):
26
                error = "The alias you have requested exists already"
27
        if not error:
28
            url = URL(
29
                target=url,
30
                public="private" not in request.form,
31
                shorty_id=alias if alias else None,
32
            )
33
            url.store()
34
            uid = url.id
35
            return redirect(url_for("display", uid=uid))
36
    return render_template("new.html", error=error, url=url)
37

38

39
@expose("/display/<uid>")
40
def display(request, uid):
41
    url = URL.load(uid)
42
    if not url:
43
        raise NotFound()
44
    return render_template("display.html", url=url)
45

46

47
@expose("/u/<uid>")
48
def link(request, uid):
49
    url = URL.load(uid)
50
    if not url:
51
        raise NotFound()
52
    return redirect(url.target, 301)
53

54

55
@expose("/list/", defaults={"page": 1})
56
@expose("/list/<int:page>")
57
def list(request, page):
58
    def wrap(doc):
59
        data = doc.value
60
        data["_id"] = doc.id
61
        return URL.wrap(data)
62

63
    code = """function(doc) { if (doc.public){ map([doc._id], doc); }}"""
64
    docResults = URL.query(code)
65
    results = [wrap(doc) for doc in docResults]
66
    pagination = Pagination(results, 1, page, "list")
67
    if pagination.page > 1 and not pagination.entries:
68
        raise NotFound()
69
    return render_template("list.html", pagination=pagination)
70

71

72
def not_found(request):
73
    return render_template("not_found.html")
74