63 lines
1.8 KiB
Python
63 lines
1.8 KiB
Python
from werkzeug.exceptions import NotFound
|
|
from werkzeug.utils import redirect
|
|
|
|
from .models import URL
|
|
from .utils import expose
|
|
from .utils import Pagination
|
|
from .utils import render_template
|
|
from .utils import session
|
|
from .utils import url_for
|
|
from .utils import validate_url
|
|
|
|
|
|
@expose("/")
|
|
def new(request):
|
|
error = url = ""
|
|
if request.method == "POST":
|
|
url = request.form.get("url")
|
|
alias = request.form.get("alias")
|
|
if not validate_url(url):
|
|
error = "I'm sorry but you cannot shorten this URL."
|
|
elif alias:
|
|
if len(alias) > 140:
|
|
error = "Your alias is too long"
|
|
elif "/" in alias:
|
|
error = "Your alias might not include a slash"
|
|
elif URL.query.get(alias):
|
|
error = "The alias you have requested exists already"
|
|
if not error:
|
|
uid = URL(url, "private" not in request.form, alias).uid
|
|
session.commit()
|
|
return redirect(url_for("display", uid=uid))
|
|
return render_template("new.html", error=error, url=url)
|
|
|
|
|
|
@expose("/display/<uid>")
|
|
def display(request, uid):
|
|
url = URL.query.get(uid)
|
|
if not url:
|
|
raise NotFound()
|
|
return render_template("display.html", url=url)
|
|
|
|
|
|
@expose("/u/<uid>")
|
|
def link(request, uid):
|
|
url = URL.query.get(uid)
|
|
if not url:
|
|
raise NotFound()
|
|
return redirect(url.target, 301)
|
|
|
|
|
|
@expose("/list/", defaults={"page": 1})
|
|
@expose("/list/<int:page>")
|
|
def list(request, page):
|
|
query = URL.query.filter_by(public=True)
|
|
pagination = Pagination(query, 30, page, "list")
|
|
if pagination.page > 1 and not pagination.entries:
|
|
raise NotFound()
|
|
return render_template("list.html", pagination=pagination)
|
|
|
|
|
|
def not_found(request):
|
|
return render_template("not_found.html")
|