maxapi-python 2.3.1__py3-none-any.whl → 2.4.1__py3-none-any.whl

pymax/types/domain/attachments/video.py

@@ -32,8 +32,8 @@ class VideoAttachment(CamelModel):
     :vartype video_id: int
     :ivar duration: Длительность видео.
     :vartype duration: int | None
-    :ivar preview_data: Данные превью.
-    :vartype preview_data: bytes
+    :ivar preview_data: Данные превью, если Max их прислал.
+    :vartype preview_data: bytes | None
     :ivar type: Тип вложения.
     :vartype type: Literal[AttachmentType.VIDEO]
     :ivar thumbnail: URL миниатюры.
@@ -48,7 +48,7 @@ class VideoAttachment(CamelModel):
     width: int
     video_id: int
     duration: int | None = None
-    preview_data: bytes
+    preview_data: bytes | None = None
     type: Literal[AttachmentType.VIDEO] = Field(alias="_type")
     thumbnail: str
     token: str
@@ -62,18 +62,18 @@ class VideoRequest(CamelModel):
     :vartype external: str | bool | None
     :ivar cache: Использовать ли кеш.
     :vartype cache: bool
-    :ivar url: URL видео.
-    :vartype url: str
+    :ivar url: Прямой URL видео или ``None`` для внешнего видео.
+    :vartype url: str | None
     """
 
     external: str | bool | None = Field(default=None, alias="EXTERNAL")
-    cache: bool
-    url: str
+    cache: bool = False  # TODO: idk maybe | None = None better
+    url: str | None = None
 
     @model_validator(mode="before")
     @classmethod
-    def unwrap_dynamic_url(cls, value: Any) -> Any:
-        """Нормализует динамический ключ URL в поле ``url``.
+    def select_video_url(cls, value: Any) -> Any:
+        """Выбирает прямой URL с максимальным доступным MP4-качеством.
 
         :param value: Значение, переданное в валидатор модели.
         :type value: Any
@@ -83,8 +83,29 @@ class VideoRequest(CamelModel):
         if not isinstance(value, dict) or "url" in value:
             return value
 
+        mp4_urls: list[tuple[int, str]] = []
         for key, url in value.items():
-            if key not in ("EXTERNAL", "cache"):
-                return {**value, "url": url}
+            if not isinstance(key, str) or not isinstance(url, str):
+                continue
+
+            normalized_key = key.upper()
+            if not normalized_key.startswith("MP4_"):
+                continue
+
+            try:
+                quality = int(normalized_key.removeprefix("MP4_"))
+            except ValueError:
+                continue
+
+            if quality > 0:
+                mp4_urls.append((quality, url))
+
+        if mp4_urls:
+            _, url = max(mp4_urls, key=lambda item: item[0])
+            return {**value, "url": url}
+
+        legacy_url = value.get("dynamicUrl", value.get("dynamic_url"))
+        if isinstance(legacy_url, str):
+            return {**value, "url": legacy_url}
 
         return value

pymax/types/domain/chat.py

@@ -156,25 +156,24 @@ class Chat(CamelModel):
 
     async def answer(
         self,
-        text: str,
+        text: str | None = None,
         reply_to: int | None = None,
         attachments: SendAttachments = None,
         *,
         notify: bool = True,
-    ) -> Message | None:
+    ) -> Message:
         """Отправляет сообщение в этот чат.
 
         :param text: Текст сообщения.
-        :type text: str
+        :type text: str | None
         :param reply_to: ID сообщения для ответа.
         :type reply_to: int | None
         :param attachments: Файлы, фотографии или видео для отправки.
         :type attachments: SendAttachments
         :param notify: Отправить ли получателям push-уведомление.
         :type notify: bool
-        :returns: Отправленное сообщение или ``None``, если сервер не вернул
-            его.
-        :rtype: Message | None
+        :returns: Отправленное сообщение.
+        :rtype: Message
         :raises RuntimeError: Если чат не привязан к клиенту.
         """
         actions, _ = self._bound()
@@ -198,7 +197,7 @@ class Chat(CamelModel):
         get_chat: bool = False,
         get_messages: bool = True,
         interactive: bool = False,
-    ) -> list[Message] | None:
+    ) -> list[Message]:
         """Загружает историю сообщений этого чата.
 
         ``from_time``, ``backward_time`` и ``forward_time`` передаются в
@@ -224,8 +223,9 @@ class Chat(CamelModel):
         :type get_messages: bool
         :param interactive: Пометить запрос как интерактивный.
         :type interactive: bool
-        :returns: Сообщения или ``None``, если сервер не вернул список.
-        :rtype: list[Message] | None
+        :returns: Список сообщений. Если сервер не вернул сообщения, список
+            пуст.
+        :rtype: list[Message]
         :raises RuntimeError: Если чат не привязан к клиенту.
         """
         actions, _ = self._bound()

pymax/types/domain/enums.py

@@ -22,3 +22,10 @@ class MessageStatus(str, Enum):
 
     EDITED = "EDITED"
     REMOVED = "REMOVED"
+
+
+class LinkType(str, Enum):
+    """Тип связи входящего сообщения с исходным сообщением."""
+
+    REPLY = "REPLY"
+    FORWARD = "FORWARD"

pymax/types/domain/handshake.py

@@ -0,0 +1,16 @@
+from pydantic import Field
+
+from .base import CamelModel
+
+
+class HandshakeResponse(CamelModel):
+    """Результат рукопожатия.
+
+    :ivar calls_seed: Сид для генерации хэшей вызовов.
+    :vartype calls_seed: int
+    """
+
+    calls_seed: int | None = None
+    app_update_type: int | None = Field(
+        alias="app-update-type", default=None
+    )  # very bizarre casing

pymax/types/domain/login.py

@@ -13,14 +13,26 @@ class LoginConfig(CamelModel):
     hash: ConfigHash | None = None
 
 
+class Login2Flags(CamelModel):
+    config_enabled: bool = False
+    contact_enabled: bool = False
+    profile_enabled: bool = False
+
+    @property
+    def enabled(self) -> bool:
+        return self.config_enabled or self.contact_enabled or self.profile_enabled
+
+
 class LoginResponse(CamelModel):
     chats: list[Chat] = Field(default_factory=list)
-    profile: Profile
+    profile: Profile | None = None
     messages: dict[int, list[Message]] = Field(default_factory=dict)  # chat_id -> [message]
     contacts: list[User | None] = Field(default_factory=list)
     token: str | None = None
     time: int | None = None
     config: LoginConfig | None = None
+    updates: int | None = None
+    login2_flags: Login2Flags | None = None
 
     def update_sync_state(self, current: SyncState) -> SyncState:
         sync_time = self.time
@@ -33,3 +45,20 @@ class LoginResponse(CamelModel):
             presence_sync=(sync_time if sync_time is not None else current.presence_sync),
             config_hash=(config_hash if config_hash is not None else current.config_hash),
         )
+
+
+class Login2Response(CamelModel):
+    profile: Profile | None = None
+    contacts: list[User | None] = Field(default_factory=list, alias="contactInfos")
+    config: LoginConfig | None = None
+
+    def update_sync_state(self, current: SyncState) -> SyncState:
+        config_hash = self.config.hash if self.config is not None else None
+
+        return SyncState(
+            chats_sync=current.chats_sync,
+            contacts_sync=current.contacts_sync,
+            drafts_sync=current.drafts_sync,
+            presence_sync=current.presence_sync,
+            config_hash=(config_hash if config_hash is not None else current.config_hash),
+        )

pymax/types/domain/message.py

@@ -1,11 +1,11 @@
 from __future__ import annotations
 
 from collections.abc import Sequence
-from typing import TYPE_CHECKING, Annotated, Any, TypeAlias
+from typing import TYPE_CHECKING, Annotated, Any, Literal, TypeAlias
 
 from pydantic import Field, PrivateAttr, model_validator
 
-from pymax.files import File, Photo, Video
+from pymax.files import File, Photo, Video, VideoNote, Voice
 from pymax.types.domain import (
     AudioAttachment,
     CallAttachment,
@@ -14,6 +14,8 @@ from pymax.types.domain import (
     FileAttachment,
     InlineKeyboardAttachment,
     PhotoAttachment,
+    Poll,
+    PollAttachment,
     ShareAttachment,
     StickerAttachment,
     UnknownAttachment,
@@ -22,10 +24,10 @@ from pymax.types.domain import (
 
 from .base import CamelModel
 from .element import Element
-from .enums import MessageStatus
+from .enums import AccessType, LinkType, MessageStatus
 
 if TYPE_CHECKING:
-    from pymax.api.messages.service import MessageService
+    from pymax.api.messages import DateTimeUnion, MessageService
 
 
 KnownAttachment: TypeAlias = Annotated[
@@ -38,12 +40,14 @@ KnownAttachment: TypeAlias = Annotated[
     | ControlAttachment
     | InlineKeyboardAttachment
     | ShareAttachment
-    | CallAttachment,
+    | CallAttachment
+    | PollAttachment,
     Field(discriminator="type"),
 ]
 Attachment: TypeAlias = KnownAttachment | UnknownAttachment
-SendAttachment: TypeAlias = Photo | File | Video
+SendAttachment: TypeAlias = Photo | File | Video | Poll | Voice | VideoNote
 SendAttachments: TypeAlias = Sequence[SendAttachment] | None
+Link: TypeAlias = Annotated["ReplyLink | ForwardLink", Field(discriminator="type")]
 
 
 class ReactionCounter(CamelModel):
@@ -88,6 +92,66 @@ class ReadState(CamelModel):
     mark: int
 
 
+class DelayedAttributes(CamelModel):
+    time_to_fire: int
+    notify_sender: bool
+    notify_opponents: bool
+
+
+class ReplyLink(CamelModel):
+    """Источник сообщения, отправленного как reply.
+
+    :ivar message: Сообщение, на которое ответили.
+    :vartype message: Message
+    :ivar chat_id: ID чата исходного сообщения.
+    :vartype chat_id: int
+    """
+
+    type: Literal[LinkType.REPLY] = LinkType.REPLY
+    message: Message
+    chat_id: int
+
+
+class ForwardLink(CamelModel):
+    """Источник пересланного сообщения.
+
+    Помимо исходного сообщения содержит доступные в payload-е имя, ссылку,
+    тип доступа и иконку исходного чата. Для скрытого источника эти метаданные
+    могут быть ``None``.
+
+    :ivar message: Исходное сообщение.
+    :vartype message: Message
+    :ivar chat_id: ID исходного чата.
+    :vartype chat_id: int
+    """
+
+    type: Literal[LinkType.FORWARD] = LinkType.FORWARD
+    message: Message
+    chat_id: int
+    chat_name: str | None = None
+    chat_link: str | None = None
+    chat_access_type: AccessType | None = None
+    chat_icon_url: str | None = None
+
+
+"""
+            "link": {
+        "type": "FORWARD",
+        "message": {
+          "id": "116641336752745888",
+          "time": 1779805553478,
+          "type": "CHANNEL",
+          "text": "Добро пожаловать на наше шоу!",
+          "attaches": []
+        },
+        "chatId": -75183661482741,
+        "chatName": "Max Bounty Test",
+        "chatLink": "https://max.ru/join/htlT5aTtlpAgcno_hc93lLM44Dxe45HvOEBqx22lKms",
+        "chatAccessType": "PRIVATE"
+      }
+"""
+
+
 class Message(CamelModel):
     """Сообщение Max с методами для действий над ним.
 
@@ -127,7 +191,8 @@ class Message(CamelModel):
     :vartype type: str
     :ivar cid: Клиентский ID сообщения, если он есть в payload-е.
     :vartype cid: int | None
-    :ivar attaches: Вложения сообщения.
+    :ivar attaches: Вложения сообщения. Неизвестный ``_type`` представлен как
+        ``UnknownAttachment`` с сохраненными дополнительными полями.
     :vartype attaches: list[Attachment]
     :ivar stats: Дополнительная статистика сообщения от Max.
     :vartype stats: dict[str, Any] | None
@@ -147,6 +212,12 @@ class Message(CamelModel):
     :vartype mark: int | None
     :ivar elements: Форматированные элементы текста сообщения.
     :vartype elements: list[Element]
+    :ivar delayed_attributes: Время срабатывания в миллисекундах и настройки
+        уведомлений для отложенного сообщения; ``None`` для обычного.
+    :vartype delayed_attributes: DelayedAttributes | None
+    :ivar link: Исходное сообщение и чат для reply или forward; ``None`` для
+        обычного сообщения без связи.
+    :vartype link: ReplyLink | ForwardLink | None
     """
 
     id: int
@@ -166,6 +237,8 @@ class Message(CamelModel):
     unread: int | None = None
     mark: int | None = None
     elements: list[Element] = Field(default_factory=list)
+    delayed_attributes: DelayedAttributes | None = None
+    link: Link | None = None
 
     _actions: MessageService | None = PrivateAttr(default=None)
 
@@ -182,22 +255,25 @@ class Message(CamelModel):
 
     async def reply(
         self,
-        text: str,
+        text: str | None = None,
         attachments: SendAttachments = None,
         *,
         notify: bool = True,
-    ) -> Message | None:
+        send_at: DateTimeUnion | None = None,
+    ) -> Message:
         """Отправляет ответ на это сообщение в тот же чат.
 
         :param text: Текст сообщения.
-        :type text: str
+        :type text: str | None
         :param attachments: Файлы, фотографии или видео для отправки.
         :type attachments: SendAttachments
         :param notify: Отправить ли получателям push-уведомление.
         :type notify: bool
-        :returns: Отправленное сообщение или ``None``, если сервер не вернул
-            его.
-        :rtype: Message | None
+        :param send_at: Абсолютный ``datetime``, относительный ``timedelta``
+            или Unix time в секундах. ``None`` и ``0`` отправляют сразу.
+        :type send_at: DateTimeUnion | None
+        :returns: Отправленное сообщение.
+        :rtype: Message
         :raises RuntimeError: Если сообщение не привязано к сервису или не
             содержит ``chat_id``.
         """
@@ -209,29 +285,33 @@ class Message(CamelModel):
             reply_to=self.id,
             attachments=attachments,
             notify=notify,
+            send_at=send_at,
         )
 
     async def answer(
         self,
-        text: str,
+        text: str | None = None,
         reply_to: int | None = None,
         attachments: SendAttachments = None,
         *,
         notify: bool = True,
-    ) -> Message | None:
+        send_at: DateTimeUnion | None = None,
+    ) -> Message:
         """Отправляет сообщение в тот же чат.
 
         :param text: Текст сообщения.
-        :type text: str
+        :type text: str | None
         :param reply_to: ID сообщения для ответа.
         :type reply_to: int | None
         :param attachments: Файлы, фотографии или видео для отправки.
         :type attachments: SendAttachments
         :param notify: Отправить ли получателям push-уведомление.
         :type notify: bool
-        :returns: Отправленное сообщение или ``None``, если сервер не вернул
-            его.
-        :rtype: Message | None
+        :param send_at: Абсолютный ``datetime``, относительный ``timedelta``
+            или Unix time в секундах. ``None`` и ``0`` отправляют сразу.
+        :type send_at: DateTimeUnion | None
+        :returns: Отправленное сообщение.
+        :rtype: Message
         :raises RuntimeError: Если сообщение не привязано к сервису или не
             содержит ``chat_id``.
         """
@@ -243,6 +323,7 @@ class Message(CamelModel):
             reply_to=reply_to,
             attachments=attachments,
             notify=notify,
+            send_at=send_at,
         )
 
     async def forward(
@@ -250,15 +331,15 @@ class Message(CamelModel):
         chat_id: int,
         *,
         notify: bool = True,
-    ) -> Message | None:
+    ) -> Message:
         """Пересылает это сообщение в другой чат.
 
         :param chat_id: ID целевого чата.
         :type chat_id: int
         :param notify: Отправить ли получателям push-уведомление.
         :type notify: bool
-        :returns: Пересланное сообщение или ``None``, если сервер его не вернул.
-        :rtype: Message | None
+        :returns: Пересланное сообщение.
+        :rtype: Message
         :raises RuntimeError: Если сообщение не привязано к сервису или не
             содержит ``chat_id``.
         """
@@ -291,13 +372,13 @@ class Message(CamelModel):
 
     async def edit(
         self,
-        text: str,
+        text: str | None = None,
         attachments: SendAttachments = None,
     ) -> Message:
         """Редактирует текст и вложения этого сообщения.
 
         :param text: Новый текст сообщения с поддержкой markdown.
-        :type text: str
+        :type text: str | None
         :param attachments: Новые файлы, фотографии или видео для сообщения.
         :type attachments: SendAttachments
         :returns: Отредактированное сообщение.
@@ -361,7 +442,7 @@ class Message(CamelModel):
 
         return await actions.add_reaction(
             chat_id=chat_id,
-            message_id=str(self.id),  # :C
+            message_id=self.id,
             reaction=reaction,
         )
 
@@ -377,7 +458,7 @@ class Message(CamelModel):
 
         return await actions.remove_reaction(
             chat_id=chat_id,
-            message_id=str(self.id),  # :C x2
+            message_id=self.id,
         )
 
     async def get_reactions(self) -> dict[str, ReactionInfo] | None:
@@ -393,7 +474,7 @@ class Message(CamelModel):
 
         return await actions.get_reactions(
             chat_id=chat_id,
-            message_ids=[str(self.id)],  # :C x3
+            message_ids=[self.id],
         )
 
     def _bound(self) -> tuple[MessageService, int]:

pymax/types/domain/user.py

@@ -71,10 +71,8 @@ class User(CamelModel):
     phone: int | None = None
     status: str | None = None
     description: str | None = None
-    # Bots may send ``gender`` as a numeric code and ``web_app`` as a URL
-    # string instead of an object; accept these so profile parsing won't fail.
     gender: str | int | None = None
-    link: str | None = None
+    link: str | int | None = None
     web_app: dict[str, Any] | str | None = None
     menu_button: dict[str, Any] | None = None
 

pymax/types/events/__init__.py

@@ -5,3 +5,4 @@ from .presence import PresenceEvent
 from .reaction import ReactionUpdateEvent
 from .typing import TypingEvent
 from .video import VideoUploadSignal
+from .voice import AudioUploadSignal

pymax/types/events/reaction.py

@@ -17,5 +17,5 @@ class ReactionUpdateEvent(CamelModel):
 
     message_id: str
     chat_id: int
-    counters: list[ReactionCounter]
-    total_count: int
+    counters: list[ReactionCounter] | None = None
+    total_count: int = 0

pymax/types/events/voice.py

@@ -0,0 +1,5 @@
+from pymax.types.domain.base import CamelModel
+
+
+class AudioUploadSignal(CamelModel):
+    audio_id: int

pymax/versions/__init__.py

@@ -0,0 +1,2 @@
+from .catalog import VersionCatalog
+from .exceptions import VersionNotFoundError

pymax/versions/catalog.py

@@ -0,0 +1,104 @@
+import json
+from importlib import resources
+from typing import Any
+
+import aiohttp
+
+from pymax.fingerprint.models import ApkBuildFingerprint
+
+from .exceptions import VersionAlreadyExistsError, VersionNotFoundError
+
+
+class VersionCatalog:
+    """Каталог fingerprints Android-клиента Max.
+
+    Встроенные версии загружаются при создании объекта. При ``remote=True``
+    вызов :meth:`load` дополнительно получает каталог с
+    ``DEFAULT_REMOTE_URL`` и объединяет его со встроенным. Ошибки сети, HTTP и
+    валидации удаленного ответа не подавляются.
+
+    Args:
+        remote: Загружать ли удаленные версии при :meth:`load`.
+    """
+
+    RECOMMENDED_APP_VERSION: str = "26.25.0"
+    DEFAULT_REMOTE_URL: str = "https://hashes.pymax.org/versions.json"
+
+    def __init__(self, remote: bool = False) -> None:
+        self.path = resources.files("pymax._data") / "apk_fingerprints.json"
+        self.remote = remote
+        self._data = self._load_fingerprints()
+        self.versions = self._convert_data()
+
+    @staticmethod
+    def recommended() -> str:
+        """Возвращает версию, используемую ``Client`` по умолчанию."""
+        return VersionCatalog.RECOMMENDED_APP_VERSION
+
+    async def load(self) -> "VersionCatalog":
+        """Загружает удаленные версии, если включен ``remote``.
+
+        Returns:
+            Этот же каталог с добавленными удаленными версиями.
+        """
+        if not self.remote:
+            return self
+
+        versions = await self._fetch()
+        self.versions.update(versions)
+
+        return self
+
+    def add(self, version: str, fingerprint: ApkBuildFingerprint, override: bool = False) -> None:
+        """Добавляет fingerprint версии в текущий каталог.
+
+        Args:
+            version: Строка версии Android-клиента.
+            fingerprint: Fingerprint и build number этой версии.
+            override: Заменить существующую запись. По умолчанию ``False``.
+
+        Raises:
+            ValueError: Если версия уже существует и ``override=False``.
+        """
+        if self.versions.get(version) is not None and override is False:
+            raise VersionAlreadyExistsError(version)
+
+        self.versions[version] = fingerprint
+
+    def resolve(self, version: str) -> ApkBuildFingerprint:
+        """Возвращает fingerprint версии.
+
+        Raises:
+            VersionNotFoundError: Если версии нет в каталоге.
+        """
+        try:
+            return self.versions[version]
+        except KeyError as e:
+            raise VersionNotFoundError(version) from e
+
+    async def _fetch(self) -> dict[str, ApkBuildFingerprint]:
+        async with (
+            aiohttp.ClientSession() as session,
+            session.get(self.DEFAULT_REMOTE_URL) as response,
+        ):
+            response.raise_for_status()
+            data = await response.json()
+
+            if not isinstance(data, dict):
+                raise ValueError  # TODO: msg
+
+            versions: dict[str, ApkBuildFingerprint] = {}
+            for k, v in data.items():
+                versions[k] = ApkBuildFingerprint.model_validate(v)
+
+            return versions
+
+    def _load_fingerprints(self) -> dict[str, Any]:
+        with self.path.open("r", encoding="utf-8") as f:
+            return json.load(f)
+
+    def _convert_data(self) -> dict[str, ApkBuildFingerprint]:
+        versions: dict[str, ApkBuildFingerprint] = {}
+        for k, v in self._data.items():
+            versions[k] = ApkBuildFingerprint.model_validate(v)
+        return versions

pymax/versions/exceptions.py

@@ -0,0 +1,10 @@
+class VersionNotFoundError(ValueError):
+    """Выбранная версия приложения отсутствует в каталоге fingerprints."""
+
+    def __init__(self, version: str) -> None:
+        super().__init__(f"Could not find version {version} in registry")
+
+
+class VersionAlreadyExistsError(ValueError):
+    def __init__(self, version: str) -> None:
+        super().__init__(f"Version {version} already exists in registry")