mirror of
https://gitlab.com/MoonTestUse1/AdministrationItDepartmens.git
synced 2025-08-14 00:25:46 +02:00
113 lines
2.9 KiB
Python
113 lines
2.9 KiB
Python
from __future__ import annotations
|
|
|
|
__all__ = [
|
|
"create_start_link",
|
|
"create_startgroup_link",
|
|
"create_deep_link",
|
|
"create_telegram_link",
|
|
"encode_payload",
|
|
"decode_payload",
|
|
]
|
|
|
|
import re
|
|
from typing import TYPE_CHECKING, Callable, Literal, Optional, cast
|
|
|
|
from aiogram.utils.link import create_telegram_link
|
|
from aiogram.utils.payload import decode_payload, encode_payload
|
|
|
|
if TYPE_CHECKING:
|
|
from aiogram import Bot
|
|
|
|
BAD_PATTERN = re.compile(r"[^A-z0-9-]")
|
|
|
|
|
|
async def create_start_link(
|
|
bot: Bot,
|
|
payload: str,
|
|
encode: bool = False,
|
|
encoder: Optional[Callable[[bytes], bytes]] = None,
|
|
) -> str:
|
|
"""
|
|
Create 'start' deep link with your payload.
|
|
|
|
If you need to encode payload or pass special characters -
|
|
set encode as True
|
|
|
|
:param bot: bot instance
|
|
:param payload: args passed with /start
|
|
:param encode: encode payload with base64url or custom encoder
|
|
:param encoder: custom encoder callable
|
|
:return: link
|
|
"""
|
|
username = (await bot.me()).username
|
|
return create_deep_link(
|
|
username=cast(str, username),
|
|
link_type="start",
|
|
payload=payload,
|
|
encode=encode,
|
|
encoder=encoder,
|
|
)
|
|
|
|
|
|
async def create_startgroup_link(
|
|
bot: Bot,
|
|
payload: str,
|
|
encode: bool = False,
|
|
encoder: Optional[Callable[[bytes], bytes]] = None,
|
|
) -> str:
|
|
"""
|
|
Create 'startgroup' deep link with your payload.
|
|
|
|
If you need to encode payload or pass special characters -
|
|
set encode as True
|
|
|
|
:param bot: bot instance
|
|
:param payload: args passed with /start
|
|
:param encode: encode payload with base64url or custom encoder
|
|
:param encoder: custom encoder callable
|
|
:return: link
|
|
"""
|
|
username = (await bot.me()).username
|
|
return create_deep_link(
|
|
username=cast(str, username),
|
|
link_type="startgroup",
|
|
payload=payload,
|
|
encode=encode,
|
|
encoder=encoder,
|
|
)
|
|
|
|
|
|
def create_deep_link(
|
|
username: str,
|
|
link_type: Literal["start", "startgroup"],
|
|
payload: str,
|
|
encode: bool = False,
|
|
encoder: Optional[Callable[[bytes], bytes]] = None,
|
|
) -> str:
|
|
"""
|
|
Create deep link.
|
|
|
|
:param username:
|
|
:param link_type: `start` or `startgroup`
|
|
:param payload: any string-convertible data
|
|
:param encode: encode payload with base64url or custom encoder
|
|
:param encoder: custom encoder callable
|
|
:return: deeplink
|
|
"""
|
|
if not isinstance(payload, str):
|
|
payload = str(payload)
|
|
|
|
if encode or encoder:
|
|
payload = encode_payload(payload, encoder=encoder)
|
|
|
|
if re.search(BAD_PATTERN, payload):
|
|
raise ValueError(
|
|
"Wrong payload! Only A-Z, a-z, 0-9, _ and - are allowed. "
|
|
"Pass `encode=True` or encode payload manually."
|
|
)
|
|
|
|
if len(payload) > 64:
|
|
raise ValueError("Payload must be up to 64 characters long.")
|
|
|
|
return create_telegram_link(username, **{cast(str, link_type): payload})
|