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)