Added iter for DB

This commit is contained in:
2022-09-02 09:47:18 +03:00
parent 646008e4d8
commit 99bf0769e5
7 changed files with 46 additions and 26 deletions

View File

@@ -1,6 +1,5 @@
import sqlite3
import disnake
from lib import logger
@@ -8,12 +7,12 @@ from lib import logger
class DB_Reader:
def __init__(self, guildid: int = None):
self.guildid = guildid
self.list = self._read_db(self.guildid)
self._guildid = guildid
self.list = self._read_db(self._guildid)
self._current_index = 0
def __str__(self) -> str:
return str(self.guildid)
return str(self._guildid)
@classmethod
def _read_db(cls, guildid: int) -> list:
@@ -32,12 +31,12 @@ class DB_Reader:
class _DBAttrs:
def __init__(self,
userid,
username,
nick,
isbot,
defaulttracks,
usertracks):
userid: int,
username: str,
nick: str,
isbot: bool,
defaulttracks: None or list,
usertracks: None or list):
self.userid = userid
self.username = username
self.nick = nick
@@ -68,7 +67,7 @@ class _ListGenerationIter:
_userid = self._list[self._current_index][0]
_username = self._list[self._current_index][1]
_nick = self._list[self._current_index][2]
_isbot = self._list[self._current_index][3]
_isbot = bool(self._list[self._current_index][3])
_defaulttracks = self._list[self._current_index][4]
_usertracks = self._list[self._current_index][5]