nathanwoodburn/linkr
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
db.py
View File

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

4
main.py
View File

@ -114,6 +114,10 @@ def add_link():
if len(link) < 5: if len(link) < 5:
return error('Link too short') 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 # Verify link is not taken
if db.get_link(link) != False: if db.get_link(link) != False:
return error('Link already taken') return error('Link already taken')