Finished audio list generator

lib/DB_Worker.py

@@ -1,6 +1,78 @@
 import sqlite3
 
 
+class User:
+    def __init__(self, userid: str = None):
+        self.userid = userid
+        self.list: list = self._lister(self.userid)
+        self._current_index = 0
+
+    def __str__(self) -> str:
+        return self.name
+
+    def __repr__(self) -> str:
+        return (
+            f'<Audio file name={self.name} path={self.userid} type={self.type}>'
+        )
+
+    @property
+    def name(self) -> str:
+        return self.list[self._current_index]
+
+    @property
+    def type(self) -> str:
+        guess = mimetypes.guess_type(f'{self.path}/{self.name}')[0].split('/')[0]
+        return guess
+
+    @classmethod
+    def _lister(cls, path) -> list:
+        _dict: list = []
+        for _files in os.walk(path):
+            _dict.extend(_files)
+            break
+        return _dict[2]
+
+    def __iter__(self):
+        return _ListGenerationIter(self)
+
+
+class _FileAttrs:
+    def __init__(self, name, path, type):
+        self.name = name
+        self.path = path
+        self.type = type
+
+    def __str__(self):
+        return 'Audio Dict Generator'
+
+    def __repr__(self):
+        return f'<File attrs name={self.name} path={self.path} type={self.path}>'
+
+
+class _ListGenerationIter:
+    def __init__(self, list_class):
+        self._list = list_class.list
+        self._name = list_class.name
+        self._type = list_class.type
+        self._path = list_class.path
+
+        self._size = len(self._list)
+        self._current_index = 0
+
+    def __iter__(self):
+        return self
+
+    def __next__(self):
+        if self._current_index < self._size:
+            _name = self._list[self._current_index]
+            _path = self._path
+            _type = self._type
+            self._current_index += 1
+            memb = _FileAttrs(_name, _path, _type)
+            return memb
+        raise StopIteration
+
+
 async def prepare_db(guild: int):
     try:
         connect = sqlite3.connect('user.db')