Extend commands

lib/DB_Worker.py

@@ -1,5 +1,5 @@
 import sqlite3
-from typing import List
+
 
 from lib import logger
 
@@ -37,36 +37,20 @@ class _DBAttrs:
                  isbot: bool,
                  defaulttracks: None or list,
                  usertracks: None or list):
-        self.id = userid
+        self.userid = userid
         self.username = username
         self.nick = nick
         self.isbot = isbot
-        if defaulttracks is not None:
-            self._def_list = defaulttracks
-            self.defaulttracks = self._defaulttracks
-        else:
-            self.defaulttracks = None
-        if usertracks is not None:
-            self._user_list = usertracks
-            self.usertracks = self._usertracks
-        else:
-            self.usertracks = None
+        self.defaulttracks = defaulttracks
+        self.usertracks = usertracks
 
     def __str__(self):
         return self.username
 
     def __repr__(self):
-        return f'<File attrs userid={self.id} username={self.username} nick={self.nick} ' \
+        return f'<File attrs userid={self.userid} username={self.username} nick={self.nick} ' \
                f'isbot={self.isbot} defaulttracks={self.defaulttracks} usertracks={self.usertracks}>'
 
-    @property
-    def _defaulttracks(self) -> List[str]:
-        return self._def_list.split(', ')
-
-    @property
-    def _usertracks(self) -> List[str]:
-        return self._user_list.split(', ')
-
 
 class _ListGenerationIter:
     def __init__(self, user_class):
@@ -166,11 +150,16 @@ async def read_db(guild: int, user: int, column: str):
         pass
 
 
-async def check_exist_audio(guild: int, user: int, audio: str):
-    _users_db = DB_Reader(guild)
-    for _user in _users_db:
-        if not _user.isbot:
-            if _user.id == user and _user.usertracks is not None and audio in _user.usertracks:
-                return True
-            else:
-                return False
+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'