mir/microbot
2
0
Fork 0
mirror of https://github.com/Nick80835/microbot synced 2025-09-01 06:55:51 +00:00
Code Issues Releases Wiki Activity

reject chats and channels in get_user

Browse Source
This commit is contained in:
Nick80835
2020-09-22 09:20:43 -04:00
parent dcffdb4b70
commit 070b7c192a
1 changed files with 13 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
ubot/modules/_sudo.py
View File

@@ -8,6 +8,7 @@ from platform import python_version
import psutil
from telethon import version
from telethon.tl.types import Channel, Chat
from ubot import ldr, micro_bot
@@ -176,7 +177,12 @@ async def get_user(event):
pass
try:
return await event.client.get_entity(event.args)
user = await event.client.get_entity(event.args)
if isinstance(user, (Chat, Channel)):
raise TypeError
return user
except (ValueError, TypeError):
await event.reply("The ID or username you provided was invalid!")
return
@@ -186,7 +192,12 @@ async def get_user(event):
if reply_id:
try:
return await event.client.get_entity(reply_id)
user = await event.client.get_entity(reply_id)
if isinstance(user, (Chat, Channel)):
raise TypeError
return user
except (ValueError, TypeError):
await event.reply("There was an error getting the user's ID!")
return