Catch ValueError in pm command. Fixes #126

This commit is contained in:
Tulir Asokan
2018-04-28 22:05:27 +03:00
parent 193dcc714b
commit 797ff06d10
+5 -1
View File
@@ -64,7 +64,11 @@ async def private_message(evt):
if len(evt.args) == 0: if len(evt.args) == 0:
return await evt.reply("**Usage:** `$cmdprefix+sp pm <user identifier>`") return await evt.reply("**Usage:** `$cmdprefix+sp pm <user identifier>`")
user = await evt.sender.client.get_entity(evt.args[0]) try:
user = await evt.sender.client.get_entity(evt.args[0])
except ValueError:
return await evt.reply("Invalid user identifier or user not found.")
if not user: if not user:
return await evt.reply("User not found.") return await evt.reply("User not found.")
elif not isinstance(user, TLUser): elif not isinstance(user, TLUser):