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

pymax/types/domain/message.py

@@ -15,6 +15,7 @@ from pymax.types.domain import (
     PhotoAttachment,
     ShareAttachment,
     StickerAttachment,
+    UnknownAttachment,
     VideoAttachment,
 )
 
@@ -26,7 +27,7 @@ if TYPE_CHECKING:
     from pymax.api.messages.service import MessageService
 
 
-Attachment: TypeAlias = Annotated[
+KnownAttachment: TypeAlias = Annotated[
     PhotoAttachment
     | VideoAttachment
     | FileAttachment
@@ -39,6 +40,7 @@ Attachment: TypeAlias = Annotated[
     | CallAttachment,
     Field(discriminator="type"),
 ]
+Attachment: TypeAlias = KnownAttachment | UnknownAttachment
 SendAttachment: TypeAlias = Photo | File | Video
 SendAttachments: TypeAlias = list[SendAttachment] | None
 
@@ -90,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`` и при работе с
@@ -259,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