From 31393fbcfec27377a968e02d1c496466e9699248 Mon Sep 17 00:00:00 2001 From: Yuuki Chan Date: Fri, 1 Mar 2024 16:45:00 +0900 Subject: [PATCH] Update db.py - fixed users not saving correctly and added `get_user_values`. --- src/db.py | 17 ++++++++++++++--- 1 file changed, 14 insertions(+), 3 deletions(-) diff --git a/src/db.py b/src/db.py index 96a4cc5..72b6b8c 100644 --- a/src/db.py +++ b/src/db.py @@ -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())