yasirarism/pyrofork
1
0
Fork 0
mirror of https://github.com/Mayuri-Chan/pyrofork.git synced 2026-01-05 06:44:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Handle mismatches in a more pythonic way

Browse source
This commit is contained in:
Dan 2018-03-18 12:12:27 +01:00
parent 1d25b84cde
commit cbd3b71b79
1 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
pyrogram/client/client.py
View file

@ -17,6 +17,7 @@
# along with Pyrogram. If not, see <http://www.gnu.org/licenses/>.
import base64
import binascii
import json
import logging
import math
@ -919,9 +920,11 @@ class Client:
match = self.INVITE_LINK_RE.match(peer_id)
if match:
decoded = base64.b64decode(match.group(1) + "=" * (-len(match.group(1)) % 4), altchars="-_")
try:
decoded = base64.b64decode(match.group(1) + "=" * (-len(match.group(1)) % 4), "-_")
return self.resolve_peer(struct.unpack(">2iq", decoded)[1])
except (AttributeError, binascii.Error, struct.error):
pass
peer_id = peer_id.lower().strip("@+")