bot: implement /logout command

1 files changed, 3 insertions, 2 deletions

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: