File size: 531 Bytes
a8e9b84 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
from pyrogram.enums import MessageEntityType
from pyrogram.types import Message, User
from DragMusic import app
async def extract_user(m: Message) -> User:
if m.reply_to_message:
return m.reply_to_message.from_user
msg_entities = m.entities[1] if m.text.startswith("/") else m.entities[0]
return await app.get_users(
msg_entities.user.id
if msg_entities.type == MessageEntityType.TEXT_MENTION
else int(m.command[1])
if m.command[1].isdecimal()
else m.command[1]
)
|