maxapi-python 0.1.2__py3-none-any.whl → 1.0.1__py3-none-any.whl

pymax/mixins/telemetry.py

@@ -0,0 +1,114 @@
+import asyncio
+import json
+import random
+import time
+
+from pymax.interfaces import ClientProtocol
+from pymax.navigation import Navigation
+from pymax.payloads import (
+    NavigationEventParams,
+    NavigationEventPayload,
+    NavigationPayload,
+)
+from pymax.static import Opcode
+
+
+class TelemetryMixin(ClientProtocol):
+    async def _send_navigation_event(
+        self, events: list[NavigationEventPayload]
+    ) -> None:
+        try:
+            payload = NavigationPayload(events=events).model_dump(by_alias=True)
+            data = await self._send_and_wait(
+                opcode=Opcode.LOG,
+                payload=payload,
+            )
+
+            payload_data = data.get("payload")
+
+            if payload_data and (error := payload_data.get("error")):
+                self.logger.error("Failed to send navigation event: %s", error)
+                return
+        except Exception:
+            self.logger.warning("Failed to send navigation event", exc_info=True)
+            return
+
+    async def _send_cold_start(self) -> None:
+        if not self.me:
+            self.logger.error("Cannot send cold start, user not set")
+            return
+
+        payload = NavigationEventPayload(
+            event="COLD_START",
+            time=int(time.time() * 1000),
+            user_id=self.me.id,
+            params=NavigationEventParams(
+                action_id=self._action_id,
+                screen_to=Navigation.get_screen_id("chats_list_tab"),
+                screen_from=1,
+                source_id=1,
+                session_id=self._session_id,
+            ),
+        )
+
+        self._action_id += 1
+
+        await self._send_navigation_event([payload])
+
+    async def _send_random_navigation(self) -> None:
+        if not self.me:
+            self.logger.error("Cannot send navigation event, user not set")
+            return
+
+        screen_from = self._current_screen
+        screen_to = Navigation.get_random_navigation(screen_from)
+
+        self._action_id += 1
+        self._current_screen = screen_to
+
+        payload = NavigationEventPayload(
+            event="NAV",
+            time=int(time.time() * 1000),
+            user_id=self.me.id,
+            params=NavigationEventParams(
+                action_id=self._action_id,
+                screen_from=Navigation.get_screen_id(screen_from),
+                screen_to=Navigation.get_screen_id(screen_to),
+                source_id=1,
+                session_id=self._session_id,
+            ),
+        )
+
+        await self._send_navigation_event([payload])
+
+    def _get_random_sleep_time(self) -> int:
+        # TODO: вынести в статик
+        sleep_options = [
+            (1000, 3000),
+            (300, 1000),
+            (60, 300),
+            (5, 60),
+            (5, 20),
+        ]
+
+        weights = [0.05, 0.10, 0.15, 0.20, 0.50]
+
+        low, high = random.choices(sleep_options, weights=weights, k=1)[0]
+        return random.randint(low, high)
+
+    async def _start(self) -> None:
+        if not self.is_connected:
+            self.logger.error("Cannot start telemetry, client not connected")
+            return
+
+        await self._send_cold_start()
+
+        try:
+            while self.is_connected:
+                await self._send_random_navigation()
+                await asyncio.sleep(self._get_random_sleep_time())
+
+        except asyncio.CancelledError:
+            self.logger.debug("Telemetry task cancelled")
+        except Exception:
+            self.logger.warning("Telemetry task failed", exc_info=True)

pymax/mixins/user.py

@@ -1,82 +1,82 @@
-from pymax.interfaces import ClientProtocol
-from pymax.payloads import FetchContactsPayload
-from pymax.static import Opcode
-from pymax.types import User
-
-
-class UserMixin(ClientProtocol):
-    def get_cached_user(self, user_id: int) -> User | None:
-        """
-        Получает юзера из кеша по его ID
-
-        Args:
-            user_id (int): ID пользователя.
-
-        Returns:
-            User | None: Объект User или None при ошибке.
-        """
-        user = self._users.get(user_id)
-        self.logger.debug("get_cached_user id=%s hit=%s", user_id, bool(user))
-        return user
-
-    async def get_users(self, user_ids: list[int]) -> list[User]:
-        """
-        Получает информацию о пользователях по их ID (с кешем).
-        """
-        self.logger.debug("get_users ids=%s", user_ids)
-        cached = {uid: self._users[uid] for uid in user_ids if uid in self._users}
-        missing_ids = [uid for uid in user_ids if uid not in self._users]
-
-        if missing_ids:
-            self.logger.debug("Fetching missing users: %s", missing_ids)
-            fetched_users = await self.fetch_users(missing_ids)
-            if fetched_users:
-                for user in fetched_users:
-                    self._users[user.id] = user
-                    cached[user.id] = user
-
-        ordered = [cached[uid] for uid in user_ids if uid in cached]
-        self.logger.debug("get_users result_count=%d", len(ordered))
-        return ordered
-
-    async def get_user(self, user_id: int) -> User | None:
-        """
-        Получает информацию о пользователе по его ID (с кешем).
-        """
-        self.logger.debug("get_user id=%s", user_id)
-        if user_id in self._users:
-            return self._users[user_id]
-
-        users = await self.fetch_users([user_id])
-        if users:
-            self._users[user_id] = users[0]
-            return users[0]
-        return None
-
-    async def fetch_users(self, user_ids: list[int]) -> None | list[User]:
-        """
-        Получает информацию о пользователях по их ID.
-        """
-        try:
-            self.logger.info("Fetching users count=%d", len(user_ids))
-
-            payload = FetchContactsPayload(contact_ids=user_ids).model_dump(
-                by_alias=True
-            )
-
-            data = await self._send_and_wait(
-                opcode=Opcode.CONTACT_INFO, payload=payload
-            )
-            if error := data.get("payload", {}).get("error"):
-                self.logger.error("Fetch users error: %s", error)
-                return None
-
-            users = [User.from_dict(u) for u in data["payload"].get("contacts", [])]
-            for user in users:
-                self._users[user.id] = user
-
-            self.logger.debug("Fetched users: %d", len(users))
-            return users
-        except Exception:
-            self.logger.exception("Fetch users failed")
-            return []
+from pymax.interfaces import ClientProtocol
+from pymax.payloads import FetchContactsPayload
+from pymax.static import Opcode
+from pymax.types import User
+
+
+class UserMixin(ClientProtocol):
+    def get_cached_user(self, user_id: int) -> User | None:
+        """
+        Получает юзера из кеша по его ID
+
+        Args:
+            user_id (int): ID пользователя.
+
+        Returns:
+            User | None: Объект User или None при ошибке.
+        """
+        user = self._users.get(user_id)
+        self.logger.debug("get_cached_user id=%s hit=%s", user_id, bool(user))
+        return user
+
+    async def get_users(self, user_ids: list[int]) -> list[User]:
+        """
+        Получает информацию о пользователях по их ID (с кешем).
+        """
+        self.logger.debug("get_users ids=%s", user_ids)
+        cached = {uid: self._users[uid] for uid in user_ids if uid in self._users}
+        missing_ids = [uid for uid in user_ids if uid not in self._users]
+
+        if missing_ids:
+            self.logger.debug("Fetching missing users: %s", missing_ids)
+            fetched_users = await self.fetch_users(missing_ids)
+            if fetched_users:
+                for user in fetched_users:
+                    self._users[user.id] = user
+                    cached[user.id] = user
+
+        ordered = [cached[uid] for uid in user_ids if uid in cached]
+        self.logger.debug("get_users result_count=%d", len(ordered))
+        return ordered
+
+    async def get_user(self, user_id: int) -> User | None:
+        """
+        Получает информацию о пользователе по его ID (с кешем).
+        """
+        self.logger.debug("get_user id=%s", user_id)
+        if user_id in self._users:
+            return self._users[user_id]
+
+        users = await self.fetch_users([user_id])
+        if users:
+            self._users[user_id] = users[0]
+            return users[0]
+        return None
+
+    async def fetch_users(self, user_ids: list[int]) -> None | list[User]:
+        """
+        Получает информацию о пользователях по их ID.
+        """
+        try:
+            self.logger.info("Fetching users count=%d", len(user_ids))
+
+            payload = FetchContactsPayload(contact_ids=user_ids).model_dump(
+                by_alias=True
+            )
+
+            data = await self._send_and_wait(
+                opcode=Opcode.CONTACT_INFO, payload=payload
+            )
+            if error := data.get("payload", {}).get("error"):
+                self.logger.error("Fetch users error: %s", error)
+                return None
+
+            users = [User.from_dict(u) for u in data["payload"].get("contacts", [])]
+            for user in users:
+                self._users[user.id] = user
+
+            self.logger.debug("Fetched users: %d", len(users))
+            return users
+        except Exception:
+            self.logger.exception("Fetch users failed")
+            return []