maxapi-python 0.1.3__py3-none-any.whl → 1.1.1__py3-none-any.whl

pymax/mixins/__init__.py

@@ -1,18 +1,35 @@
-from .auth import AuthMixin
-from .channel import ChannelMixin
-from .handler import HandlerMixin
-from .message import MessageMixin
-from .self import SelfMixin
-from .user import UserMixin
-from .websocket import WebSocketMixin
-
-
-class ApiMixin(
-    AuthMixin,
-    HandlerMixin,
-    UserMixin,
-    ChannelMixin,
-    SelfMixin,
-    MessageMixin,
-):
-    pass
+from .auth import AuthMixin
+from .channel import ChannelMixin
+from .handler import HandlerMixin
+from .message import MessageMixin
+from .self import SelfMixin
+from .socket import SocketMixin
+from .telemetry import TelemetryMixin
+from .user import UserMixin
+from .websocket import WebSocketMixin
+
+
+class ApiMixin(
+    AuthMixin,
+    HandlerMixin,
+    UserMixin,
+    ChannelMixin,
+    SelfMixin,
+    MessageMixin,
+    TelemetryMixin,
+):
+    pass
+
+
+__all__ = [
+    "ApiMixin",
+    "AuthMixin",
+    "ChannelMixin",
+    "HandlerMixin",
+    "MessageMixin",
+    "SelfMixin",
+    "SocketMixin",
+    "TelemetryMixin",
+    "UserMixin",
+    "WebSocketMixin",
+]

pymax/mixins/auth.py

@@ -1,81 +1,81 @@
-import asyncio
-import re
-from typing import Any
-
-from pymax.interfaces import ClientProtocol
-from pymax.payloads import RequestCodePayload, SendCodePayload
-from pymax.static import AuthType, Constants, Opcode
-
-
-class AuthMixin(ClientProtocol):
-    def _check_phone(self) -> bool:
-        return bool(re.match(Constants.PHONE_REGEX.value, self.phone))
-
-    async def _request_code(
-        self, phone: str, language: str = "ru"
-    ) -> dict[str, int | str]:
-        try:
-            self.logger.info("Requesting auth code")
-
-            payload = RequestCodePayload(
-                phone=phone, type=AuthType.START_AUTH, language=language
-            ).model_dump(by_alias=True)
-
-            data = await self._send_and_wait(
-                opcode=Opcode.AUTH_REQUEST, payload=payload
-            )
-            self.logger.debug(
-                "Code request response opcode=%s seq=%s",
-                data.get("opcode"),
-                data.get("seq"),
-            )
-            return data.get("payload")
-        except Exception:
-            self.logger.error("Request code failed", exc_info=True)
-            raise RuntimeError("Request code failed")
-
-    async def _send_code(self, code: str, token: str) -> dict[str, Any]:
-        try:
-            self.logger.info("Sending verification code")
-
-            payload = SendCodePayload(
-                token=token,
-                verify_code=code,
-                auth_token_type=AuthType.CHECK_CODE,
-            ).model_dump(by_alias=True)
-
-            data = await self._send_and_wait(opcode=Opcode.AUTH, payload=payload)
-            self.logger.debug(
-                "Send code response opcode=%s seq=%s",
-                data.get("opcode"),
-                data.get("seq"),
-            )
-            return data.get("payload")
-        except Exception:
-            self.logger.error("Send code failed", exc_info=True)
-            raise RuntimeError("Send code failed")
-
-    async def _login(self) -> None:
-        self.logger.info("Starting login flow")
-        request_code_payload = await self._request_code(self.phone)
-        temp_token = request_code_payload.get("token")
-        if not temp_token or not isinstance(temp_token, str):
-            self.logger.critical("Failed to request code: token missing")
-            raise ValueError("Failed to request code")
-
-        code = await asyncio.to_thread(input, "Введите код: ")
-        if len(code) != 6 or not code.isdigit():
-            self.logger.error("Invalid code format entered")
-            raise ValueError("Invalid code format")
-
-        login_resp = await self._send_code(code, temp_token)
-        token: str | None = (
-            login_resp.get("tokenAttrs", {}).get("LOGIN", {}).get("token")
-        )
-        if not token:
-            self.logger.critical("Failed to login, token not received")
-            raise ValueError("Failed to login, token not received")
-
-        self._token = token
-        self._database.update_auth_token(self._device_id, self._token)
-        self.logger.info("Login successful, token saved to database")
+import asyncio
+import re
+from typing import Any
+
+from pymax.interfaces import ClientProtocol
+from pymax.payloads import RequestCodePayload, SendCodePayload
+from pymax.static import AuthType, Constants, Opcode
+
+
+class AuthMixin(ClientProtocol):
+    def _check_phone(self) -> bool:
+        return bool(re.match(Constants.PHONE_REGEX.value, self.phone))
+
+    async def _request_code(
+        self, phone: str, language: str = "ru"
+    ) -> dict[str, int | str]:
+        try:
+            self.logger.info("Requesting auth code")
+
+            payload = RequestCodePayload(
+                phone=phone, type=AuthType.START_AUTH, language=language
+            ).model_dump(by_alias=True)
+
+            data = await self._send_and_wait(
+                opcode=Opcode.AUTH_REQUEST, payload=payload
+            )
+            self.logger.debug(
+                "Code request response opcode=%s seq=%s",
+                data.get("opcode"),
+                data.get("seq"),
+            )
+            return data.get("payload")
+        except Exception:
+            self.logger.error("Request code failed", exc_info=True)
+            raise RuntimeError("Request code failed")
+
+    async def _send_code(self, code: str, token: str) -> dict[str, Any]:
+        try:
+            self.logger.info("Sending verification code")
+
+            payload = SendCodePayload(
+                token=token,
+                verify_code=code,
+                auth_token_type=AuthType.CHECK_CODE,
+            ).model_dump(by_alias=True)
+
+            data = await self._send_and_wait(opcode=Opcode.AUTH, payload=payload)
+            self.logger.debug(
+                "Send code response opcode=%s seq=%s",
+                data.get("opcode"),
+                data.get("seq"),
+            )
+            return data.get("payload")
+        except Exception:
+            self.logger.error("Send code failed", exc_info=True)
+            raise RuntimeError("Send code failed")
+
+    async def _login(self) -> None:
+        self.logger.info("Starting login flow")
+        request_code_payload = await self._request_code(self.phone)
+        temp_token = request_code_payload.get("token")
+        if not temp_token or not isinstance(temp_token, str):
+            self.logger.critical("Failed to request code: token missing")
+            raise ValueError("Failed to request code")
+
+        code = await asyncio.to_thread(input, "Введите код: ")
+        if len(code) != 6 or not code.isdigit():
+            self.logger.error("Invalid code format entered")
+            raise ValueError("Invalid code format")
+
+        login_resp = await self._send_code(code, temp_token)
+        token: str | None = (
+            login_resp.get("tokenAttrs", {}).get("LOGIN", {}).get("token")
+        )
+        if not token:
+            self.logger.critical("Failed to login, token not received")
+            raise ValueError("Failed to login, token not received")
+
+        self._token = token
+        self._database.update_auth_token(self._device_id, self._token)
+        self.logger.info("Login successful, token saved to database")

pymax/mixins/channel.py

@@ -1,25 +1,25 @@
-from pymax.interfaces import ClientProtocol
-from pymax.payloads import ResolveLinkPayload
-from pymax.static import Opcode
-
-
-class ChannelMixin(ClientProtocol):
-    async def resolve_channel_by_name(self, name: str) -> bool:
-        """
-        Пытается найти канал по его имени
-
-        Args:
-            name (str): Имя канала
-
-        Returns:
-            bool: True, если канал найден
-        """
-        payload = ResolveLinkPayload(
-            link=f"https://max.ru/{name}",
-        ).model_dump(by_alias=True)
-
-        data = await self._send_and_wait(opcode=Opcode.LINK_INFO, payload=payload)
-        if error := data.get("payload", {}).get("error"):
-            self.logger.error("Resolve link error: %s", error)
-            return False
-        return True
+from pymax.interfaces import ClientProtocol
+from pymax.payloads import ResolveLinkPayload
+from pymax.static import Opcode
+
+
+class ChannelMixin(ClientProtocol):
+    async def resolve_channel_by_name(self, name: str) -> bool:
+        """
+        Пытается найти канал по его имени
+
+        Args:
+            name (str): Имя канала
+
+        Returns:
+            bool: True, если канал найден
+        """
+        payload = ResolveLinkPayload(
+            link=f"https://max.ru/{name}",
+        ).model_dump(by_alias=True)
+
+        data = await self._send_and_wait(opcode=Opcode.LINK_INFO, payload=payload)
+        if error := data.get("payload", {}).get("error"):
+            self.logger.error("Resolve link error: %s", error)
+            return False
+        return True