def keyboard(bot, update): chat = update.effective_chat # type: Optional[Chat] user = update.effective_user # type: Optional[User] conn_id = con_sql.get_connected_chat(user.id) if conn_id and not conn_id == False: btn1 = "/disconnect - {}".format(tld(chat.id, "keyboard_disconnect")) btn2 = "" btn3 = "" else: if con_sql.get_history(user.id): history = con_sql.get_history(user.id) try: chat_name1 = dispatcher.bot.getChat(history.chat_id1).title except: chat_name1 = "" try: chat_name2 = dispatcher.bot.getChat(history.chat_id2).title except: chat_name2 = "" try: chat_name3 = dispatcher.bot.getChat(history.chat_id3).title except: chat_name3 = "" if chat_name1: btn1 = "/connect {} - {}".format(history.chat_id1, chat_name1) else: btn1 = "/connect - {}".format(tld(chat.id, "keyboard_connect")) if chat_name2: btn2 = "/connect {} - {}".format(history.chat_id2, chat_name2) else: btn2 = "" if chat_name3: btn3 = "/connect {} - {}".format(history.chat_id3, chat_name3) else: btn3 = "" #TODO: Remove except garbage update.effective_message.reply_text( tld(chat.id, "keyboard_updated"), reply_markup=ReplyKeyboardMarkup([[ KeyboardButton("/help - {}".format(tld(chat.id, "keyboard_bothelp"))), KeyboardButton("/notes - {}".format(tld(chat.id, "keyboard_notes"))) ], [KeyboardButton(btn1)], [KeyboardButton(btn2)], [KeyboardButton(btn3)]]))
def connect_chat(bot, update, args): chat = update.effective_chat # type: Optional[Chat] user = update.effective_user # type: Optional[User] if update.effective_chat.type == 'private': if len(args) >= 1: try: connect_chat = int(args[0]) except ValueError: update.effective_message.reply_text( tld(chat.id, "common_err_invalid_chatid")) return if (bot.get_chat_member( connect_chat, update.effective_message.from_user.id).status in ('administrator', 'creator') or (sql.allow_connect_to_chat(connect_chat) == True) and bot.get_chat_member( connect_chat, update.effective_message.from_user.id).status in ('member')) or (user.id in SUDO_USERS): connection_status = sql.connect( update.effective_message.from_user.id, connect_chat) if connection_status: chat_name = dispatcher.bot.getChat( connected(bot, update, chat, user.id, need_admin=False)).title update.effective_message.reply_text( tld(chat.id, "connection_success").format(chat_name), parse_mode=ParseMode.MARKDOWN) #Add chat to connection history history = sql.get_history(user.id) if history: #Vars if history.chat_id1: history1 = int(history.chat_id1) if history.chat_id2: history2 = int(history.chat_id2) if history.chat_id3: history3 = int(history.chat_id3) if history.updated: number = history.updated if number == 1 and connect_chat != history2 and connect_chat != history3: history1 = connect_chat number = 2 elif number == 2 and connect_chat != history1 and connect_chat != history3: history2 = connect_chat number = 3 elif number >= 3 and connect_chat != history2 and connect_chat != history1: history3 = connect_chat number = 1 else: print("Error") print(history.updated) print(number) sql.add_history(user.id, history1, history2, history3, number) # print(history.user_id, history.chat_id1, history.chat_id2, history.chat_id3, history.updated) else: sql.add_history(user.id, connect_chat, "0", "0", 2) # Rebuild user's keyboard keyboard(bot, update) else: update.effective_message.reply_text( tld(chat.id, "connection_fail")) else: update.effective_message.reply_text( tld(chat.id, "connection_err_not_allowed")) else: update.effective_message.reply_text( tld(chat.id, "connection_err_no_chatid")) history = sql.get_history(user.id) # print(history.user_id, history.chat_id1, history.chat_id2, history.chat_id3, history.updated) elif update.effective_chat.type == 'supergroup': connect_chat = chat.id if (bot.get_chat_member(connect_chat, update.effective_message.from_user.id).status in ('administrator', 'creator') or (sql.allow_connect_to_chat(connect_chat) == True) and bot.get_chat_member( connect_chat, update.effective_message.from_user.id).status in 'member') or (user.id in SUDO_USERS): connection_status = sql.connect( update.effective_message.from_user.id, connect_chat) if connection_status: update.effective_message.reply_text( tld(chat.id, "connection_success").format(chat.id), parse_mode=ParseMode.MARKDOWN) else: update.effective_message.reply_text( tld(chat.id, "connection_fail"), parse_mode=ParseMode.MARKDOWN) else: update.effective_message.reply_text( tld(chat.id, "common_err_no_admin")) else: update.effective_message.reply_text(tld(chat.id, "common_cmd_pm_only"))