Compare commits
No commits in common. "0198f10463938fa9ae20c3900e7f47acd360a88b" and "058983cab5a7a83ef4147f10680a2a1f6cb9076b" have entirely different histories.
0198f10463
...
058983cab5
2 changed files with 4 additions and 15 deletions
2
plugins
2
plugins
|
@ -1 +1 @@
|
|||
Subproject commit 80152d30028068311733cc8eea5a0b81d5bd8f9a
|
||||
Subproject commit 375668d49010fdaaf1ff0df0e646704ab83a42be
|
17
src/db.py
17
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_values(self, user: str, values: dict):
|
||||
def set_user(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,22 +24,11 @@ class Database:
|
|||
if cnt == 1:
|
||||
conn.execute((
|
||||
update(self.table).
|
||||
values(values).
|
||||
where(self.table.c.name == user)
|
||||
values(values)
|
||||
))
|
||||
conn.commit()
|
||||
|
||||
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):
|
||||
def get_user(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())
|
||||
|
|
Loading…
Add table
Reference in a new issue