Третий коммит, добавление share, share_kb, а также ADMIN_ID

myenv/Lib/site-packages/aiogram/utils/payload.py

@@ -0,0 +1,109 @@
+"""
+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"
+
+"""
+
+from base64 import urlsafe_b64decode, urlsafe_b64encode
+from typing import Callable, Optional
+
+
+def encode_payload(
+    payload: str,
+    encoder: Optional[Callable[[bytes], bytes]] = None,
+) -> str:
+    """Encode payload with encoder.
+
+    Result also will be encoded with URL-safe base64url.
+    """
+    if not isinstance(payload, str):
+        payload = str(payload)
+
+    payload_bytes = payload.encode("utf-8")
+    if encoder is not None:
+        payload_bytes = encoder(payload_bytes)
+
+    return _encode_b64(payload_bytes)
+
+
+def decode_payload(
+    payload: str,
+    decoder: Optional[Callable[[bytes], bytes]] = None,
+) -> str:
+    """Decode URL-safe base64url payload with decoder."""
+    original_payload = _decode_b64(payload)
+
+    if decoder is None:
+        return original_payload.decode()
+
+    return decoder(original_payload).decode()
+
+
+def _encode_b64(payload: bytes) -> str:
+    """Encode with URL-safe base64url."""
+    bytes_payload: bytes = urlsafe_b64encode(payload)
+    str_payload = bytes_payload.decode()
+    return str_payload.replace("=", "")
+
+
+def _decode_b64(payload: str) -> bytes:
+    """Decode with URL-safe base64url."""
+    payload += "=" * (4 - len(payload) % 4)
+    return urlsafe_b64decode(payload.encode())