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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: maxapi-python
-Version: 1.1.4
+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.4.dist-info → maxapi_python-1.1.5.dist-info}/RECORD

@@ -7,22 +7,22 @@ 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=pC87lOfRmM33ed2LY_S2c2KJTMvWJfrkTqerTp-dluk,18179
+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=NFdq6VExTKR80tE9XzVBjuG_eRMKEdYSm2HZop5yPYE,7809
 pymax/mixins/handler.py,sha256=I1iNPaEgpvFnphaxV6liLwVaBCJ8sN6-h7908-_tPFk,2104
-pymax/mixins/message.py,sha256=fPQ69N-9O03BEsbhBQzHn6HTcHH3tWjR6VJtn_F2kzc,10099
+pymax/mixins/message.py,sha256=GtwhoZaX2JPadhE9wq7M0YHBV1V2jbBIFwtQVIWP8sY,12896
 pymax/mixins/self.py,sha256=V0gbkY3jfX9fnd7v06n4_s7P3HTPcS1KNPqUzA0vNi8,1169
 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=L958bXTYh6U89jnmd0VWIUVtmJweUVFOMqjEuH8jWGo,10079
-maxapi_python-1.1.4.dist-info/METADATA,sha256=DziHH8WPX4Ed1xzJ8at5MEGp0kMcKORFGNlKW1bb_-0,5930
-maxapi_python-1.1.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-maxapi_python-1.1.4.dist-info/licenses/LICENSE,sha256=oe-AGp86WMKawV4KmqF28Q0m-kGAhPfAOPrEUm4MnVw,1064
-maxapi_python-1.1.4.dist-info/RECORD,,
+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/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):
@@ -283,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/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,
@@ -538,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,