optimised imports

lib/DB_Worker.py

@@ -1,4 +1,4 @@
-import sqlite3
+from sqlite3 import connect, Error
 
 from lib.Logger import logger
 
@@ -16,13 +16,13 @@ class DBReader:
     @classmethod
     def _read_db(cls, guildid: int) -> list:
         try:
-            sql_con = sqlite3.connect("user.db")
+            sql_con = connect("user.db")
             cursor = sql_con.cursor()
             cursor.execute("""SELECT * FROM "(guildid)" """,
                            {'guildid': guildid})
             record = cursor.fetchall()
             return record
-        except sqlite3.Error as _e:
+        except Error as _e:
             logger.info(f'Error reading DB\n{_e}')
 
     def __iter__(self):
@@ -84,13 +84,13 @@ class _ListGenerationIter:
 
 async def prepare_db(guild: int):
     try:
-        connect = sqlite3.connect('user.db')
-        cursor = connect.cursor()
+        _connect = connect('user.db')
+        cursor = _connect.cursor()
         cursor.execute(f'''CREATE TABLE IF NOT EXISTS "{guild}"
         ([userid] INTEGER PRIMARY KEY, [username] TEXT,
         [nick] TEXT, [isbot] BOOL, [defaulttracks] TEXT, [usertracks] TEXT)''')
         cursor.close()
-    except sqlite3.Error as _error:
+    except Error as _error:
         logger.info(_error)
 
 
@@ -101,12 +101,12 @@ async def work_with_db(db_func: str, data_turple: tuple):
     :param data_turple:
     """
     try:
-        connect = sqlite3.connect('user.db')
-        cursor = connect.cursor()
+        _connect = connect('user.db')
+        cursor = _connect.cursor()
         cursor.execute(db_func, data_turple)
-        connect.commit()
+        _connect.commit()
         cursor.close()
-    except sqlite3.Error as _error:
+    except Error as _error:
         logger.critical(_error)
 
 
@@ -126,10 +126,10 @@ async def read_db(guild: int, user: int, column: str):
                  'defaulttracks': 4,
                  'usertracks': 5}
     try:
-        sql_con = sqlite3.connect("user.db")
+        sql_con = connect("user.db")
         cursor = sql_con.cursor()
         cursor.execute(f"""SELECT * FROM "{guild}" where userid = {user}""")
         record = cursor.fetchone()
         return record[_col_dict[column]]
-    except sqlite3.Error as _error:
+    except Error as _error:
         logger.critical(_error)