spuuntries
commited on
Commit
·
79045e1
1
Parent(s):
2161c06
fix: tighten query
Browse filesgwe mwnya yg intended aja yh
app.py
CHANGED
@@ -24,7 +24,8 @@ def setup_db():
|
|
24 |
def check_login(username, password):
|
25 |
conn = sqlite3.connect("users.db")
|
26 |
c = conn.cursor()
|
27 |
-
|
|
|
28 |
try:
|
29 |
c.execute(query)
|
30 |
result = c.fetchone()
|
@@ -33,10 +34,10 @@ def check_login(username, password):
|
|
33 |
return "admin"
|
34 |
elif result and result[0] == "user":
|
35 |
return "user"
|
36 |
-
return None
|
37 |
except:
|
38 |
conn.close()
|
39 |
-
return None
|
40 |
|
41 |
|
42 |
def respond(
|
|
|
24 |
def check_login(username, password):
|
25 |
conn = sqlite3.connect("users.db")
|
26 |
c = conn.cursor()
|
27 |
+
|
28 |
+
query = f"SELECT role FROM (SELECT username, password, role FROM users WHERE role='user') AS t WHERE username='{username}' AND password='{password}' LIMIT 1"
|
29 |
try:
|
30 |
c.execute(query)
|
31 |
result = c.fetchone()
|
|
|
34 |
return "admin"
|
35 |
elif result and result[0] == "user":
|
36 |
return "user"
|
37 |
+
return None
|
38 |
except:
|
39 |
conn.close()
|
40 |
+
return None
|
41 |
|
42 |
|
43 |
def respond(
|