refractor lib code

lib/DB.py

@@ -0,0 +1,85 @@
+import sqlite3
+
+
+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.close()
+    except sqlite3.Error as _error:
+        pass
+
+
+async def work_with_db(db_func: str, data_turple: tuple):
+    """
+    Writing to db per server userinfo
+    :param db_func:
+    :param data_turple:
+    """
+    try:
+        connect = sqlite3.connect('user.db')
+        cursor = connect.cursor()
+        cursor.execute(db_func, data_turple)
+        connect.commit()
+        cursor.close()
+    except sqlite3.Error as _error:
+        pass
+
+
+async def fill_bd(name: str, userid: int, isbot: bool, nick: str, guild: int):
+    sqlite_insert_with_param = (f"""INSERT OR IGNORE INTO "{guild}"
+                          (username, userid, nick, isbot) 
+                          VALUES (?, ?, ?, ?)""")
+    data_tuple: tuple[str, int, str, bool] = (name, userid, nick, isbot)
+    await work_with_db(sqlite_insert_with_param, data_tuple)
+
+
+async def add_audio(guild: int, user: int, audio: str, track: str = 'usertracks'):
+    """
+    Adding audio into folder and DB
+    :param guild: Guild id
+    :param user:
+    :param audio:
+    :param track: usertracks or defaulttracks
+    """
+    # audio = f'{DB.read_db(guild, user, track)}, {audio}'
+    sql_update_query = f"""UPDATE "{guild}" set {track} = ? where userid = ?"""
+    data_tuple = (audio, user)
+    await work_with_db(sql_update_query, data_tuple)
+
+
+async def read_db(guild: int, user: int, column: str):
+    _col_dict = {'userid': 0,
+                 'username': 1,
+                 'nick': 2,
+                 'isbot': 3,
+                 'defaulttracks': 4,
+                 'usertracks': 5}
+    try:
+        sql_con = sqlite3.connect("user.db")
+        cursor = sql_con.cursor()
+        sql_read = f"""SELECT * FROM "{guild}" where userid = {user}"""
+        cursor.execute(sql_read)
+        record = cursor.fetchone()
+        return record[_col_dict[column]]
+    except sqlite3.Error as _error:
+        pass
+
+
+async def check_exist_audio(ctx, guild: int, user: int, column: str, audio: str):
+    _list_str = await read_db(guild, user, column)
+    print(type(_list_str))
+    if _list_str is not None:
+        _list = _list_str.split(',')
+        if audio in _list:
+            await ctx.reply("File in list")
+
+        else:
+            pass
+
+    else:
+        _list = 'None'