def connected(bot, update, chat, user_id, need_admin=True): chat = update.effective_chat # type: Optional[Chat] if chat.type == chat.PRIVATE and sql.get_connected_chat(user_id): conn_id = sql.get_connected_chat(user_id).chat_id if (bot.get_chat_member( conn_id, user_id).status in ('administrator', 'creator') or (sql.allow_connect_to_chat(connect_chat) == True) and bot.get_chat_member( user_id, update.effective_message.from_user.id).status in ('member')) or (user_id in SUDO_USERS): if need_admin: if bot.get_chat_member( conn_id, update.effective_message.from_user.id).status in ( 'administrator', 'creator') or user_id in SUDO_USERS: return conn_id else: update.effective_message.reply_text( tld(chat.id, "connection_err_no_admin")) return else: return conn_id else: update.effective_message.reply_text( tld(chat.id, "connection_err_unknown")) disconnect_chat(bot, update) return else: return False
def connected(bot, update, chat, user_id, need_admin=True): if chat.type == chat.PRIVATE and sql.get_connected_chat(user_id): conn_id = sql.get_connected_chat(user_id).chat_id if (bot.get_chat_member( conn_id, user_id).status in ('administrator', 'creator') or (sql.allow_connect_to_chat(connect_chat) == True) and bot.get_chat_member( user_id, update.effective_message.from_user.id).status in ('member')) or (user_id in SUDO_USERS): if need_admin: if bot.get_chat_member( conn_id, update.effective_message.from_user.id).status in ( 'administrator', 'creator') or user_id in SUDO_USERS: return conn_id else: update.effective_message.reply_text( "You need to be a admin in a connected group!") exit(1) else: return conn_id else: update.effective_message.reply_text( "Group changed rights connection or you are not admin anymore.\nI'll disconnect you." ) disconnect_chat(bot, update) exit(1) else: return False
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"))