Compare commits
4 commits
058983cab5
...
0198f10463
Author | SHA1 | Date | |
---|---|---|---|
0198f10463 | |||
![]() |
8b92b260f6 | ||
![]() |
31393fbcfe | ||
![]() |
eeb38858db |
2 changed files with 15 additions and 4 deletions
2
plugins
2
plugins
|
@ -1 +1 @@
|
||||||
Subproject commit 375668d49010fdaaf1ff0df0e646704ab83a42be
|
Subproject commit 80152d30028068311733cc8eea5a0b81d5bd8f9a
|
17
src/db.py
17
src/db.py
|
@ -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_user(self, user: str, values: dict):
|
def set_user_values(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())
|
||||||
|
@ -24,11 +24,22 @@ class Database:
|
||||||
if cnt == 1:
|
if cnt == 1:
|
||||||
conn.execute((
|
conn.execute((
|
||||||
update(self.table).
|
update(self.table).
|
||||||
values(values)
|
values(values).
|
||||||
|
where(self.table.c.name == user)
|
||||||
))
|
))
|
||||||
conn.commit()
|
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:
|
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())
|
||||||
|
|
Loading…
Add table
Reference in a new issue