Commit cf049fa5 authored by TheTechRobo's avatar TheTechRobo
Browse files

Add regex verification to the backend

parent 0d00f06e
Loading
Loading
Loading
Loading
+9 −0
Original line number Diff line number Diff line
import re

from flask import *
from findmedia import *

@@ -5,6 +7,8 @@ app = Flask(__name__)

@app.route("/find/<id>")
async def find(id):
    if not re.match(r"^[A-Za-z0-9_-]{10}[AEIMQUYcgkosw048]$", id):
        return {"status": "bad.id", "true": True, "id": None}, 400
    return {
            "status": "find.id",
            "true": True,
@@ -21,6 +25,11 @@ async def find(id):
async def ui():
    return render_template("init.html")

@app.route("/nojs")
async def formsubmit():
    vid = request.args.get("vid") or abort(400)
    return render_template("fid.html", data=(await find(vid)), vid=vid, nonoscript=True)

@app.route("/ui/fid")
async def fid():
    vid = request.args.get("vid") or abort(400)