def reply_afk(update, context): message = update.effective_message userc = update.effective_user userc_id = userc.id if message.entities and message.parse_entities( [MessageEntity.TEXT_MENTION, MessageEntity.MENTION] ): entities = message.parse_entities( [MessageEntity.TEXT_MENTION, MessageEntity.MENTION] ) chk_users = [] for ent in entities: if ent.type == MessageEntity.TEXT_MENTION: user_id = ent.user.id fst_name = ent.user.first_name if user_id in chk_users: return chk_users.append(user_id) elif ent.type == MessageEntity.MENTION: user_id = get_user_id( message.text[ent.offset : ent.offset + ent.length] ) if not user_id: # Should never happen, since for a user to become AFK they must have spoken. Maybe changed username? return if user_id in chk_users: return chk_users.append(user_id) try: chat = context.bot.get_chat(user_id) except BadRequest: print( "Error: Could not fetch userid {} for AFK module".format( user_id ) ) return fst_name = chat.first_name else: return check_afk(update, context, user_id, fst_name, userc_id) elif message.reply_to_message: user_id = message.reply_to_message.from_user.id fst_name = message.reply_to_message.from_user.first_name check_afk(update, context, user_id, fst_name, userc_id)
def extract_unt_fedban( message: Message, args: List[str], ) -> (Optional[int], Optional[str]): prev_message = message.reply_to_message split_text = message.text.split(None, 1) if len(split_text) < 2: return id_from_reply(message) # only option possible text_to_parse = split_text[1] text = "" entities = list(message.parse_entities([MessageEntity.TEXT_MENTION])) ent = entities[0] if entities else None # if entity offset matches (command end/text start) then all good if entities and ent and ent.offset == len( message.text) - len(text_to_parse): ent = entities[0] user_id = ent.user.id text = message.text[ent.offset + ent.length:] elif len(args) >= 1 and args[0][0] == "@": user = args[0] user_id = get_user_id(user) if not user_id and not isinstance(user_id, int): message.reply_text( "I don't have that user in my db. " "You'll be able to interact with them if you reply to that person's message instead, or forward one of that user's messages.", ) return None, None res = message.text.split(None, 2) if len(res) >= 3: text = res[2] elif len(args) >= 1 and args[0].isdigit(): user_id = int(args[0]) res = message.text.split(None, 2) if len(res) >= 3: text = res[2] elif prev_message: user_id, text = id_from_reply(message) else: return None, None try: message.bot.get_chat(user_id) except BadRequest as excp: if excp.message in ("User_id_invalid", "Chat not found") and not isinstance( user_id, int, ): message.reply_text( "I don't seem to have interacted with this user before " "please forward a message from them to give me control! " "(like a voodoo doll, I need a piece of them to be able to execute certain commands...)", ) return None, None if excp.message != "Chat not found": LOGGER.exception("Exception %s on user %s", excp.message, user_id) return None, None if not isinstance(user_id, int): return None, None return user_id, text