Added iter for DB
This commit is contained in:
@@ -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]
|
||||
|
||||
|
||||
Reference in New Issue
Block a user