maxapi-python 1.1.10__tar.gz → 1.1.11__tar.gz

{maxapi_python-1.1.10 → maxapi_python-1.1.11}/.github/FUNDING.yml

@@ -12,4 +12,4 @@ lfx_crowdfunding: # Replace with a single LFX Crowdfunding project-name e.g., cl
 polar: # Replace with a single Polar username
 buy_me_a_coffee: # Replace with a single Buy Me a Coffee username
 thanks_dev: # Replace with a single thanks.dev username
-custom: ['https://www.donationalerts.com/r/pymax']
+custom: ['']

{maxapi_python-1.1.10 → maxapi_python-1.1.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: maxapi-python
-Version: 1.1.10
+Version: 1.1.11
 Summary: Python wrapper для API мессенджера Max
 Project-URL: Homepage, https://github.com/noxzion/PyMax
 Project-URL: Repository, https://github.com/noxzion/PyMax

{maxapi_python-1.1.10 → maxapi_python-1.1.11}/examples/example.py

@@ -21,12 +21,27 @@ async def handle_message(message: Message) -> None:
 @client.on_start
 async def handle_start() -> None:
     print("Client started successfully!")
-    sessions = await client.get_sessions()
-    for session in sessions:
-        print(session.client)
+    react_info = await client.add_reaction(
+        chat_id=0, message_id="115368067020359151", reaction="👍"
+    )
+    if react_info:
+        print("Reaction added!")
+        print(react_info.total_count)
+    react_info = await client.get_reactions(
+        chat_id=0, message_ids=["115368067020359151"]
+    )
+    if react_info:
+        print("Reactions fetched!")
+        for msg_id, info in react_info.items():
+            print(f"Message ID: {msg_id}, Total Reactions: {info.total_count}")
+    react_info = await client.remove_reaction(
+        chat_id=0, message_id="115368067020359151"
+    )
+    if react_info:
+        print("Reaction removed!")
+        print(react_info.total_count)
     # print(client.dialogs)
-    chat = await client.join_group("join/sdfdfsfdf")
-    print(chat.title)
+
     # if history:
     #     for message in history:
     #         if message.link:

{maxapi_python-1.1.10 → maxapi_python-1.1.11}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "maxapi-python"
-version = "1.1.10"
+version = "1.1.11"
 description = "Python wrapper для API мессенджера Max"
 readme = "README.md"
 requires-python = ">=3.10"

{maxapi_python-1.1.10 → maxapi_python-1.1.11}/src/pymax/mixins/handler.py

@@ -1,4 +1,5 @@
-from typing import Any, Awaitable, Callable
+from collections.abc import Awaitable, Callable
+from typing import Any
 
 from pymax.interfaces import ClientProtocol, Filter
 from pymax.types import Message

{maxapi_python-1.1.10 → maxapi_python-1.1.11}/src/pymax/mixins/message.py

@@ -6,20 +6,30 @@ from aiohttp import ClientSession
 from pymax.files import File, Photo, Video
 from pymax.interfaces import ClientProtocol
 from pymax.payloads import (
+    AddReactionPayload,
     AttachPhotoPayload,
     DeleteMessagePayload,
     EditMessagePayload,
     FetchHistoryPayload,
     GetFilePayload,
+    GetReactionsPayload,
     GetVideoPayload,
     PinMessagePayload,
+    ReactionInfoPayload,
+    RemoveReactionPayload,
     ReplyLink,
     SendMessagePayload,
     SendMessagePayloadMessage,
     UploadPhotoPayload,
 )
 from pymax.static import AttachType, Opcode
-from pymax.types import Attach, FileRequest, Message, VideoRequest
+from pymax.types import (
+    Attach,
+    FileRequest,
+    Message,
+    ReactionInfo,
+    VideoRequest,
+)
 
 
 class MessageMixin(ClientProtocol):
@@ -375,3 +385,140 @@ class MessageMixin(ClientProtocol):
         except Exception:
             self.logger.exception("Get video error")
             return None
+
+    async def add_reaction(
+        self,
+        chat_id: int,
+        message_id: str,
+        reaction: str,
+    ) -> ReactionInfo | None:
+        """
+        Добавляет реакцию к сообщению.
+
+        Args:
+            chat_id (int): ID чата
+            message_id (int): ID сообщения
+            reaction (str): Реакция (эмодзи)
+
+        Returns:
+            ReactionInfo | None: Информация о реакции или None при ошибке.
+        """
+        try:
+            self.logger.info(
+                "Adding reaction to message chat_id=%s message_id=%s reaction=%s",
+                chat_id,
+                message_id,
+                reaction,
+            )
+
+            payload = AddReactionPayload(
+                chat_id=chat_id,
+                message_id=message_id,
+                reaction=ReactionInfoPayload(id=reaction),
+            ).model_dump(by_alias=True)
+
+            data = await self._send_and_wait(
+                opcode=Opcode.MSG_REACTION, payload=payload
+            )
+            if error := data.get("payload", {}).get("error"):
+                self.logger.error("Add reaction error: %s", error)
+                return None
+
+            self.logger.debug("add_reaction success")
+            return (
+                ReactionInfo.from_dict(data["payload"]["reactionInfo"])
+                if data.get("payload")
+                else None
+            )
+        except Exception:
+            self.logger.exception("Add reaction failed")
+            return None
+
+    async def get_reactions(
+        self, chat_id: int, message_ids: list[str]
+    ) -> dict[str, ReactionInfo] | None:
+        """
+        Получает реакции на сообщения.
+
+        Args:
+            chat_id (int): ID чата
+            message_ids (list[str]): Список ID сообщений
+
+        Returns:
+            dict[str, ReactionInfo] | None: Словарь с ID сообщений и информацией о реакциях или None при ошибке.
+        """
+        try:
+            self.logger.info(
+                "Getting reactions for messages chat_id=%s message_ids=%s",
+                chat_id,
+                message_ids,
+            )
+
+            payload = GetReactionsPayload(
+                chat_id=chat_id, message_ids=message_ids
+            ).model_dump(by_alias=True)
+
+            data = await self._send_and_wait(
+                opcode=Opcode.MSG_GET_REACTIONS, payload=payload
+            )
+            if error := data.get("payload", {}).get("error"):
+                self.logger.error("Get reactions error: %s", error)
+                return None
+
+            reactions = {}
+            for msg_id, reaction_data in (
+                data.get("payload", {}).get("messagesReactions", {}).items()
+            ):
+                reactions[msg_id] = ReactionInfo.from_dict(reaction_data)
+
+            self.logger.debug("get_reactions success")
+
+            return reactions
+
+        except Exception:
+            self.logger.exception("Get reactions failed")
+            return None
+
+    async def remove_reaction(
+        self,
+        chat_id: int,
+        message_id: str,
+    ) -> ReactionInfo | None:
+        """
+        Удаляет реакцию с сообщения.
+
+        Args:
+            chat_id (int): ID чата
+            message_id (str): ID сообщения
+
+        Returns:
+            ReactionInfo | None: Информация о реакции или None при ошибке.
+        """
+        try:
+            self.logger.info(
+                "Removing reaction from message chat_id=%s message_id=%s",
+                chat_id,
+                message_id,
+            )
+
+            payload = RemoveReactionPayload(
+                chat_id=chat_id,
+                message_id=message_id,
+            ).model_dump(by_alias=True)
+
+            data = await self._send_and_wait(
+                opcode=Opcode.MSG_CANCEL_REACTION, payload=payload
+            )
+            if error := data.get("payload", {}).get("error"):
+                self.logger.error("Remove reaction error: %s", error)
+                return None
+
+            self.logger.debug("remove_reaction success")
+            return (
+                ReactionInfo.from_dict(data["payload"]["reactionInfo"])
+                if data.get("payload")
+                else None
+            )
+        except Exception:
+            self.logger.exception("Remove reaction failed")
+            return None

{maxapi_python-1.1.10 → maxapi_python-1.1.11}/src/pymax/payloads.py

@@ -213,3 +213,24 @@ class SearchByPhonePayload(CamelModel):
 
 class JoinGroupPayload(CamelModel):
     link: str
+
+
+class ReactionInfoPayload(CamelModel):
+    reaction_type: str = "EMOJI"
+    id: str
+
+
+class AddReactionPayload(CamelModel):
+    chat_id: int
+    message_id: str
+    reaction: ReactionInfoPayload
+
+
+class GetReactionsPayload(CamelModel):
+    chat_id: int
+    message_ids: list[str]
+
+
+class RemoveReactionPayload(CamelModel):
+    chat_id: int
+    message_id: str

{maxapi_python-1.1.10 → maxapi_python-1.1.11}/src/pymax/types.py

@@ -317,13 +317,51 @@ class MessageLink:
         return f"MessageLink: {self.chat_id}/{self.message.id}"
 
 
+class ReactionCounter:
+    def __init__(self, count: int, reaction: str) -> None:
+        self.count = count
+        self.reaction = reaction
+
+    @classmethod
+    def from_dict(cls, data: dict[str, Any]) -> "ReactionCounter":
+        return cls(count=data["count"], reaction=data["reaction"])
+
+    @override
+    def __repr__(self) -> str:
+        return f"ReactionCounter(count={self.count!r}, reaction={self.reaction!r})"
+
+    @override
+    def __str__(self) -> str:
+        return f"{self.reaction}: {self.count}"
+
+
+class ReactionInfo:
+    def __init__(
+        self,
+        total_count: int,
+        counters: list[ReactionCounter],
+        your_reaction: str | None = None,
+    ) -> None:
+        self.total_count = total_count
+        self.counters = counters
+        self.your_reaction = your_reaction
+
+    @classmethod
+    def from_dict(cls, data: dict[str, Any]) -> "ReactionInfo":
+        return cls(
+            total_count=data.get("totalCount", 0),
+            counters=[ReactionCounter.from_dict(c) for c in data.get("counters", [])],
+            your_reaction=data.get("yourReaction"),
+        )
+
+
 class Message:
     def __init__(
         self,
         chat_id: int | None,
         sender: int | None,
         elements: list[Element] | None,
-        reaction_info: dict[str, Any] | None,
+        reaction_info: ReactionInfo | None,
         options: int | None,
         id: int,
         time: int,
@@ -373,7 +411,9 @@ class Message:
             link=MessageLink.from_dict(message.get("link"))
             if message.get("link")
             else None,
-            reaction_info=message.get("reactionInfo"),
+            reaction_info=ReactionInfo.from_dict(message.get("reactionInfo"))
+            if message.get("reactionInfo")
+            else None,
         )
 
     @override