maxapi-python 1.1.3__py3-none-any.whl → 1.1.5__py3-none-any.whl

{maxapi_python-1.1.3.dist-info → maxapi_python-1.1.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: maxapi-python
-Version: 1.1.3
+Version: 1.1.5
 Summary: Python wrapper для API мессенджера Max
 Project-URL: Homepage, https://github.com/noxzion/PyMax
 Project-URL: Repository, https://github.com/noxzion/PyMax
@@ -17,7 +17,7 @@ Requires-Dist: aiohttp>=3.12.15
 Requires-Dist: lz4>=4.4.4
 Requires-Dist: msgpack>=1.1.1
 Requires-Dist: sqlmodel>=0.0.24
-Requires-Dist: websockets>=11.0
+Requires-Dist: websockets>=15.0
 Description-Content-Type: text/markdown
 
 <p align="center">

{maxapi_python-1.1.3.dist-info → maxapi_python-1.1.5.dist-info}/RECORD

@@ -3,26 +3,26 @@ pymax/core.py,sha256=6fqnP-eAgbUTM6F0Kj09TPJ3IOgvkHz5EsjJTI18r5A,7813
 pymax/crud.py,sha256=Mk-c87GItS91BlJu6INDbw1-ovXyoB2D9rXHK8voxpU,3207
 pymax/exceptions.py,sha256=msS11MD7qZPm0qZ6O8fobTm-GTldm2IA3uQLTX6eDxc,919
 pymax/files.py,sha256=Tpv-43gS7I4Pwlaimb8mZ2B-ZkF3aMsrLYT20NPaqhE,2656
-pymax/filters.py,sha256=EejNuJMmSBhw3bUqDoqXEnCnLjGy_sw5aH3Vynpxc0A,1306
+pymax/filters.py,sha256=fxC3Bl5AqrB8BSdN-CtiLWucbHCRA2IODpsOy2NagGU,1471
 pymax/interfaces.py,sha256=_D6iMQI74Gdtl6-HMoE1acRFRHeITgOZPVFpt5pvoow,2394
 pymax/models.py,sha256=PsPGbOkERxesZZltjNrmqhOfRcO44Is2ThbEToREcB8,201
 pymax/navigation.py,sha256=16c1_FZrw24uFlP6W5-F8OrEQE73bkQA3HSFqTdBtgo,5725
-pymax/payloads.py,sha256=yeBRxiMq6ixUQjMBBFcBDtBpYzqqfaEII3Z1kJq8pe8,3907
+pymax/payloads.py,sha256=t-6xM3s8laQqBWyVdcQNV080sjfRsOtbQ-o6CKpAh20,4087
 pymax/static.py,sha256=wwSV1ue5s5buqWz6TvCzjzN2ZWI-wITposTRvcS151g,4738
-pymax/types.py,sha256=gQc-XYhx2zZ05FE9eIVkPqPh2rERxil96i2mTXdhEPE,17831
+pymax/types.py,sha256=2huOPuGxwka_6s-Fw-7KbeAAoiFJrB7e07ArBKIRA6w,19020
 pymax/utils.py,sha256=F2TdoWfSwDLeh2uIcMIE_GTdXd7hU7gWti2i5P727bA,1364
 pymax/mixins/__init__.py,sha256=-PSMwTVioS-VTy-EGfV-epaKFLy58R4N2b-rX6wJf-M,649
 pymax/mixins/auth.py,sha256=vTNSZ6AunvDIMPQAvgYozpIZaCWMYiMDiabCBI7Sm6c,3079
 pymax/mixins/channel.py,sha256=Stnf63GPtlQnsMPVEC9P0oardEOz50I4DCXN5H5s1SM,823
-pymax/mixins/group.py,sha256=QJCd5MLYCVRrClcuAuRkLV3oylJRAOaGw0xUqFm2uXk,7820
+pymax/mixins/group.py,sha256=NFdq6VExTKR80tE9XzVBjuG_eRMKEdYSm2HZop5yPYE,7809
 pymax/mixins/handler.py,sha256=I1iNPaEgpvFnphaxV6liLwVaBCJ8sN6-h7908-_tPFk,2104
-pymax/mixins/message.py,sha256=0y7fO61zg9XG46SKxOPCOLMJW1LJBWfTi-PR3HjJfJo,10249
+pymax/mixins/message.py,sha256=GtwhoZaX2JPadhE9wq7M0YHBV1V2jbBIFwtQVIWP8sY,12896
 pymax/mixins/self.py,sha256=V0gbkY3jfX9fnd7v06n4_s7P3HTPcS1KNPqUzA0vNi8,1169
-pymax/mixins/socket.py,sha256=sgGxX7FmLyO3M82Nu2L-n00dCn0W2fV5TnrCipiKhaE,15738
+pymax/mixins/socket.py,sha256=UNMU1Fh9J3xDV76J2Mx8bVHXo0gcS81qocOw8MM0XXs,15643
 pymax/mixins/telemetry.py,sha256=0sQl6kvFVxobLthNAPNS9LzMrwwzZFA1xmOnvfiHWos,3522
 pymax/mixins/user.py,sha256=U-epgvLruTDHBCrLDE0N0iWeOypGE1_SU8cKD3TE90U,3045
-pymax/mixins/websocket.py,sha256=ndgcezJ1xqYCpaxc17rXxiXDQrTwTmtrgvx_d1H2Pcc,10094
-maxapi_python-1.1.3.dist-info/METADATA,sha256=P3lRNm2tS__wY5cMe3awWXIvKx8YC2SvmxYBSFGu-fA,5930
-maxapi_python-1.1.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-maxapi_python-1.1.3.dist-info/licenses/LICENSE,sha256=oe-AGp86WMKawV4KmqF28Q0m-kGAhPfAOPrEUm4MnVw,1064
-maxapi_python-1.1.3.dist-info/RECORD,,
+pymax/mixins/websocket.py,sha256=L958bXTYh6U89jnmd0VWIUVtmJweUVFOMqjEuH8jWGo,10079
+maxapi_python-1.1.5.dist-info/METADATA,sha256=k2lznWoJb4fgMJg_abFlOXE064a5_eZjbMDknokgZh8,5930
+maxapi_python-1.1.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+maxapi_python-1.1.5.dist-info/licenses/LICENSE,sha256=oe-AGp86WMKawV4KmqF28Q0m-kGAhPfAOPrEUm4MnVw,1064
+maxapi_python-1.1.5.dist-info/RECORD,,

pymax/filters.py

@@ -5,6 +5,7 @@ from .types import Message
 class Filter:
     def __init__(
         self,
+        chat_id: int | None = None,
         user_id: int | None = None,
         text: list[str] | None = None,
         status: MessageStatus | str | None = None,
@@ -12,6 +13,7 @@ class Filter:
         text_contains: str | None = None,
         reaction_info: bool | None = None,
     ) -> None:
+        self.chat_id = chat_id
         self.user_id = user_id
         self.text = text
         self.status = status
@@ -20,6 +22,8 @@ class Filter:
         self.text_contains = text_contains
 
     def match(self, message: Message) -> bool:
+        if self.chat_id is not None and message.chat_id != self.chat_id:
+            return False
         if self.user_id is not None and message.sender != self.user_id:
             return False
         if self.text is not None and any(

pymax/mixins/group.py

@@ -51,7 +51,7 @@ class GroupMixin(ClientProtocol):
                 return None
 
             chat = Chat.from_dict(data["payload"]["chat"])
-            message = Message.from_dict(data["payload"]["message"])
+            message = Message.from_dict(data["payload"])
 
             if chat:
                 cached_chat = await self._get_chat(chat.id)

pymax/mixins/message.py

@@ -15,9 +15,11 @@ from pymax.payloads import (
     SendMessagePayload,
     SendMessagePayloadMessage,
     UploadPhotoPayload,
+    GetVideoPayload,
+    GetFilePayload,
 )
 from pymax.static import AttachType, Opcode
-from pymax.types import Attach, Message
+from pymax.types import Attach, Message, FileRequest, VideoRequest
 
 
 class MessageMixin(ClientProtocol):
@@ -143,11 +145,7 @@ class MessageMixin(ClientProtocol):
                 self.logger.error("Send message error: %s", error)
                 print(data)
                 return None
-            msg = (
-                Message.from_dict(data["payload"]["message"])
-                if data.get("payload")
-                else None
-            )
+            msg = Message.from_dict(data["payload"]) if data.get("payload") else None
             self.logger.debug("send_message result: %r", msg)
             return msg
         except Exception:
@@ -174,11 +172,7 @@ class MessageMixin(ClientProtocol):
             data = await self._send_and_wait(opcode=Opcode.MSG_EDIT, payload=payload)
             if error := data.get("payload", {}).get("error"):
                 self.logger.error("Edit message error: %s", error)
-            msg = (
-                Message.from_dict(data["payload"]["message"])
-                if data.get("payload")
-                else None
-            )
+            msg = Message.from_dict(data["payload"]) if data.get("payload") else None
             self.logger.debug("edit_message result: %r", msg)
             return msg
         except Exception:
@@ -291,3 +285,83 @@ class MessageMixin(ClientProtocol):
         except Exception:
             self.logger.exception("Fetch history failed")
             return None
+
+
+    async def get_video_by_id(
+        self,
+        chat_id: int,
+        message_id: int,
+        video_id: str,
+    ) -> VideoRequest | None:
+        """
+        Получает видео
+
+        Args:
+            chat_id (int): ID чата
+            message_id (int): ID сообщения
+            video_id (int): ID видео
+
+        Returns:
+            external (str): Странная ссылка из апи
+            cache (bool): True, если видео кэшировано
+            url (str): Ссылка на видео
+        """
+        try:
+            self.logger.info(
+                "Getting video_id=%s message_id=%s", video_id, message_id
+            )
+
+            payload = GetVideoPayload(
+                chat_id=chat_id,
+                message_id=message_id,
+                video_id=video_id
+            ).model_dump(by_alias=True)
+
+            data = await self._send_and_wait(opcode=Opcode.VIDEO_PLAY, payload=payload)
+
+            if error := data.get("payload", {}).get("error"):
+                self.logger.error("Get video error: %s", error)
+            
+            video = VideoRequest.from_dict(data["payload"]) if data.get("payload") else None 
+            self.logger.debug(" result: %r", video)
+            return video
+        except Exception:
+            self.logger.exception("Get video error")
+            return None
+        
+    async def get_file_by_id(
+        self,
+        chat_id: int,
+        message_id: int,
+        file_id: str,
+    ) -> FileRequest | None:
+        """
+        Получает файл
+
+        Args:
+            chat_id (int): ID чата
+            message_id (int): ID сообщения
+            file_id (int): ID видео
+
+        Returns:
+            unsafe (bool): Проверка файла на безопасность максом
+            url (str): Ссылка на скачивание файла
+        """
+        try:
+            self.logger.info(
+                "Getting file_id=%s message_id=%s", file_id, message_id
+            )
+            payload = GetFilePayload(
+                chat_id=chat_id,
+                message_id=message_id,
+                file_id=file_id
+            ).model_dump(by_alias=True)
+            data = await self._send_and_wait(opcode=Opcode.FILE_DOWNLOAD, payload=payload)
+            if error := data.get("payload", {}).get("error"):
+                self.logger.error("Get file error: %s", error)
+            file = FileRequest.from_dict(data["payload"]) if data.get("payload") else None 
+            self.logger.debug(" result: %r", file)
+            return file
+        except Exception:
+            self.logger.exception("Get video error")
+            return None

pymax/mixins/socket.py

@@ -214,9 +214,7 @@ Socket connections may be unstable, SSL issues are possible.
                                 for handler, filter in self._on_message_handlers:
                                     payload = data_item.get("payload", {})
                                     msg_dict = (
-                                        payload.get("message")
-                                        if isinstance(payload, dict)
-                                        else None
+                                        payload if isinstance(payload, dict) else None
                                     )
                                     msg = (
                                         Message.from_dict(msg_dict)

pymax/mixins/websocket.py

@@ -118,7 +118,7 @@ class WebSocketMixin(ClientProtocol):
                         try:
                             for handler, filter in self._on_message_handlers:
                                 payload = data.get("payload", {})
-                                msg = Message.from_dict(payload.get("message"))
+                                msg = Message.from_dict(payload)
                                 if msg:
                                     if msg.status:
                                         continue  # TODO: заглушка! сделать отдельный хендлер

pymax/payloads.py

@@ -193,3 +193,13 @@ class NavigationEventPayload(CamelModel):
 
 class NavigationPayload(CamelModel):
     events: list[NavigationEventPayload]
+
+class GetVideoPayload(CamelModel):
+    chat_id: int
+    message_id: str
+    video_id: int
+
+class GetFilePayload(CamelModel):
+    chat_id: int
+    message_id: str
+    file_id: int

pymax/types.py

@@ -167,6 +167,46 @@ class FileAttach:
         return f"FileAttach: {self.file_id}"
 
 
+class FileRequest:
+    def __init__(
+        self,
+        unsafe: bool,
+        url: str,
+    ) -> None:
+        self.unsafe = unsafe
+        self.url = url
+
+    @classmethod
+    def from_dict(cls, data: dict[str, Any]) -> "FileRequest":
+        return cls(
+            unsafe=data["unsafe"],
+            url=data["url"],
+    )
+
+
+class VideoRequest:
+    def __init__(
+        self,
+        external: str,
+        cache: bool,
+        url: str,
+    ) -> None:
+        self.external = external
+        self.cache = cache
+        self.url = url
+
+    @classmethod
+    def from_dict(cls, data: dict[str, Any]) -> "VideoRequest":
+
+        listdata = list(data.values()) # Костыль ✅
+
+        return cls(
+            external=data["EXTERNAL"],
+            cache=data["cache"],
+            url=listdata[2],
+    )
+        
+
 class Me:
     def __init__(
         self,
@@ -230,6 +270,7 @@ class Element:
 class Message:
     def __init__(
         self,
+        chat_id: int | None,
         sender: int | None,
         elements: list[Element] | None,
         reaction_info: dict[str, Any] | None,
@@ -241,6 +282,7 @@ class Message:
         type: MessageType | str,
         attaches: list[PhotoAttach | VideoAttach | FileAttach],
     ) -> None:
+        self.chat_id = chat_id
         self.sender = sender
         self.elements = elements
         self.options = options
@@ -254,8 +296,9 @@ class Message:
 
     @classmethod
     def from_dict(cls, data: dict[Any, Any]) -> "Message":
+        message = data["message"] if data.get("message") else data
         attaches = []
-        for a in data.get("attaches", []):
+        for a in message.get("attaches", []):
             if a["_type"] == AttachType.PHOTO:
                 attaches.append(PhotoAttach.from_dict(a))
             elif a["_type"] == AttachType.VIDEO:
@@ -263,16 +306,17 @@ class Message:
             elif a["_type"] == AttachType.FILE:
                 attaches.append(FileAttach.from_dict(a))
         return cls(
-            sender=data.get("sender"),
-            elements=[Element.from_dict(e) for e in data.get("elements", [])],
-            options=data.get("options"),
-            id=data["id"],
-            time=data["time"],
-            text=data["text"],
-            type=data["type"],
+            chat_id=data.get("chatId"),
+            sender=message.get("sender"),
+            elements=[Element.from_dict(e) for e in message.get("elements", [])],
+            options=message.get("options"),
+            id=message["id"],
+            time=message["time"],
+            text=message["text"],
+            type=message["type"],
             attaches=attaches,
-            status=data.get("status"),
-            reaction_info=data.get("reactionInfo"),
+            status=message.get("status"),
+            reaction_info=message.get("reactionInfo"),
         )
 
     @override
@@ -280,6 +324,7 @@ class Message:
         return (
             f"Message(id={self.id!r}, sender={self.sender!r}, text={self.text!r}, "
             f"type={self.type!r}, status={self.status!r}, elements={self.elements!r})"
+            f"attaches={self.attaches!r}, chat_id={self.chat_id!r}, time={self.time!r}, options={self.options!r}, reactionInfo={self.reactionInfo!r})"
         )
 
     @override
@@ -533,7 +578,7 @@ class User:
         return f"User {self.id}: {', '.join(str(n) for n in self.names)}"
 
 
-class Attach:  # УБРАТЬ ГАДА!!!
+class Attach:  # УБРАТЬ ГАДА!!! или нет...
     def __init__(
         self,
         _type: AttachType,