dh~ dZddlmZmZddlmZmZ d dedeeegefdefdZ d ded eeegefdefd Z dedefd Z dedefd 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" )urlsafe_b64decodeurlsafe_b64encode)CallableOptionalNpayloadencoderreturnct|ts t|}|jd}|||}t|S)z[Encode payload with encoder. Result also will be encoded with URL-safe base64url. zutf-8) isinstancestrencode _encode_b64)rr payload_bytess O/var/www/netwell/bot/venv/lib/python3.12/site-packages/aiogram/utils/payload.pyencode_payloadrDs@ gs #g,NN7+M . } %%decodercht|}||jS||jS)z/Decode URL-safe base64url payload with decoder.) _decode_b64decode)rroriginal_payloads rdecode_payloadrVs8 #7+&&(( # $ + + --rc\t|}|j}|jddS)zEncode with URL-safe base64url.=)rrreplace)r bytes_payload str_payloads rrrcs-,W5M&&(K   sB ''rcb|ddt|dzz zz }t|jS)zDecode with URL-safe base64url.r)lenrr )rs rrrjs1 sa#g,**++G W^^- ..r)N) __doc__base64rrtypingrrr bytesrrrrrrr's=~8% 37& & hw~. /& &(37 .  . hw~. / .  .((3(///r