Yuuki/EliteBot
1
0
Fork 0
forked from Raiza.dev/EliteBot
Code Pull requests Activity

Update db.py - fixed users not saving correctly and added get_user_values.

Browse source
This commit is contained in:
Yuuki Chan 2024-03-01 16:45:00 +09:00
parent eeb38858db
commit 31393fbcfe
1 changed files with 14 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
src/db.py
View file

@ -16,7 +16,7 @@ class Database:
if not inspect(self.engine).has_table(table_name):
self.meta.create_all(self.engine)
def set_user(self, user: str, values: dict):
def set_user_values(self, user: str, values: dict):
with self.engine.connect() as conn:
stmt = select(self.table).where(self.table.c.name == user)
cnt = len(conn.execute(stmt).fetchall())
@ -24,11 +24,22 @@ class Database:
if cnt == 1:
conn.execute((
update(self.table).
values(values)
values(values).
where(self.table.c.name == user)
))
conn.commit()
def get_user(self, user: str, index: int):
def get_user_values(self, user: str):
with self.engine.connect() as conn:
stmt = select(self.table).where(self.table.c.name == user)
cnt = len(conn.execute(stmt).fetchall())
if cnt == 1:
return conn.execute(select(self.table).where(self.table.c.name == user)).fetchall()
else:
return -1
def get_user_value(self, user: str, index: int):
with self.engine.connect() as conn:
stmt = select(self.table).where(self.table.c.name == user)
cnt = len(conn.execute(stmt).fetchall())