diff --git a/sukuinote/plugins/log_user_joins.py b/sukuinote/plugins/log_user_joins.py index 8fe8dd1..8ae3341 100644 --- a/sukuinote/plugins/log_user_joins.py +++ b/sukuinote/plugins/log_user_joins.py @@ -2,7 +2,7 @@ import html import asyncio from pyrogram import Client, ContinuePropagation from pyrogram.errors.exceptions.flood_420 import FloodWait -from pyrogram.raw.types import UpdateNewChannelMessage, UpdateNewMessage, MessageService, PeerChat, PeerChannel, MessageActionChatAddUser, MessageActionChatJoinedByLink +from pyrogram.raw.types import UpdateNewChannelMessage, UpdateNewMessage, MessageService, PeerChat, PeerChannel, MessageActionChatAddUser, MessageActionChatJoinedByLink, PeerUser from .. import config, log_errors, slave def sexy_user_name(user): @@ -46,12 +46,16 @@ async def log_user_joins(client, update, users, chats): text += f"{atext} [{sexy_chat_id}]\n" async with lock: if (sexy_chat_id, message.id) not in handled: + if isinstance(message.from_id, PeerUser): + adder = sexy_user_name(users[message.from_id.user_id]) + else: + adder = 'Anonymous' if is_join: - text += f'- User: {sexy_user_name(users[message.from_id])}\n' + text += f'- User: {adder}\n' if isinstance(action, MessageActionChatJoinedByLink): text += f'- Inviter: {sexy_user_name(users[action.inviter_id])}' else: - text += f'- Adder: {sexy_user_name(users[message.from_id])}\n- Added Users:\n' + text += f'- Adder: {adder}\n- Added Users:\n' for user in action.users: text += f'--- {sexy_user_name(users[user])}\n' while True: