add user to db on any action

1 files changed, 16 insertions, 0 deletions

diff --git a/supergenerator.py b/supergenerator.py
index 4b5dbf9..e3827f5 100644
--- a/supergenerator.py
+++ b/supergenerator.py
@@ -133,6 +133,22 @@ def set_user_config(
         }
 
 
+def ensure_user(
+    user_id,
+    model_name=DEFAULT_MODEL,
+    stream_mode=DEFAULT_STREAM,
+    call_mode=DEFAULT_CALL
+):
+    with shelve.open("users") as db:
+        key = str(user_id)
+
+        if key not in db:
+            db[key] = {
+                "model": model_name,
+                "stream": stream_mode,
+                "call": call_mode,
+            }
+
 def get_user_config(user_id):
     with shelve.open("users") as db:
         return db.get(