LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2024-11-23 01:46:35 +03:00
Code Issues Packages Projects Releases Wiki Activity

Allow using parenthesis or spaces in phones when .get_entity

Browse Source
This commit is contained in:
Lonami Exo 2017-10-01 14:21:17 +02:00
parent 5b6f9b6542
commit 71b6e527a6
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
telethon/telegram_client.py
View File

@ -3,6 +3,8 @@ from datetime import datetime, timedelta
from functools import lru_cache
from mimetypes import guess_type
import re
try:
import socks
except ImportError:
@ -852,10 +854,10 @@ class TelegramClient(TelegramBareClient):
return self(GetChannelsRequest([input_entity]))[0]
if isinstance(entity, str):
if entity.startswith('+') or entity.isdigit():
stripped_phone = re.sub(r'[+-()\s]', '', entity)
if stripped_phone.isdigit():
contacts = self(GetContactsRequest(0))
try:
stripped_phone = entity.strip('+')
return next(
u for u in contacts.users
if u.phone and u.phone.endswith(stripped_phone)