From cbfde03e620188d80213a8b15a4dffdb8948b257 Mon Sep 17 00:00:00 2001 From: clarkzjw Date: Thu, 23 Feb 2023 21:18:49 -0800 Subject: bot: implement /logout command --- command.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'command.py') diff --git a/command.py b/command.py index 6feeec7..b12483a 100644 --- a/command.py +++ b/command.py @@ -2,7 +2,7 @@ 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, get_user_home_instance +from dbstore.peewee_store import get_user_access_key, get_user_home_instance, delete_user_by_id from config import * @@ -46,7 +46,8 @@ async def list_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> No async def logout_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None: - await update.message.reply_text(PROMPT_LOGOUT, parse_mode=ParseMode.HTML, reply_markup=MAIN_MENU) + if delete_user_by_id(str(update.effective_user.id)): + await update.message.reply_text(PROMPT_LOGOUT_SUCCESS, parse_mode=ParseMode.HTML, reply_markup=LOGIN_MENU) async def toggle_visibility_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None: -- cgit v1.2.3