code cleanup

This commit is contained in:
bacon
2024-03-08 11:55:27 +03:00
parent 961b260887
commit ed31858a8c

View File

@@ -86,7 +86,7 @@ async def prepare_db(guild: int):
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) ''')
[nick] TEXT, [isbot] BOOL, [defaulttracks] TEXT, [usertracks] TEXT)''')
cursor.execute(create_table)
cursor.close()
except sqlite3.Error as _error:
@@ -141,7 +141,7 @@ async def read_db(guild: int, user: int, column: str):
try:
sql_con = sqlite3.connect("user.db")
cursor = sql_con.cursor()
cursor.execute(f"""SELECT * FROM "%(guild)" where userid = %(user)""",
cursor.execute("""SELECT * FROM "%(guild)" where userid = %(user)""",
{'guild': guild, 'user': user})
record = cursor.fetchone()
return record[_col_dict[column]]