Move sql query for has_user to db_structure.py

iScsc · Aug 31, 2024 · de5fc08 · de5fc08
1 parent 6997ca9
commit de5fc08
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/src/database/db_manager.py b/src/database/db_manager.py
@@ -7,6 +7,7 @@
     sql_add_user,
     sql_get_user,
     sql_get_users,
+    sql_has_user,
 )
 from classes import User
 from classes import Challenge
@@ -72,7 +73,7 @@ async def add_user(self, idx):
 
     def has_user(self, idx):
         cur = self.db.cursor()
-        res = cur.execute("SELECT * FROM users WHERE id=(?)", (idx,)).fetchone()
+        res = cur.execute(sql_has_user, (idx, )).fetchone()
         cur.close()
         return res is not None
 

diff --git a/src/database/db_structure.py b/src/database/db_structure.py
@@ -14,3 +14,6 @@
 
 
 sql_get_users = """SELECT * FROM users;"""
+
+
+sql_has_user = """SELECT * FROM users WHERE id=(?)"""
Original file line number	Diff line number	Diff line change
Expand Up		@@ -14,3 +14,6 @@


		sql_get_users = """SELECT * FROM users;"""


		sql_has_user = """SELECT * FROM users WHERE id=(?)"""