code cleanup

This commit is contained in:
bacon
2024-03-08 12:01:48 +03:00
parent ed31858a8c
commit 0e9394b057

View File

@@ -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)