j |dZddlmZddlmZmZddlmZerddlm Z  d d dZ d d dZ dd Z dd Z y)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" ) annotations)urlsafe_b64decodeurlsafe_b64encode) TYPE_CHECKING)CallableNct|ts t|}|jd}|||}t|S)z[Encode payload with encoder. Result also will be encoded with URL-safe base64url. zutf-8) isinstancestrencode _encode_b64)payloadencoder payload_bytess H/var/www/html/venv/lib/python3.12/site-packages/aiogram/utils/payload.pyencode_payloadrIs@ gs #g,NN7+M . } %%cht|}||jS||jS)z/Decode URL-safe base64url payload with decoder.) _decode_b64decode)r decoderoriginal_payloads rdecode_payloadr[s8 #7+&&(( # $ + + --rc\t|}|j}|jddS)zEncode with URL-safe base64url.=)rrreplace)r bytes_payload str_payloads rr r hs-,W5M&&(K   sB ''rcb|ddt|dzz zz }t|jS)zDecode with URL-safe base64url.r)lenrr )r s rrros1 sa#g,**++G W^^- ..r)N)r r rCallable[[bytes], bytes] | Nonereturnr )r r rr"r#r )r bytesr#r )r r r#r$)__doc__ __future__rbase64rrtypingrcollections.abcrrrr rrrr+sn=~#7 ( 04& & ,& &(04 .  . , .  .(/r