From d7c1835ed41d50d8e0fd42fd8c33fa760ce24bb5 Mon Sep 17 00:00:00 2001 From: clarkzjw Date: Thu, 23 Feb 2023 21:11:05 -0800 Subject: bot: implement /list command --- command.py | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) (limited to 'command.py') diff --git a/command.py b/command.py index 22f7066..6feeec7 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 +from dbstore.peewee_store import get_user_access_key, get_user_home_instance from config import * @@ -33,7 +33,16 @@ async def tos_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> Non async def list_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None: - await update.message.reply_text(PROMPT_LIST, parse_mode=ParseMode.HTML, reply_markup=MAIN_MENU) + result = get_user_home_instance(str(update.effective_user.id)) + if len(result) == 0: + await update.message.reply_text(PROMPT_LIST_NO_RESULT, parse_mode=ParseMode.HTML) + else: + await update.message.reply_text(f"You are linked with the following Fediverse accounts:\n\n" + f"Instance: {result['home_instance']}\n" + f"Instance type: {result['home_instance_type']}\n" + f"Default visibility: {result['default_visibility']}\n", + parse_mode=ParseMode.HTML, + reply_markup=MAIN_MENU) async def logout_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None: -- cgit v1.2.3