mir/pyrogram
2
0
Fork 0
mirror of https://github.com/pyrogram/pyrogram synced 2025-08-28 21:07:59 +00:00
Code Issues Releases Wiki Activity

Use base64.urlsafe_b64encode/decode instead of manually passing altchars

Browse Source
This commit is contained in:
Dan 2019-06-19 16:07:22 +02:00
parent 8465c4a977
commit edaced35a7
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pyrogram/client/ext/utils.py
View File

@ -18,16 +18,16 @@
import base64 import base64
import struct import struct
from base64 import b64decode, b64encode
from typing import Union, List from typing import Union, List
import pyrogram import pyrogram
from . import BaseClient from . import BaseClient
from ...api import types from ...api import types
def decode(s: str) -> bytes: def decode(s: str) -> bytes:
s = b64decode(s + "=" * (-len(s) % 4), "-_") s = base64.urlsafe_b64decode(s + "=" * (-len(s) % 4))
r = b"" r = b""
assert s[-1] == 2 assert s[-1] == 2
@ -59,7 +59,7 @@ def encode(s: bytes) -> str:
r += bytes([i]) r += bytes([i])
return b64encode(r, b"-_").decode().rstrip("=") return base64.urlsafe_b64encode(r).decode().rstrip("=")
def get_peer_id(input_peer) -> int: def get_peer_id(input_peer) -> int: