spuuntries commited on
Commit
79045e1
·
1 Parent(s): 2161c06

fix: tighten query

Browse files

gwe mwnya yg intended aja yh

Files changed (1) hide show
  1. app.py +4 -3
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
- query = f"SELECT role FROM users WHERE username='{username}' AND password='{password}' LIMIT 1"
 
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 # Return None for failed login
37
  except:
38
  conn.close()
39
- return None # Return None for any errors
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(