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 12:01:48 +03:00
parent ed31858a8c
commit 0e9394b057
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
lib/DB_Worker.py
View File

@@ -18,7 +18,8 @@ class DBReader:
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 "{guildid}" """) cursor.execute("""SELECT * FROM "(guildid)" """,
{'guildid': guildid})
record = cursor.fetchall() record = cursor.fetchall()
return record return record
except sqlite3.Error as _e: except sqlite3.Error as _e:
@@ -85,9 +86,9 @@ async def prepare_db(guild: int):
try: try:
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, cursor.execute('''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) {'guild': guild})
cursor.close() cursor.close()
except sqlite3.Error as _error: except sqlite3.Error as _error:
logger.info(_error) logger.info(_error)