bacon/discord_bot
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

code cleanup

Browse Source
This commit is contained in:
bacon
2024-03-08 11:55:27 +03:00
parent 961b260887
commit ed31858a8c
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lib/DB_Worker.py
View File

@@ -86,7 +86,7 @@ async def prepare_db(guild: int):
connect = sqlite3.connect('user.db') connect = sqlite3.connect('user.db')
cursor = connect.cursor() cursor = connect.cursor()
create_table = (f'''CREATE TABLE IF NOT EXISTS "{guild}" ([userid] INTEGER PRIMARY KEY, [username] TEXT, 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.execute(create_table)
cursor.close() cursor.close()
except sqlite3.Error as _error: except sqlite3.Error as _error:
@@ -141,7 +141,7 @@ async def read_db(guild: int, user: int, column: str):
try: try:
sql_con = sqlite3.connect("user.db") sql_con = sqlite3.connect("user.db")
cursor = sql_con.cursor() 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}) {'guild': guild, 'user': user})
record = cursor.fetchone() record = cursor.fetchone()
return record[_col_dict[column]] return record[_col_dict[column]]