mirror of
https://github.com/Nick80835/microbot
synced 2025-08-24 02:58:26 +00:00
41 lines
1.3 KiB
Python
41 lines
1.3 KiB
Python
|
from telethon.tl.types import Channel, Chat
|
||
|
|
|
||
|
|
|
||
|
|
async def get_user(event, allow_channel=False):
|
||
|
|
if event.args:
|
||
|
|
if event.args.isnumeric():
|
||
|
|
user_id = int(event.args)
|
||
|
|
else:
|
||
|
|
user_id = None
|
||
|
|
|
||
|
|
try:
|
||
|
|
user = await event.client.get_entity(user_id or event.args)
|
||
|
|
|
||
|
|
if isinstance(user, Chat) or (isinstance(user, Channel) and not allow_channel):
|
||
|
|
raise TypeError
|
||
|
|
|
||
|
|
return user
|
||
|
|
except (ValueError, TypeError):
|
||
|
|
await event.reply("The ID or username you provided was invalid!")
|
||
|
|
return
|
||
|
|
elif event.is_reply:
|
||
|
|
reply = await event.get_reply_message()
|
||
|
|
|
||
|
|
if reply and reply.sender_id:
|
||
|
|
try:
|
||
|
|
user = await event.client.get_entity(reply.sender_id)
|
||
|
|
|
||
|
|
if isinstance(user, Chat) or (isinstance(user, Channel) and not allow_channel):
|
||
|
|
raise TypeError
|
||
|
|
|
||
|
|
return user
|
||
|
|
except (ValueError, TypeError):
|
||
|
|
await event.reply("There was an error getting the user's ID!")
|
||
|
|
return
|
||
|
|
else:
|
||
|
|
await event.reply("Give me a user ID, username or reply!")
|
||
|
|
return
|
||
|
|
else:
|
||
|
|
await event.reply("Give me a user ID, username or reply!")
|
||
|
|
return
|