LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-02-17 03:51:05 +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 functools import lru_cache
from mimetypes import guess_type from mimetypes import guess_type
import re
try: try:
import socks import socks
except ImportError: except ImportError:
@ -852,10 +854,10 @@ class TelegramClient(TelegramBareClient):
return self(GetChannelsRequest([input_entity]))[0] return self(GetChannelsRequest([input_entity]))[0]
if isinstance(entity, str): 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)) contacts = self(GetContactsRequest(0))
try: try:
stripped_phone = entity.strip('+')
return next( return next(
u for u in contacts.users u for u in contacts.users
if u.phone and u.phone.endswith(stripped_phone) if u.phone and u.phone.endswith(stripped_phone)