fix: Limit link chars to letters and numbers
All checks were successful
Build Docker / Build Main Image (push) Successful in 20s

This commit is contained in:
Nathan Woodburn 2023-11-17 10:55:00 +11:00
parent e0d2e89c55
commit a1122419e9
Signed by: nathanwoodburn
GPG Key ID: 203B000478AD0EF1
2 changed files with 6 additions and 0 deletions

2
db.py
View File

@ -52,6 +52,7 @@ def get_link_count():
cursor = connection.cursor()
cursor.execute("SELECT COUNT(*) FROM links")
result = cursor.fetchone()
cursor.fetchall()
cursor.close()
connection.close()
@ -70,6 +71,7 @@ def get_account_count():
cursor = connection.cursor()
cursor.execute("SELECT * FROM links")
result = cursor.fetchone()
cursor.fetchall()
cursor.close()
connection.close()
last_check_account = time.time()

View File

@ -114,6 +114,10 @@ def add_link():
if len(link) < 5:
return error('Link too short')
regexmatch = re.match(r"^[a-zA-Z0-9]+$", link)
if not regexmatch:
return error('Invalid link')
# Verify link is not taken
if db.get_link(link) != False:
return error('Link already taken')