Raiza.dev/EliteBot
4
0
Fork 1
Code Issues Pull requests Projects Releases Packages Wiki Activity

Updated db.py

Browse source
This commit is contained in:
Yuuki Chan 2024-02-22 20:31:54 +09:00
parent ace06a855c
commit ebfc76f17b
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/db.py
View file

@ -16,7 +16,7 @@ class Database:
if not inspect(self.engine).has_table(table_name): if not inspect(self.engine).has_table(table_name):
self.meta.create_all(self.engine) self.meta.create_all(self.engine)
def set(self, user: str, values: dict): def set_user(self, user: str, values: dict):
with self.engine.connect() as conn: with self.engine.connect() as conn:
stmt = select(self.table).where(self.table.c.name == user) stmt = select(self.table).where(self.table.c.name == user)
cnt = len(conn.execute(stmt).fetchall()) cnt = len(conn.execute(stmt).fetchall())
@ -28,7 +28,7 @@ class Database:
)) ))
conn.commit() conn.commit()
def get(self, user: str, index: int): def get_user(self, user: str, index: int):
with self.engine.connect() as conn: with self.engine.connect() as conn:
stmt = select(self.table).where(self.table.c.name == user) stmt = select(self.table).where(self.table.c.name == user)
cnt = len(conn.execute(stmt).fetchall()) cnt = len(conn.execute(stmt).fetchall())