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

pymax/types/domain/message.py

@@ -1,7 +1,7 @@
 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
 
@@ -24,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[
@@ -47,6 +47,7 @@ KnownAttachment: TypeAlias = Annotated[
 Attachment: TypeAlias = KnownAttachment | UnknownAttachment
 SendAttachment: TypeAlias = Photo | File | Video | Poll | Voice | VideoNote
 SendAttachments: TypeAlias = Sequence[SendAttachment] | None
+Link: TypeAlias = Annotated["ReplyLink | ForwardLink", Field(discriminator="type")]
 
 
 class ReactionCounter(CamelModel):
@@ -91,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 с методами для действий над ним.
 
@@ -130,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
@@ -150,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
@@ -169,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)
 
@@ -189,6 +259,7 @@ class Message(CamelModel):
         attachments: SendAttachments = None,
         *,
         notify: bool = True,
+        send_at: DateTimeUnion | None = None,
     ) -> Message:
         """Отправляет ответ на это сообщение в тот же чат.
 
@@ -198,6 +269,9 @@ class Message(CamelModel):
         :type attachments: SendAttachments
         :param notify: Отправить ли получателям push-уведомление.
         :type notify: bool
+        :param send_at: Абсолютный ``datetime``, относительный ``timedelta``
+            или Unix time в секундах. ``None`` и ``0`` отправляют сразу.
+        :type send_at: DateTimeUnion | None
         :returns: Отправленное сообщение.
         :rtype: Message
         :raises RuntimeError: Если сообщение не привязано к сервису или не
@@ -211,6 +285,7 @@ class Message(CamelModel):
             reply_to=self.id,
             attachments=attachments,
             notify=notify,
+            send_at=send_at,
         )
 
     async def answer(
@@ -220,6 +295,7 @@ class Message(CamelModel):
         attachments: SendAttachments = None,
         *,
         notify: bool = True,
+        send_at: DateTimeUnion | None = None,
     ) -> Message:
         """Отправляет сообщение в тот же чат.
 
@@ -231,6 +307,9 @@ class Message(CamelModel):
         :type attachments: SendAttachments
         :param notify: Отправить ли получателям push-уведомление.
         :type notify: bool
+        :param send_at: Абсолютный ``datetime``, относительный ``timedelta``
+            или Unix time в секундах. ``None`` и ``0`` отправляют сразу.
+        :type send_at: DateTimeUnion | None
         :returns: Отправленное сообщение.
         :rtype: Message
         :raises RuntimeError: Если сообщение не привязано к сервису или не
@@ -244,6 +323,7 @@ class Message(CamelModel):
             reply_to=reply_to,
             attachments=attachments,
             notify=notify,
+            send_at=send_at,
         )
 
     async def forward(
@@ -362,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,
         )
 
@@ -378,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:
@@ -394,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/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")