afk: don't crash if an anonymous message mentions user

squash change from:
    1. afk: don't crash if an anonymous message mentions user [KenHV/KensurBot@7dc089c]
        get_sender() returns None if telegram didn't send the sender.
        this leads to the bot crashing whenever an anonymous message mentions
        the user, as the afk event handler doesn't account for this case.

        let's add a check to handle this case.
    2. afk: fully handle anonymous admin [MrMissx/UserButt@1c29d68]
        follow up for commit MrMissx/UserButt@46a239d

        now the anonymous admin info saved and
        notified on BOTLOG as a group name.

Co-authored-by: KenHV <[email protected]>
Co-authored-by: Mr.Miss <[email protected]>
Signed-off-by: MoveAngel <[email protected]>

Author: 2 people

userbot/modules/afk.py

@@ -121,20 +121,20 @@ async def type_afk_is_not_true(notafk):
                 + " chats while you were away",
             )
             for i in USERS:
-                name = await notafk.client.get_entity(i)
-                name0 = str(name.first_name)
-                await notafk.client.send_message(
-                    BOTLOG_CHATID,
-                    "["
-                    + name0
-                    + "](tg://user?id="
-                    + str(i)
-                    + ")"
-                    + " sent you "
-                    + "`"
-                    + str(USERS[i])
-                    + " messages`",
-                )
+                if str(i).isnumeric():
+                    name = await notafk.client.get_entity(i)
+                    name0 = str(name.first_name)
+                    await notafk.client.send_message(
+                        BOTLOG_CHATID,
+                        "[" + name0 + "](tg://user?id=" + str(i) + ")" +
+                        " sent you " + "`" + str(USERS[i]) + " message(s)`",
+                    )
+                else:  # anon admin
+                    await notafk.client.send_message(
+                        BOTLOG_CHATID,
+                        "Anonymous admin in `" + i + "` sent you " + "`" +
+                        str(USERS[i]) + " message(s)`",
+                    )
         COUNT_MSG = 0
         USERS = {}
         AFKREASON = None
@@ -153,8 +153,7 @@ async def mention_afk(mention):
     back_alivee = datetime.now()
     afk_end = back_alivee.replace(microsecond=0)
     afk_since = "a while ago"
-    if mention.message.mentioned and not (await mention.get_sender()).bot:
-        if ISAFK:
+    if ISAFK and mention.message.mentioned:
             now = datetime.now()
             datime_since_afk = now - afk_time  # pylint:disable=E0602
             time = float(datime_since_afk.seconds)
@@ -182,17 +181,28 @@ async def mention_afk(mention):
                 afk_since = f"`{int(minutes)}m{int(seconds)}s` ago"
             else:
                 afk_since = f"`{int(seconds)}s` ago"
-            if mention.sender_id not in USERS:
+            
+            is_bot = False
+            if (sender := await mention.get_sender()):
+                is_bot = sender.bot
+                if is_bot: return  # ignore bot
+
+            chat_obj = await mention.client.get_entity(mention.chat_id)
+            chat_title = chat_obj.title
+
+            if mention.sender_id not in USERS or chat_title not in USERS:
                 if AFKREASON:
                     await mention.reply(
                         f"I'm AFK since {afk_since}.\
                         \nReason: `{AFKREASON}`"
                     )
                 else:
                     await mention.reply(str(choice(AFKSTR)))
-                USERS.update({mention.sender_id: 1})
-                COUNT_MSG = COUNT_MSG + 1
-            elif mention.sender_id in USERS:
+                if mention.sender_id is not None:
+                    USERS.update({mention.sender_id: 1})
+                else:
+                    USERS.update({chat_title: 1})
+            else:
                 if USERS[mention.sender_id] % randint(2, 4) == 0:
                     if AFKREASON:
                         await mention.reply(
@@ -201,11 +211,11 @@ async def mention_afk(mention):
                         )
                     else:
                         await mention.reply(str(choice(AFKSTR)))
-                    USERS[mention.sender_id] = USERS[mention.sender_id] + 1
-                    COUNT_MSG = COUNT_MSG + 1
-                else:
-                    USERS[mention.sender_id] = USERS[mention.sender_id] + 1
-                    COUNT_MSG = COUNT_MSG + 1
+                    if mention.sender_id is not None:
+                        USERS[mention.sender_id] += 1
+                    else:
+                        USERS[chat_title] += 1
+                COUNT_MSG += 1
 
 
 @register(incoming=True, disable_errors=True)