From adb574ab528fbc4401fb7cd50b8748d1fb529450 Mon Sep 17 00:00:00 2001 From: clarkzjw Date: Thu, 23 Feb 2023 00:30:52 -0800 Subject: bot: multi user working --- command.py | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) (limited to 'command.py') 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: -- cgit v1.2.3