maxapi-python 1.1.20__py3-none-any.whl → 1.2.1__py3-none-any.whl

{maxapi_python-1.1.20.dist-info → maxapi_python-1.2.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: maxapi-python
-Version: 1.1.20
+Version: 1.2.1
 Summary: Python wrapper для API мессенджера Max
 Project-URL: Homepage, https://github.com/ink-developer/PyMax
 Project-URL: Repository, https://github.com/ink-developer/PyMax
@@ -16,6 +16,7 @@ Requires-Dist: aiofiles>=24.1.0
 Requires-Dist: aiohttp>=3.12.15
 Requires-Dist: lz4>=4.4.4
 Requires-Dist: msgpack>=1.1.1
+Requires-Dist: qrcode>=8.2
 Requires-Dist: sqlmodel>=0.0.24
 Requires-Dist: websockets>=15.0
 Description-Content-Type: text/markdown
@@ -35,6 +36,12 @@ Description-Content-Type: text/markdown
     <img src="https://img.shields.io/badge/packaging-uv-D7FF64.svg" alt="Packaging">
 </p>
 
+> [!IMPORTANT]
+> (20.12.25) Из за резкого изменения апи большая часть библиотеки не работает.
+Смотрите [новость](https://t.me/pymax_news/111)
+> 
+> P.s добавил логин по qr в dev/1.2.1
+
 ---
 > ⚠️ **Дисклеймер**
 >
@@ -79,65 +86,47 @@ uv add -U maxapi-python
 
 ```python
 import asyncio
+
 from pymax import MaxClient, Message
+from pymax.filters import Filters
+
+client = MaxClient(
+    phone="+1234567890",
+    work_dir="cache",  # директория для сессий
+)
+
 
-# Инициализация клиента
-phone = "+1234567890"
-client = MaxClient(phone=phone, work_dir="cache")
+# Обработка входящих сообщений
+@client.on_message(Filters.chat(0))  # фильтр по ID чата
+async def on_message(msg: Message) -> None:
+    print(f"[{msg.sender}] {msg.text}")
+
+    await client.send_message(
+        chat_id=msg.chat_id,
+        text="Привет, я бот на PyMax!",
+    )
+
+    await client.add_reaction(
+        chat_id=msg.chat_id,
+        message_id=str(msg.id),
+        reaction="👍",
+    )
 
-# Обработчик входящих сообщений
-@client.on_message()
-async def handle_message(message: Message) -> None:
-    print(f"{message.sender}: {message.text}")
 
-# Обработчик запуска клиента
 @client.on_start
-async def handle_start() -> None:
-    print("Клиент запущен")
+async def on_start() -> None:
+    print(f"Клиент запущен. Ваш ID: {client.me.id}")
 
-    # Получение истории сообщений
+    # Получение истории
     history = await client.fetch_history(chat_id=0)
-    if history:
-        for message in history:
-            user = await client.get_user(message.sender)
-            if user:
-                print(f"{user.names[0].name}: {message.text}")
-
-async def main() -> None:
-    await client.start()
-
-    # Работа с чатами
-    for chat in client.chats:
-        print(f"Чат: {chat.title}")
-
-        # Отправка сообщения
-        message = await client.send_message(
-            "Привет от PyMax!",
-            chat.id,
-            notify=True
-        )
-
-        # Редактирование сообщения
-        await asyncio.sleep(2)
-        await client.edit_message(
-            chat.id,
-            message.id,
-            "Привет от PyMax! (отредактировано)"
-        )
-
-        # Удаление сообщения
-        await asyncio.sleep(2)
-        await client.delete_message(chat.id, [message.id], for_me=False)
-
-    # Работа с диалогами
-    for dialog in client.dialogs:
-        print(f"Диалог: {dialog.last_message.text}")
-
-    # Работа с каналами
-    for channel in client.channels:
-        print(f"Канал: {channel.title}")
-
-    await client.close()
+    print("Последние сообщения из чата 0:")
+    for m in history:
+        print(f"- {m.text}")
+
+
+async def main():
+    await client.start()  # подключение и авторизация
+
 
 if __name__ == "__main__":
     asyncio.run(main())
@@ -145,7 +134,8 @@ if __name__ == "__main__":
 
 ## Документация
 
-[WIP](https://ink-developer.github.io/PyMax)
+[GitHub Pages](https://maxapiteam.github.io/PyMax/)
+[DeepWiki](https://deepwiki.com/MaxApiTeam/PyMax)
 
 ## Лицензия
 

maxapi_python-1.2.1.dist-info/RECORD

@@ -0,0 +1,32 @@
+pymax/__init__.py,sha256=6wUKKwsyxFpWG3b7kwptOvHd-w78C-ygw42iCDBYQvc,1915
+pymax/core.py,sha256=Y4sXaUEtK4pVyYiKf5folyhW_xMFUJ3W-5Hc33b8v0E,14892
+pymax/crud.py,sha256=YC92TyhA2mr1tJCcfd-tvh8umtXKgqJfgiLo7nXUl3Q,3076
+pymax/exceptions.py,sha256=nDUNx7bM-Yjugj-qfIllcrnwLg9JpZroYqfXapjYbMQ,3178
+pymax/files.py,sha256=AvFIr34Desq2p4CNWXIngRqeyTBKMT98VmcnI-zvUU0,3462
+pymax/filters.py,sha256=gSHPJ1Vi37HKPxf0jRRv9Q3iGwhiQjw1MGrCaouqHzs,4325
+pymax/formatter.py,sha256=RJ_5VbY7Li8UM3xL1AvcXo8v1iYnY8GvDDkreaFqtnY,860
+pymax/formatting.py,sha256=XRtuXJGweuNZevJFdPxksDftIrfuMGEA-AOUc_v6IhQ,2484
+pymax/interfaces.py,sha256=wKF1z1QRw8LcjvM9rzSHWXTK6gPb6sDt2UGiQLvyMf8,8790
+pymax/models.py,sha256=PsPGbOkERxesZZltjNrmqhOfRcO44Is2ThbEToREcB8,201
+pymax/navigation.py,sha256=4ia6RGY2pXMArboNhHkbWlWX7LtcYK1VGVXorPX0Pb4,5747
+pymax/payloads.py,sha256=-GEJVXXlmJiFSTX4ToVNzmSZSrvSRe-BLOwYyRxGkWY,7280
+pymax/types.py,sha256=_ARcVXLGHyiGAJKYPd6EU9QDKzz4VwS6kjTu3YEH_u4,35523
+pymax/mixins/__init__.py,sha256=5sXJME34S1EssuDETaN4DLRH7vhMw_Q3Jmay9myAIZM,775
+pymax/mixins/auth.py,sha256=e90vIpEOwAjUxgYMYaG7R6jR_5t9rKsei_mTBQUirL4,14716
+pymax/mixins/channel.py,sha256=W52YnBay1sUYXxF9oAWsz44ZUh_s45jSvKmAyjTbULM,5357
+pymax/mixins/group.py,sha256=LqI1QHmZlmtuQ0-4H1MrNeBV-O9SMDMfHT9f4B_2poE,15189
+pymax/mixins/handler.py,sha256=ETnI8fA386LYJGjWtUhhWzQHREUA78di1aO1oWwtscA,12523
+pymax/mixins/message.py,sha256=AznKKmTMxdzsYl8IecT43RjWpGvlQM85GzSNGFbI8BA,33279
+pymax/mixins/scheduler.py,sha256=rcMfgfZnzu5V6MkcCg6uRgbi-jkc7UyqOjemulydWbc,964
+pymax/mixins/self.py,sha256=Be5L64eNYylGM-NmoxFpQZv1ohsC1Dx_Cs3Om__V96s,6976
+pymax/mixins/socket.py,sha256=tdHgd1NwWoEZhHCDd74XLOHFKUq-rladxhXV8Z_-APU,22860
+pymax/mixins/telemetry.py,sha256=LWr68DNQkPhAjGRDYQ5lORXxC3Yw6M9E8sF0TCNISTE,3609
+pymax/mixins/user.py,sha256=RSZd4t-aq8P2k3cVzNVWBkUf-_xTWILrBzwxLRgk1pw,9450
+pymax/mixins/utils.py,sha256=s3FUf3i_wjn2Gbg5YY1rWZB-90ZEGrrcUuND_MqqSTE,853
+pymax/mixins/websocket.py,sha256=m2swhSHIcFG6iABAik_oWxIpHfr0sxZ74I6VRU-iVO8,17809
+pymax/static/constant.py,sha256=-qJz65V_ESagy7DYHRj3QsfKwyAyVofWzZh0AqBJtXo,1112
+pymax/static/enum.py,sha256=Hk0e6zSbGOJC_9Aw7gNXX3hcavnjzQfDyr8vjW22cFo,4648
+maxapi_python-1.2.1.dist-info/METADATA,sha256=JU86mxRn8zdfsRGWFnNvJ6ea6KHKmmADCvDWvuvIZb4,5855
+maxapi_python-1.2.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+maxapi_python-1.2.1.dist-info/licenses/LICENSE,sha256=hOR249ItqMdcly1A0amqEWRNRTq4Gv5NJtmQ3A5qK4E,1070
+maxapi_python-1.2.1.dist-info/RECORD,,

pymax/core.py

@@ -1,23 +1,24 @@
+from __future__ import annotations
+
 import asyncio
 import contextlib
 import logging
 import socket
 import ssl
 import time
-import traceback
 from collections.abc import Awaitable
 from pathlib import Path
 from typing import TYPE_CHECKING, Any, Literal
+from uuid import UUID
 
-from typing_extensions import Self, override
+from typing_extensions import override
 
 from .crud import Database
 from .exceptions import (
     InvalidPhoneError,
     SocketNotConnectedError,
-    WebSocketNotConnectedError,
 )
-from .formatter import ColoredFormatter
+from .interfaces import BaseClient
 from .mixins import ApiMixin, SocketMixin, WebSocketMixin
 from .payloads import UserAgentPayload
 from .static.constant import (
@@ -27,45 +28,53 @@ from .static.constant import (
 )
 
 if TYPE_CHECKING:
-    from collections.abc import Awaitable, Callable
-    from typing import Any
+    from collections.abc import Callable
 
     import websockets
 
-    from .filters import Filter
+    from pymax.filters import BaseFilter
+
+    from .filters import Filters
     from .types import Channel, Chat, Dialog, Me, Message, ReactionInfo, User
 
 
 logger = logging.getLogger(__name__)
 
 
-class MaxClient(ApiMixin, WebSocketMixin):
+class MaxClient(ApiMixin, WebSocketMixin, BaseClient):
     """
     Основной клиент для работы с WebSocket API сервиса Max.
 
-
-    Args:
-        phone (str): Номер телефона для авторизации.
-        uri (str, optional): URI WebSocket сервера. По умолчанию Constants.WEBSOCKET_URI.value.
-        work_dir (str, optional): Рабочая директория для хранения базы данных. По умолчанию ".".
-        logger (logging.Logger | None): Пользовательский логгер. Если не передан — используется
-            логгер модуля с именем f"{__name__}.MaxClient".
-        headers (UserAgentPayload): Заголовки для подключения к WebSocket.
-        token (str | None, optional): Токен авторизации. Если не передан, будет выполнен
-            процесс логина по номеру телефона.
-        host (str, optional): Хост API сервера. По умолчанию Constants.HOST.value.
-        port (int, optional): Порт API сервера. По умолчанию Constants.PORT.value.
-        registration (bool, optional): Флаг регистрации нового пользователя. По умолчанию False.
-        first_name (str, optional): Имя пользователя для регистрации. Требуется, если registration=True.
-        last_name (str | None, optional): Фамилия пользователя для регистрации.
-        send_fake_telemetry (bool, optional): Флаг отправки фейковой телеметрии. По умолчанию True.
-        proxy (str | Literal[True] | None, optional): Прокси для подключения к WebSocket.
-            (См. https://websockets.readthedocs.io/en/stable/topics/proxies.html).
-        reconnect (bool, optional): Флаг автоматического переподключения при потере соединения. По умолчанию True.
-
-
-    Raises:
-        InvalidPhoneError: Если формат номера телефона неверный.
+    :param phone: Номер телефона для авторизации.
+    :type phone: str
+    :param uri: URI WebSocket сервера.
+    :type uri: str, optional
+    :param work_dir: Рабочая директория для хранения базы данных.
+    :type work_dir: str, optional
+    :param logger: Пользовательский логгер. Если не передан, используется логгер модуля с именем f"{__name__}.MaxClient".
+    :type logger: logging.Logger | None
+    :param headers: Заголовки для подключения к WebSocket.
+    :type headers: UserAgentPayload
+    :param token: Токен авторизации. Если не передан, будет выполнен процесс логина по номеру телефона.
+    :type token: str | None, optional
+    :param host: Хост API сервера.
+    :type host: str, optional
+    :param port: Порт API сервера.
+    :type port: int, optional
+    :param registration: Флаг регистрации нового пользователя.
+    :type registration: bool, optional
+    :param first_name: Имя пользователя для регистрации. Требуется, если registration=True.
+    :type first_name: str, optional
+    :param last_name: Фамилия пользователя для регистрации.
+    :type last_name: str | None, optional
+    :param send_fake_telemetry: Флаг отправки фейковой телеметрии.
+    :type send_fake_telemetry: bool, optional
+    :param proxy: Прокси для подключения к WebSocket (см. https://websockets.readthedocs.io/en/stable/topics/proxies.html).
+    :type proxy: str | Literal[True] | None, optional
+    :param reconnect: Флаг автоматического переподключения при потере соединения.
+    :type reconnect: bool, optional
+
+    :raises InvalidPhoneError: Если формат номера телефона неверный.
     """
 
     def __init__(
@@ -82,6 +91,7 @@ class MaxClient(ApiMixin, WebSocketMixin):
         registration: bool = False,
         first_name: str = "",
         last_name: str | None = None,
+        device_id: UUID | None = None,
         logger: logging.Logger | None = None,
         reconnect: bool = True,
         reconnect_delay: float = 1.0,
@@ -127,7 +137,7 @@ class MaxClient(ApiMixin, WebSocketMixin):
         self._circuit_breaker: bool = False
         self._last_error_time: float = 0.0
 
-        self._device_id = self._database.get_device_id()
+        self._device_id = device_id if device_id is not None else self._database.get_device_id()
         self._file_upload_waiters: dict[int, asyncio.Future[dict[str, Any]]] = {}
 
         self._token = self._database.get_auth_token() or token
@@ -138,19 +148,20 @@ class MaxClient(ApiMixin, WebSocketMixin):
         self._current_screen: str = "chats_list_tab"
 
         self._on_message_handlers: list[
-            tuple[Callable[[Message], Any], Filter | None]
+            tuple[Callable[[Message], Any], BaseFilter[Message] | None]
         ] = []
         self._on_message_edit_handlers: list[
-            tuple[Callable[[Message], Any], Filter | None]
+            tuple[Callable[[Message], Any], BaseFilter[Message] | None]
         ] = []
         self._on_message_delete_handlers: list[
-            tuple[Callable[[Message], Any], Filter | None]
+            tuple[Callable[[Message], Any], BaseFilter[Message] | None]
         ] = []
         self._on_start_handler: Callable[[], Any | Awaitable[Any]] | None = None
-        self._on_reaction_change_handlers: list[
-            tuple[Callable[[str, int, ReactionInfo], Any]]
-        ] = []
-        self._on_chat_update_handlers: list[tuple[Callable[[Chat], Any]]] = []
+        self._on_stop_handler: Callable[[], Any | Awaitable[Any]] | None = None
+        self._on_reaction_change_handlers: list[Callable[[str, int, ReactionInfo], Any]] = []
+        self._on_chat_update_handlers: list[Callable[[Chat], Any | Awaitable[Any]]] = []
+        self._on_raw_receive_handlers: list[Callable[[dict[str, Any]], Any | Awaitable[Any]]] = []
+        self._scheduled_tasks: list[tuple[Callable[[], Any | Awaitable[Any]], float]] = []
 
         self._ssl_context = ssl.create_default_context()
         self._ssl_context.set_ciphers("DEFAULT")
@@ -168,37 +179,18 @@ class MaxClient(ApiMixin, WebSocketMixin):
             self._work_dir,
         )
 
-    def _setup_logger(self) -> None:
-        if not self.logger.handlers:
-            if not self.logger.level:
-                self.logger.setLevel(logging.INFO)
-            handler = logging.StreamHandler()
-            formatter = ColoredFormatter(
-                "%(asctime)s [%(levelname)s] %(name)s: %(message)s",
-                datefmt="%Y-%m-%d %H:%M:%S",
-            )
-            handler.setFormatter(formatter)
-            self.logger.addHandler(handler)
-
     async def _wait_forever(self) -> None:
         try:
             await self.ws.wait_closed()
         except asyncio.CancelledError:
             self.logger.debug("wait_closed cancelled")
 
-    async def _safe_execute(self, coro, *, context: str = "unknown") -> Any:
-        """
-        Безопасно выполняет пользовательскую корутину.
-        Логирует traceback, но не роняет event loop.
+    async def close(self) -> None:
         """
-        try:
-            return await coro
-        except Exception as e:
-            self.logger.error(
-                f"Unhandled exception in {context}: {e}\n{traceback.format_exc()}"
-            )
+        Закрывает клиент и освобождает ресурсы.
 
-    async def close(self) -> None:
+        :return: None
+        """
         try:
             self.logger.info("Closing client")
             if self._recv_task:
@@ -220,100 +212,46 @@ class MaxClient(ApiMixin, WebSocketMixin):
         except Exception:
             self.logger.exception("Error closing client")
 
-    @override
-    def _create_safe_task(
-        self, coro: Awaitable[Any], *, name: str | None = None
-    ) -> asyncio.Task[Any | None]:
-        async def runner():
-            try:
-                return await coro
-            except asyncio.CancelledError:
-                raise
-            except Exception as e:
-                self.logger.exception(
-                    f"Unhandled exception in task {name or coro}: {e}",
-                    exc_info=e,
-                )
-                return None
-
-        task = asyncio.create_task(runner(), name=name)
-        self._background_tasks.add(task)
-        return task
-
     async def _post_login_tasks(self, sync: bool = True) -> None:
         if sync:
             await self._sync()
 
-        if self._on_start_handler:
-            self.logger.debug("Calling on_start handler")
-            result = self._on_start_handler()
-            if asyncio.iscoroutine(result):
-                await self._safe_execute(result, context="on_start handler")
-
+        self.logger.debug("is_connected=%s before starting ping", self.is_connected)
         ping_task = asyncio.create_task(self._send_interactive_ping())
         ping_task.add_done_callback(self._log_task_exception)
         self._background_tasks.add(ping_task)
 
+        start_scheduled_task = asyncio.create_task(self._start_scheduled_tasks())
+        start_scheduled_task.add_done_callback(self._log_task_exception)
+
         if self._send_fake_telemetry:
             telemetry_task = asyncio.create_task(self._start())
             telemetry_task.add_done_callback(self._log_task_exception)
             self._background_tasks.add(telemetry_task)
 
-    async def _cleanup_client(self) -> None:
-        for task in list(self._background_tasks):
-            task.cancel()
-            try:
-                await task
-            except asyncio.CancelledError:
-                pass
-            except Exception:
-                self.logger.debug(
-                    "Background task raised during cancellation", exc_info=True
-                )
-            self._background_tasks.discard(task)
-
-        if self._recv_task:
-            self._recv_task.cancel()
-            with contextlib.suppress(asyncio.CancelledError):
-                await self._recv_task
-            self._recv_task = None
-
-        if self._outgoing_task:
-            self._outgoing_task.cancel()
-            with contextlib.suppress(asyncio.CancelledError):
-                await self._outgoing_task
-            self._outgoing_task = None
-
-        for fut in self._pending.values():
-            if not fut.done():
-                fut.set_exception(WebSocketNotConnectedError)
-        self._pending.clear()
-
-        if self._ws:
-            try:
-                await self._ws.close()
-            except Exception:
-                self.logger.debug("Error closing ws during cleanup", exc_info=True)
-            self._ws = None
-
-        self.is_connected = False
-        self.logger.info("Client start() cleaned up")
+        if self._on_start_handler:
+            self.logger.debug("Calling on_start handler")
+            result = self._on_start_handler()
+            if asyncio.iscoroutine(result):
+                await self._safe_execute(result, context="on_start handler")
 
-    async def login_with_code(
-        self, temp_token: str, code: str, start: bool = False
-    ) -> None:
+    async def login_with_code(self, temp_token: str, code: str, start: bool = False) -> None:
         """
         Завершает кастомный login flow: отправляет код, сохраняет токен и запускает пост-логин задачи.
 
-        Args:
-            temp_token (str): Временный токен, полученный из request_code()
-            code (str): Код, введённый пользователем
-
-        Returns:
-            str: Токен для входа
+        :param temp_token: Временный токен, полученный из request_code.
+        :type temp_token: str
+        :param code: Код верификации (6 цифр).
+        :type code: str
+        :param start: Флаг запуска пост-логин задач и ожидания навсегда. Если False, только сохраняет токен.
+        :type start: bool, optional
+        :return: None
+        :rtype: None
         """
         resp = await self._send_code(code, temp_token)
         token = resp.get("tokenAttrs", {}).get("LOGIN", {}).get("token")
+        if not token:
+            raise ValueError("Login response did not contain tokenAttrs.LOGIN.token")
         self._token = token
         self._database.update_auth_token(self._device_id, token)
         if start:
@@ -336,6 +274,9 @@ class MaxClient(ApiMixin, WebSocketMixin):
         Запускает клиент, подключается к WebSocket, авторизует
         пользователя (если нужно) и запускает фоновый цикл.
         Теперь включает безопасный reconnect-loop, если self.reconnect=True.
+
+        :return: None
+        :rtype: None
         """
 
         while True:
@@ -353,8 +294,8 @@ class MaxClient(ApiMixin, WebSocketMixin):
 
                 if self._token is None:
                     await self._login()
-                else:
-                    await self._sync()
+
+                await self._sync()
 
                 await self._post_login_tasks(sync=False)
 
@@ -376,18 +317,6 @@ class MaxClient(ApiMixin, WebSocketMixin):
             self.logger.info("Reconnect enabled — restarting client")
             await asyncio.sleep(self.reconnect_delay)
 
-    async def idle(self):
-        await asyncio.Event().wait()
-
-    async def __aenter__(self) -> Self:
-        self._create_safe_task(self.start(), name="start")
-        while not self.is_connected:
-            await asyncio.sleep(0.05)
-        return self
-
-    async def __aexit__(self, exc_type, exc, tb) -> None:
-        await self.close()
-
 
 class SocketMaxClient(SocketMixin, MaxClient):
     @override
@@ -402,12 +331,6 @@ class SocketMaxClient(SocketMixin, MaxClient):
 
     @override
     async def _cleanup_client(self):
-        """
-        Socket-specific cleanup: cancel background tasks, set pending futures
-        exceptions to SocketNotConnectedError, and close socket.
-        """
-        from .exceptions import SocketNotConnectedError
-
         for task in list(self._background_tasks):
             task.cancel()
             try:

pymax/crud.py

@@ -31,9 +31,8 @@ class Database:
 
     def get_device_id(self) -> UUID:
         with self.get_session() as session:
-            device_id = cast(
-                UUID | None, session.exec(select(Auth.device_id)).first()
-            )
+            device_id = session.exec(select(Auth.device_id)).first()
+
             if device_id is None:
                 auth = Auth()
                 session.add(auth)
@@ -51,9 +50,7 @@ class Database:
 
     def update_auth_token(self, device_id: UUID, token: str) -> None:
         with self.get_session() as session:
-            auth = session.exec(
-                select(Auth).where(Auth.device_id == device_id)
-            ).first()
+            auth = session.exec(select(Auth).where(Auth.device_id == device_id)).first()
             if auth:
                 auth.token = token
                 session.add(auth)
@@ -86,7 +83,6 @@ class Database:
         with self.get_session() as session:
             rows = session.exec(select(Auth)).all()
             if not rows:
-                # Create default Auth with device type from enum
                 auth = Auth(device_type=DeviceType.WEB.value)
                 session.add(auth)
                 session.commit()