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

pymax/protocol/tcp/payload.py

@@ -11,6 +11,8 @@ logger = get_logger(__name__)
 
 
 class MsgpackPayloadCodec:
+    WRAPPED_VALUE_EXT_CODE = 1
+
     def _to_msgpack_value(self, value: Any) -> Any:
         if isinstance(value, Enum):
             return value.value
@@ -30,10 +32,25 @@ class MsgpackPayloadCodec:
     def _unpack_stream(
         self, payload_bytes: bytes, *, raw: bool
     ) -> list[Any]:  # TODO: deprecate? idk
-        unpacker = msgpack.Unpacker(raw=raw, strict_map_key=False)
+        unpacker = msgpack.Unpacker(
+            raw=raw,
+            strict_map_key=False,
+            ext_hook=self._decode_ext,
+        )
         unpacker.feed(payload_bytes)
         return list(unpacker)
 
+    def _decode_ext(self, code: int, data: bytes) -> Any:
+        if code != self.WRAPPED_VALUE_EXT_CODE:
+            return msgpack.ExtType(code, data)
+
+        return msgpack.unpackb(
+            data,
+            raw=False,
+            strict_map_key=False,
+            ext_hook=self._decode_ext,
+        )
+
     def decode(self, payload_bytes: bytes) -> Any:
         if not payload_bytes:
             return {}
@@ -43,6 +60,7 @@ class MsgpackPayloadCodec:
                 payload_bytes,
                 raw=False,
                 strict_map_key=False,
+                ext_hook=self._decode_ext,
             )
 
         except msgpack.exceptions.ExtraData as e:
@@ -81,6 +99,8 @@ class TcpPayloadDecoder:
             return {self._normalize_key(k): self._normalize_keys(v) for k, v in obj.items()}
         if isinstance(obj, list):
             return [self._normalize_keys(item) for item in obj]
+        if isinstance(obj, msgpack.ExtType):
+            return obj
         if isinstance(obj, tuple):
             return tuple(self._normalize_keys(item) for item in obj)
         return obj

pymax/transport/tcp.py

@@ -7,6 +7,7 @@ from pymax.logging import get_logger
 from .base import Transport
 
 logger = get_logger(__name__)
+_CLOSE_TIMEOUT = 5.0
 
 
 class TCPTransport(Transport):
@@ -66,7 +67,11 @@ class TCPTransport(Transport):
         if writer:
             logger.debug("tcp close")
             writer.close()
-            await writer.wait_closed()
+            try:
+                await asyncio.wait_for(writer.wait_closed(), _CLOSE_TIMEOUT)
+            except (OSError, TimeoutError) as e:
+                logger.warning("tcp close did not finish cleanly: %s", e)
+                writer.transport.abort()
             logger.debug("tcp closed")
 
     async def send(self, data: bytes | str) -> None:

pymax/transport/websocket.py

@@ -17,11 +17,14 @@ class WebSocketTransport(Transport):
     async def connect(self) -> None:
         if self.proxy:
             self.ws = await client.connect(
-                self.url, origin=Origin("https://web.max.ru"), proxy=self.proxy
+                self.url,
+                origin=Origin("https://web.max.ru"),
+                proxy=self.proxy,
+                max_size=1024 * 1024 * 10,  # 10 MB
             )
         else:
             self.ws = await client.connect(
-                self.url, origin=Origin("https://web.max.ru")
+                self.url, origin=Origin("https://web.max.ru"), max_size=1024 * 1024 * 10
             )  # TODO: origin should be configurable
 
     async def close(self) -> None:
@@ -41,7 +44,7 @@ class WebSocketTransport(Transport):
         if self.ws is None or not self.connected:
             raise ConnectionError("Not connected to the server")
 
-        return await self.ws.recv(decode=True)
+        return await self.ws.recv(decode=False)
 
     @property
     def connected(self) -> bool:

pymax/types/domain/__init__.py

@@ -3,7 +3,8 @@ from .bots import InitData
 from .chat import Chat
 from .error import MaxApiError
 from .folder import Folder, FolderList, FolderUpdate
-from .login import LoginResponse
+from .handshake import HandshakeResponse
+from .login import Login2Flags, Login2Response, LoginResponse
 from .member import Member
 from .message import Message, ReactionCounter, ReactionInfo, ReadState
 from .name import Name

pymax/types/domain/attachments/__init__.py

@@ -2,10 +2,11 @@ from .audio import AudioAttachment
 from .call import CallAttachment
 from .contact import ContactAttachment
 from .control import ControlAttachment
-from .enums import AttachmentType
+from .enums import AttachmentType, PollFlags
 from .file import FileAttachment, FileRequest
 from .keyboards import InlineKeyboardAttachment
 from .photo import PhotoAttachment
+from .poll import Poll, PollAnswer, PollAttachment, PollState
 from .share import ShareAttachment
 from .sticker import StickerAttachment
 from .unknown import UnknownAttachment

pymax/types/domain/attachments/enums.py

@@ -1,4 +1,4 @@
-from enum import Enum
+from enum import Enum, IntFlag
 
 
 class AttachmentType(str, Enum):
@@ -14,6 +14,7 @@ class AttachmentType(str, Enum):
     CALL = "CALL"
     SHARE = "SHARE"
     INLINE_KEYBOARD = "INLINE_KEYBOARD"
+    POLL = "POLL"
     UNKNOWN = "UNKNOWN"
 
 
@@ -26,3 +27,18 @@ class TranscriptionStatus(str, Enum):
     PROCESSING = "PROCESSING"
     SUCCESS = "SUCCESS"
     UNKNOWN = "UNKNOWN"
+
+
+class PollFlags(IntFlag):
+    """Настройки опроса, представленные битовой маской.
+
+    Несколько настроек объединяются оператором ``|``. Числовые значения,
+    полученные от Max, автоматически преобразуются в ``PollFlags``.
+    """
+
+    FLAG_SETTINGS_ANONYMOUS = 1
+    FLAG_SETTINGS_MULTISELECT = 2
+    FLAG_SETTINGS_REVOTE = 4
+    FLAG_SETTINGS_CLOSED = 8
+    FLAG_SETTINGS_QUIZ = 16
+    FLAG_SETTINGS_CAN_FORWARD = 32

pymax/types/domain/attachments/poll.py

@@ -0,0 +1,111 @@
+from typing import Literal
+
+from pydantic import Field
+
+from pymax.types.domain.base import CamelModel
+
+from .enums import AttachmentType, PollFlags
+
+
+class PollAnswer(CamelModel):
+    """Вариант ответа в опросе.
+
+    :ivar text: Текст варианта ответа.
+    :vartype text: str
+    :ivar answer_id: ID варианта, назначенный Max.
+    :vartype answer_id: int | None
+    """
+
+    text: str
+    answer_id: int | None = None
+
+
+class PollVote(CamelModel):
+    """Голос пользователя за один вариант ответа.
+
+    :ivar timestamp: Время голосования в формате Unix time.
+    :vartype timestamp: int
+    :ivar user_id: ID проголосовавшего пользователя.
+    :vartype user_id: int
+    """
+
+    timestamp: int
+    user_id: int
+
+
+class PollResult(CamelModel):
+    """Результат голосования по одному варианту ответа.
+
+    :ivar answer_id: ID варианта ответа.
+    :vartype answer_id: int
+    :ivar vote_count: Количество голосов.
+    :vartype vote_count: int
+    :ivar votes: Голоса пользователей, доступные текущему аккаунту.
+    :vartype votes: list[PollVote]
+    :ivar rate: Доля голосов в формате, возвращаемом Max.
+    :vartype rate: int
+    :ivar options: Дополнительные параметры результата от Max.
+    :vartype options: int
+    """
+
+    answer_id: int
+    vote_count: int
+    votes: list[PollVote]
+    rate: int
+    options: int
+
+
+class PollState(CamelModel):
+    """Текущее состояние голосования.
+
+    :ivar total: Общее количество голосов.
+    :vartype total: int
+    :ivar result: Результаты по вариантам ответа.
+    :vartype result: list[PollResult] | None
+    :ivar voter_preview_ids: ID пользователей для предпросмотра списка
+        проголосовавших.
+    :vartype voter_preview_ids: list[int]
+    """
+
+    total: int = 0
+    result: list[PollResult] | None = None
+    voter_preview_ids: list[int]
+
+
+class Poll(CamelModel):
+    """Опрос для отправки в сообщении.
+
+    :ivar title: Вопрос или заголовок опроса.
+    :vartype title: str
+    :ivar answers: Варианты ответа.
+    :vartype answers: list[PollAnswer]
+    :ivar settings: Настройки опроса. Несколько ``PollFlags`` объединяются
+        оператором ``|``.
+    :vartype settings: PollFlags
+    :ivar type: Тип вложения.
+    :vartype type: Literal[AttachmentType.POLL]
+    """
+
+    title: str
+    answers: list[PollAnswer]
+    settings: PollFlags
+    type: Literal[AttachmentType.POLL] = Field(alias="_type", default=AttachmentType.POLL)
+
+
+class PollAttachment(Poll):
+    """Опрос, полученный как вложение сообщения.
+
+    Помимо параметров опроса содержит назначенный Max ID, версию и текущее
+    состояние голосования.
+
+    :ivar poll_id: ID опроса.
+    :vartype poll_id: int
+    :ivar version: Версия данных опроса.
+    :vartype version: int
+    :ivar state: Текущее состояние голосования.
+    :vartype state: PollState
+    """
+
+    poll_id: int
+    version: int
+    state: PollState

pymax/types/domain/attachments/sticker.py

@@ -42,7 +42,7 @@ class StickerAttachment(CamelModel):
     sticker_id: int
     tags: list[str] | None = None
     width: int
-    set_id: int
+    set_id: int | None = None
     time: int
     sticker_type: str
     audio: bool

pymax/types/domain/attachments/video.py

@@ -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/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

@@ -5,7 +5,7 @@ from typing import TYPE_CHECKING, Annotated, Any, 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,
@@ -38,11 +40,12 @@ 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
 
 
@@ -182,22 +185,21 @@ class Message(CamelModel):
 
     async def reply(
         self,
-        text: str,
+        text: str | None = None,
         attachments: SendAttachments = None,
         *,
         notify: bool = True,
-    ) -> Message | 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
+        :returns: Отправленное сообщение.
+        :rtype: Message
         :raises RuntimeError: Если сообщение не привязано к сервису или не
             содержит ``chat_id``.
         """
@@ -213,25 +215,24 @@ class Message(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: Если сообщение не привязано к сервису или не
             содержит ``chat_id``.
         """
@@ -250,15 +251,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 +292,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: Отредактированное сообщение.

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