From 0e9394b0578a298c41a26677b7f65911b4045049 Mon Sep 17 00:00:00 2001 From: bacon Date: Fri, 8 Mar 2024 12:01:48 +0300 Subject: [PATCH] code cleanup --- lib/DB_Worker.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/lib/DB_Worker.py b/lib/DB_Worker.py index fcb4238..1c0a600 100644 --- a/lib/DB_Worker.py +++ b/lib/DB_Worker.py @@ -18,7 +18,8 @@ class DBReader: try: sql_con = sqlite3.connect("user.db") cursor = sql_con.cursor() - cursor.execute(f"""SELECT * FROM "{guildid}" """) + cursor.execute("""SELECT * FROM "(guildid)" """, + {'guildid': guildid}) record = cursor.fetchall() return record except sqlite3.Error as _e: @@ -85,9 +86,9 @@ async def prepare_db(guild: int): try: connect = sqlite3.connect('user.db') cursor = connect.cursor() - create_table = (f'''CREATE TABLE IF NOT EXISTS "{guild}" ([userid] INTEGER PRIMARY KEY, [username] TEXT, - [nick] TEXT, [isbot] BOOL, [defaulttracks] TEXT, [usertracks] TEXT)''') - cursor.execute(create_table) + cursor.execute('''CREATE TABLE IF NOT EXISTS "(guild)" ([userid] INTEGER PRIMARY KEY, [username] TEXT, + [nick] TEXT, [isbot] BOOL, [defaulttracks] TEXT, [usertracks] TEXT)''', + {'guild': guild}) cursor.close() except sqlite3.Error as _error: logger.info(_error)