Firepup650/bitbot-3.11-fork
2
0
Fork 0
Code Issues Pull requests Activity Actions

add database.users.get_nickname, to get nickname from ID

Browse source
This commit is contained in:
jesopo 2019-12-01 07:05:17 +00:00
parent b385323386
commit 3ce947bb77
1 changed files with 7 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
src/Database.py
View file

@ -80,10 +80,15 @@ class Users(Table):
self.database.execute("DELETE FROM users WHERE user_id=?", self.database.execute("DELETE FROM users WHERE user_id=?",
[user_id]) [user_id])
def get_id(self, server_id: int, nickname: str): def get_id(self, server_id: int, nickname: str):
value = self.database.execute_fetchone("""SELECT user_id FROM value = self.database.execute_fetchone(
users WHERE server_id=? and nickname=?""", "SELECT user_id FROM users WHERE server_id=? and nickname=?",
[server_id, nickname]) [server_id, nickname])
return value if value == None else value[0] return value if value == None else value[0]
def get_nickname(self, server_id: int, user_id: int):
value = self.database.execute_fetchone(
"SELECT nickname FROM users WHERe server_id=? AND user_id=?",
[server_id, user_id])
return (value or [None])[0]
class BotSettings(Table): class BotSettings(Table):
def set(self, setting: str, value: typing.Any): def set(self, setting: str, value: typing.Any):