bot: multi user working

1 files changed, 9 insertions, 0 deletions

diff --git a/dbstore/peewee_store.py b/dbstore/peewee_store.py
index 3d37140..272e230 100644
--- a/dbstore/peewee_store.py
+++ b/dbstore/peewee_store.py
@@ -41,6 +41,15 @@ def get_user_by_state(state: str) -> dict:
             return {}
 
 
+def get_user_access_key(telegram_user_id: str) -> str:
+    with db.connection_context():
+        try:
+            user = User.get(User.telegram_user_id == telegram_user_id)
+            return user.access_key
+        except DoesNotExist:
+            return ""
+
+
 class Location(BaseModel):
     fsq_id = CharField(unique=True, primary_key=True)
     name = CharField(max_length=128)