nahakubuilder/flask_url_shortener
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added a way to create custom tokens

Browse Source
This commit is contained in:
xemeds
2020-08-21 12:17:17 +03:00
parent 17838b61ab
commit f02158387e
7 changed files with 86 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tiny0/routes.py
+19 -7
View File
@@ -13,15 +13,27 @@ def index():
# If the form was valid
if form.validate_on_submit():
# Generate a valid token
token = gen_valid_token()
# Add the token and the given url to the database
db.session.add(URL(token=token, url=form.url.data))
db.session.commit()
# If a token was given
if form.token.data:
# Add the token and the given url to the database
db.session.add(URL(token=form.token.data, url=form.url.data))
db.session.commit()
# Return the url page with the shortened url
return render_template("url.html", url=WEBSITE_DOMAIN + "/" + token)
# Return the url page with the shortened url
return render_template("url.html", url=WEBSITE_DOMAIN + "/" + form.token.data)
# Else if a token was not given
else:
# Generate a valid token
token = gen_valid_token()
# Add the token and the given url to the database
db.session.add(URL(token=token, url=form.url.data))
db.session.commit()
# Return the url page with the shortened url
return render_template("url.html", url=WEBSITE_DOMAIN + "/" + token)
# If the form was invalid or not submitted
else: