bot: implement delayed checkin, (kind of), some TODO leftfeature/delayed_checkin

1 files changed, 10 insertions, 0 deletions

diff --git a/dbstore/peewee_store.py b/dbstore/peewee_store.py
index d05c642..ebf3536 100644
--- a/dbstore/peewee_store.py
+++ b/dbstore/peewee_store.py
@@ -23,6 +23,15 @@ class User(BaseModel):
     client_id = CharField(max_length=128)
     client_secret = CharField(max_length=128)
     toot_visibility = CharField(max_length=128, default=TOOT_VISIBILITY_PRIVATE)
+    # delayed checkin in minutes
+    delayed_checkin = IntegerField(default=0)
+
+
+def update_delayed_checkin(telegram_user_id: str, delayed_checkin: int) -> int:
+    with db.connection_context():
+        return User.update(delayed_checkin=delayed_checkin).where(
+            User.telegram_user_id == telegram_user_id
+        ).execute()
 
 
 def update_user_visibility(telegram_user_id: str, visibility: str) -> int:
@@ -45,6 +54,7 @@ def get_user_by_id(telegram_user_id: str) -> dict:
                 "client_id": user.client_id,
                 "client_secret": user.client_secret,
                 "toot_visibility": user.toot_visibility,
+                "delayed_checkin": user.delayed_checkin,
             }
         except DoesNotExist:
             return {}