a y{Jh} ã@sŒdZddlmZmZddlmZmZdeeeegefedœdd„Z deeeegefedœd d „Z eed œd d „Z eed œdd„Z dS)aš Payload preparing We have added some utils to make work with payload easier. Basic encode example: .. code-block:: python from aiogram.utils.payload import encode_payload encoded = encode_payload("foo") # result: "Zm9v" Basic decode it back example: .. code-block:: python from aiogram.utils.payload import decode_payload encoded = "Zm9v" decoded = decode_payload(encoded) # result: "foo" Encoding and decoding with your own methods: 1. Create your own cryptor .. code-block:: python from Cryptodome.Cipher import AES from Cryptodome.Util.Padding import pad, unpad class Cryptor: def __init__(self, key: str): self.key = key.encode("utf-8") self.mode = AES.MODE_ECB # never use ECB in strong systems obviously self.size = 32 @property def cipher(self): return AES.new(self.key, self.mode) def encrypt(self, data: bytes) -> bytes: return self.cipher.encrypt(pad(data, self.size)) def decrypt(self, data: bytes) -> bytes: decrypted_data = self.cipher.decrypt(data) return unpad(decrypted_data, self.size) 2. Pass cryptor callable methods to aiogram payload tools .. code-block:: python cryptor = Cryptor("abcdefghijklmnop") encoded = encode_payload("foo", encoder=cryptor.encrypt) decoded = decode_payload(encoded_payload, decoder=cryptor.decrypt) # result: decoded == "foo" é)Úurlsafe_b64decodeÚurlsafe_b64encode)ÚCallableÚOptionalN)ÚpayloadÚencoderÚreturncCs4t|tƒst|ƒ}| d¡}|dur,||ƒ}t|ƒS)z[Encode payload with encoder. Result also will be encoded with URL-safe base64url. zutf-8N)Ú isinstanceÚstrÚencodeÚ _encode_b64)rrZ payload_bytes©r úG/var/www/shaz/venv/lib/python3.9/site-packages/aiogram/utils/payload.pyÚencode_payloadCs   r)rÚdecoderrcCs$t|ƒ}|dur| ¡S||ƒ ¡S)z/Decode URL-safe base64url payload with decoder.N)Ú _decode_b64Údecode)rrZoriginal_payloadr r rÚdecode_payloadUsr)rrcCst|ƒ}| ¡}| dd¡S)zEncode with URL-safe base64url.ú=Ú)rrÚreplace)rZ bytes_payloadZ str_payloadr r rr bsr cCs$|ddt|ƒd7}t| ¡ƒS)zDecode with URL-safe base64url.ré)Úlenrr )rr r rrisr)N)N) Ú__doc__Úbase64rrÚtypingrrr Úbytesrrr rr r r rÚs >þý þý