maxapi-python 2.1.3__py3-none-any.whl → 2.2.0__py3-none-any.whl

pymax/types/domain/attachments/unknown.py

@@ -30,8 +30,6 @@ class UnknownAttachment(CamelModel):
 
         attachment_type = value.get("_type", value.get("type"))
         if attachment_type in KNOWN_ATTACHMENT_TYPES:
-            raise ValueError(
-                "Known attachment type should be parsed by its own model"
-            )
+            raise ValueError("Known attachment type should be parsed by its own model")
 
         return value

pymax/types/domain/auth.py

@@ -1,7 +1,10 @@
 from pydantic import BaseModel, Field
 
+from pymax.api.auth.enums import AuthType
 from pymax.api.models import CamelModel
 
+from .profile import Profile
+
 
 class StartAuthResponse(CamelModel):
     """Ответ на начало авторизации.
@@ -70,7 +73,7 @@ class CheckCodeResponse(CamelModel):
     :vartype password_challenge: PasswordChallenge | None
     """
 
-    token_attrs: TokenAttrs = Field(default_factory=TokenAttrs)
+    token_attrs: TokenAttrs = Field(default_factory=lambda: TokenAttrs.model_validate({}))
     password_challenge: PasswordChallenge | None = None
 
     @property
@@ -103,7 +106,7 @@ class CheckPasswordResponse(CamelModel):
     :vartype error: str | None
     """
 
-    token_attrs: TokenAttrs = Field(default_factory=TokenAttrs)
+    token_attrs: TokenAttrs = Field(default_factory=lambda: TokenAttrs.model_validate({}))
     error: str | None = None
 
     @property
@@ -159,3 +162,22 @@ class CheckQrResponse(CamelModel):
     """
 
     status: QrStatus
+
+
+class ConfirmRegistrationResponse(CamelModel):
+    """Ответ Max после регистрации нового аккаунта.
+
+    :ivar user_token: Внутренний ID зарегистрированного пользователя.
+    :vartype user_token: int
+    :ivar profile: Профиль зарегистрированного аккаунта.
+    :vartype profile: Profile
+    :ivar token_type: Тип выданного токена.
+    :vartype token_type: AuthType
+    :ivar token: Токен входа для новой сессии.
+    :vartype token: str
+    """
+
+    user_token: int
+    profile: Profile
+    token_type: AuthType
+    token: str

pymax/types/domain/chat.py

@@ -19,7 +19,8 @@ class Chat(CamelModel):
 
     Объекты чатов, полученные через клиент, обычно уже привязаны к сервисам
     сообщений и чатов. После этого можно вызывать удобные методы объекта:
-    :meth:`answer`, :meth:`history`, :meth:`leave`, :meth:`invite`,
+    :meth:`answer`, :meth:`history`, :meth:`get_message`,
+    :meth:`get_messages`, :meth:`leave`, :meth:`invite`,
     :meth:`remove_users`, :meth:`pin_message`, :meth:`update_settings` и
     :meth:`rework_invite_link`.
 
@@ -147,6 +148,10 @@ class Chat(CamelModel):
         """
         self._message_actions = message_actions
         self._chat_actions = chat_actions
+        if self.last_message is not None:
+            self.last_message.bind(message_actions)
+        if self.pinned_message is not None:
+            self.pinned_message.bind(message_actions)
         return self
 
     async def answer(
@@ -242,6 +247,38 @@ class Chat(CamelModel):
             interactive=interactive,
         )
 
+    async def get_message(self, message_id: int) -> Message | None:
+        """Возвращает сообщение этого чата по ID.
+
+        :param message_id: ID сообщения.
+        :type message_id: int
+        :returns: Сообщение или ``None``, если сервер его не вернул.
+        :rtype: Message | None
+        :raises RuntimeError: Если чат не привязан к клиенту.
+        """
+        actions, _ = self._bound()
+
+        return await actions.get_message(
+            chat_id=self.id,
+            message_id=message_id,
+        )
+
+    async def get_messages(self, message_ids: list[int]) -> list[Message]:
+        """Возвращает сообщения этого чата по ID.
+
+        :param message_ids: ID сообщений.
+        :type message_ids: list[int]
+        :returns: Список найденных сообщений.
+        :rtype: list[Message]
+        :raises RuntimeError: Если чат не привязан к клиенту.
+        """
+        actions, _ = self._bound()
+
+        return await actions.get_messages(
+            chat_id=self.id,
+            message_ids=message_ids,
+        )
+
     async def leave(self) -> None:
         """Выходит из группы или канала.
 

pymax/types/domain/message.py

@@ -92,7 +92,7 @@ class Message(CamelModel):
 
     Сообщения, полученные через клиент, обычно уже привязаны к сервису
     сообщений. После этого можно вызывать удобные методы объекта:
-    :meth:`reply`, :meth:`answer`, :meth:`pin`, :meth:`delete`,
+    :meth:`reply`, :meth:`answer`, :meth:`edit`, :meth:`pin`, :meth:`delete`,
     :meth:`read`, :meth:`react`, :meth:`unreact` и :meth:`get_reactions`.
 
     Используйте ``Message`` в обработчиках ``on_message`` и при работе с
@@ -261,6 +261,36 @@ class Message(CamelModel):
             notify_pin=notify_pin,
         )
 
+    async def edit(
+        self,
+        text: str,
+        attachment: SendAttachment | None = None,
+        attachments: SendAttachments = None,
+    ) -> Message:
+        """Редактирует текст и вложения этого сообщения.
+
+        :param text: Новый текст сообщения с поддержкой markdown.
+        :type text: str
+        :param attachment: Одно новое вложение.
+        :type attachment: SendAttachment | None
+        :param attachments: Список новых вложений. Имеет приоритет над
+            ``attachment``.
+        :type attachments: SendAttachments
+        :returns: Отредактированное сообщение.
+        :rtype: Message
+        :raises RuntimeError: Если сообщение не привязано к сервису или не
+            содержит ``chat_id``.
+        """
+        actions, chat_id = self._bound()
+
+        return await actions.edit_message(
+            chat_id=chat_id,
+            message_id=self.id,
+            text=text,
+            attachment=attachment,
+            attachments=attachments,
+        )
+
     async def delete(self, for_me: bool = False) -> bool:
         """Удаляет это сообщение.
 

pymax/types/domain/presence.py

@@ -7,9 +7,9 @@ class Presence(CamelModel):
     :ivar seen: Время последней активности в формате Unix time, если оно
         передано сервером.
     :vartype seen: int | None
-    :ivar status: Код статуса присутствия Max.
-    :vartype status: int
+    :ivar status: Код статуса присутствия Max, если он передан сервером.
+    :vartype status: int | None
     """
 
     seen: int | None = None
-    status: int
+    status: int | None = None

pymax/types/domain/sync.py

@@ -68,29 +68,13 @@ class SyncOverrides(BaseModel):
         :rtype: SyncState
         """
         return SyncState(
-            chats_sync=(
-                self.chats_sync
-                if self.chats_sync is not None
-                else saved.chats_sync
-            ),
+            chats_sync=(self.chats_sync if self.chats_sync is not None else saved.chats_sync),
             contacts_sync=(
-                self.contacts_sync
-                if self.contacts_sync is not None
-                else saved.contacts_sync
-            ),
-            drafts_sync=(
-                self.drafts_sync
-                if self.drafts_sync is not None
-                else saved.drafts_sync
+                self.contacts_sync if self.contacts_sync is not None else saved.contacts_sync
             ),
+            drafts_sync=(self.drafts_sync if self.drafts_sync is not None else saved.drafts_sync),
             presence_sync=(
-                self.presence_sync
-                if self.presence_sync is not None
-                else saved.presence_sync
-            ),
-            config_hash=(
-                self.config_hash
-                if self.config_hash is not None
-                else saved.config_hash
+                self.presence_sync if self.presence_sync is not None else saved.presence_sync
             ),
+            config_hash=(self.config_hash if self.config_hash is not None else saved.config_hash),
         )

pymax/types/events/__init__.py

@@ -1,3 +1,7 @@
 from .file import FileUploadSignal
+from .mark import MessageReadEvent
 from .message import MessageDeleteEvent
+from .presence import PresenceEvent
+from .reaction import ReactionUpdateEvent
+from .typing import TypingEvent
 from .video import VideoUploadSignal

pymax/types/events/mark.py

@@ -0,0 +1,23 @@
+from pymax.types.domain.base import CamelModel
+
+
+class MessageReadEvent(CamelModel):
+    """Событие изменения отметки прочтения чата.
+
+    Handler ``on_message_read`` получает объект, когда пользователь отмечает
+    сообщения прочитанными или возвращает чат в непрочитанное состояние.
+
+    :ivar set_as_unread: Чат был явно отмечен непрочитанным.
+    :vartype set_as_unread: bool
+    :ivar chat_id: ID чата.
+    :vartype chat_id: int
+    :ivar user_id: ID пользователя, изменившего отметку.
+    :vartype user_id: int
+    :ivar mark: Временная отметка прочтения Max.
+    :vartype mark: int
+    """
+
+    set_as_unread: bool
+    chat_id: int
+    user_id: int
+    mark: int

pymax/types/events/message.py

@@ -1,15 +1,19 @@
 from __future__ import annotations
 
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Any
 
-from pydantic import PrivateAttr
+from pydantic import PrivateAttr, model_validator
 
+from pymax.logging import get_logger
 from pymax.types.domain import Chat
 from pymax.types.domain.base import CamelModel
+from pymax.types.domain.message import Message
 
 if TYPE_CHECKING:
     from pymax.api.messages import MessageService
 
+logger = get_logger(__name__)
+
 
 class MessageDeleteEvent(CamelModel):
     """Событие удаления сообщений.
@@ -17,20 +21,68 @@ class MessageDeleteEvent(CamelModel):
     Handler ``on_message_delete`` получает этот объект, когда Max сообщает об
     удалении одного или нескольких сообщений в чате.
 
-    :ivar chat: Чат, в котором удалены сообщения.
-    :vartype chat: Chat
     :ivar message_ids: ID удаленных сообщений.
     :vartype message_ids: list[int]
+    :ivar chat_id: ID чата.
+    :vartype chat_id: int
+    :ivar chat: Чат, если Max прислал полный объект.
+    :vartype chat: Chat | None
+    :ivar message: Удаленное сообщение для WebSocket-события.
+    :vartype message: Message | None
     :ivar ttl: Признак удаления из-за TTL, если Max его прислал.
     :vartype ttl: bool
     """
 
-    chat: Chat
     message_ids: list[int]
+    chat_id: int
+    chat: Chat | None = None
+    message: Message | None = None
     ttl: bool = False
 
     _actions: MessageService | None = PrivateAttr(default=None)
 
+    @model_validator(mode="before")
+    @classmethod
+    def normalize_payload(cls, data: Any) -> Any:
+        # i really hate it cause of stupid web version thats send other type
+        # of payload (128, expect 142)
+        # TODO: impl it in the better way maybe
+
+        if not isinstance(data, dict):
+            return data
+
+        if "chat" in data:  # case opcode == 142
+            chat = data["chat"]
+            chat_id = chat.get("id") if isinstance(chat, dict) else getattr(chat, "id", None)
+            message_ids = data.get("messageIds", data.get("message_ids"))
+            if chat_id is None or message_ids is None:
+                return data
+
+            return {
+                "chat": chat,
+                "ttl": data.get("ttl", False),
+                "messageIds": message_ids,
+                "chatId": chat_id,
+            }
+        if "message" in data:  # case opcode == 128
+            message = data["message"]
+            message_id = (
+                message.get("id") if isinstance(message, dict) else getattr(message, "id", None)
+            )
+            chat_id = data.get("chatId", data.get("chat_id"))
+            if chat_id is None or message_id is None:
+                return data
+
+            return {
+                "chatId": chat_id,
+                "message": message,
+                "ttl": data.get("ttl", False),
+                "messageIds": [message_id],
+            }
+
+        logger.warning("Illegal state during MessageDeleteEvent validation. Starting fallback")
+        return data  # stupid fallback but who cares. Still better than KeyError
+
     def bind(self, actions: MessageService) -> MessageDeleteEvent:
         """Привязывает сервис сообщений к событию удаления."""
         self._actions = actions

pymax/types/events/presence.py

@@ -0,0 +1,15 @@
+from pymax.types.domain.base import CamelModel
+from pymax.types.domain.presence import Presence
+
+
+class PresenceEvent(CamelModel):
+    """Событие изменения присутствия пользователя.
+
+    :ivar presence: Новое состояние присутствия.
+    :vartype presence: Presence
+    :ivar user_id: ID пользователя.
+    :vartype user_id: int
+    """
+
+    presence: Presence
+    user_id: int

pymax/types/events/reaction.py

@@ -0,0 +1,21 @@
+from pymax.types.domain.base import CamelModel
+from pymax.types.domain.message import ReactionCounter
+
+
+class ReactionUpdateEvent(CamelModel):
+    """Событие обновления реакций сообщения.
+
+    :ivar message_id: ID сообщения.
+    :vartype message_id: str
+    :ivar chat_id: ID чата.
+    :vartype chat_id: int
+    :ivar counters: Счетчики реакций по типам.
+    :vartype counters: list[ReactionCounter]
+    :ivar total_count: Общее количество реакций.
+    :vartype total_count: int
+    """
+
+    message_id: str
+    chat_id: int
+    counters: list[ReactionCounter]
+    total_count: int

pymax/types/events/typing.py

@@ -0,0 +1,14 @@
+from pymax.types.domain.base import CamelModel
+
+
+class TypingEvent(CamelModel):
+    """Событие набора текста пользователем.
+
+    :ivar chat_id: ID чата.
+    :vartype chat_id: int
+    :ivar user_id: ID пользователя, который набирает текст.
+    :vartype user_id: int
+    """
+
+    chat_id: int
+    user_id: int