bot: multi user working

1 files changed, 10 insertions, 4 deletions

diff --git a/command.py b/command.py
index 901b792..7426901 100644
--- a/command.py
+++ b/command.py
@@ -2,15 +2,21 @@ from telegram import Update
 from telegram.constants import ParseMode
 from telegram.error import BadRequest
 from telegram.ext import ContextTypes, ConversationHandler
-
+from dbstore.peewee_store import get_user_access_key
 from config import *
 
 
 async def start_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> int:
     await update.message.reply_text(PROMPT_START, parse_mode=ParseMode.MARKDOWN)
-    await update.message.reply_text(PROMPT_CHOOSE_ACTION, reply_markup=MAIN_MENU)
-
-    return WAIT_LOCATION
+    user_access_key = get_user_access_key(str(update.effective_user.id))
+    # TODO
+    # verify user access key still valid
+    if len(user_access_key) == 0:
+        await update.message.reply_text(PROMPT_FEDI_LOGIN_WHERE_IS_INSTANCE, parse_mode=ParseMode.MARKDOWN)
+        return FEDI_LOGIN
+    else:
+        await update.message.reply_text(PROMPT_CHOOSE_ACTION, reply_markup=MAIN_MENU)
+        return WAIT_LOCATION
 
 
 async def fedi_login_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> int: