Добавил варн для питона 3.12

Author: ink-developer

examples/example.py

@@ -10,7 +10,7 @@
 
 
 # client = MaxClient(phone=phone, work_dir="cache")
-client = SocketMaxClient(phone=phone, work_dir="cache", send_fake_telemetry=False)
+client = SocketMaxClient(phone=phone, work_dir="cache")
 
 
 async def main() -> None:

src/pymax/core.py

@@ -7,18 +7,17 @@
 from pathlib import Path
 from typing import TYPE_CHECKING, Any
 
-import websockets
 from typing_extensions import override
 
-from pymax.mixins import self
-
 from .crud import Database
-from .exceptions import InvalidPhoneError, WebSocketNotConnectedError
+from .exceptions import InvalidPhoneError
 from .mixins import ApiMixin, SocketMixin, WebSocketMixin
 from .static import Constants
 from .types import Channel, Chat, Dialog, Me, Message, User
 
 if TYPE_CHECKING:
+    import websockets
+
     from .filters import Filter
 
 logger = logging.getLogger(__name__)
@@ -95,6 +94,11 @@ def __init__(
         self._on_start_handler: Callable[[], Any | Awaitable[Any]] | None = None
         self._background_tasks: set[asyncio.Task[Any]] = set()
         self._ssl_context = ssl.create_default_context()
+        self._ssl_context.set_ciphers("DEFAULT")
+        self._ssl_context.check_hostname = True
+        self._ssl_context.verify_mode = ssl.CERT_REQUIRED
+        self._ssl_context.minimum_version = ssl.TLSVersion.TLSv1_2
+        self._ssl_context.load_default_certs()
         self._socket: socket.socket | None = None
         self.logger = logger or logging.getLogger(f"{__name__}.MaxClient")
         self._setup_logger()

src/pymax/mixins/socket.py

@@ -1,5 +1,7 @@
 import asyncio
 import socket
+import ssl
+import sys
 from typing import Any
 
 import lz4.block
@@ -8,7 +10,6 @@
 
 from pymax.filters import Message
 from pymax.interfaces import ClientProtocol
-from pymax.mixins import self
 from pymax.payloads import BaseWebSocketMessage, SyncPayload
 from pymax.static import Opcode
 from pymax.types import Channel, Chat, Dialog, Me
@@ -69,6 +70,15 @@ def _pack_packet(
 
     async def _connect(self, user_agent: dict[str, Any]) -> dict[str, Any]:
         try:
+            if sys.version_info[:2] == (3, 12):
+                self.logger.warning(
+                    """
+===============================================================
+         ⚠️⚠️ \033[0;31mWARNING: Python 3.12 detected!\033[0m ⚠️⚠️
+Socket connections may be unstable, SSL issues are possible.
+===============================================================
+    """
+                )
             self.logger.info("Connecting to socket %s:%s", self.host, self.port)
             loop = asyncio.get_running_loop()
             raw_sock = await loop.run_in_executor(
@@ -108,7 +118,6 @@ async def _recv_loop(self) -> None:
             self.logger.warning("Recv loop started without socket instance")
             return
 
-        self.logger.warning(">>> Recv loop started (socket)")
         loop = asyncio.get_running_loop()
 
         def _recv_exactly(n: int) -> bytes:
@@ -283,6 +292,8 @@ async def _send_and_wait(
         cmd: int = 0,
         timeout: float = 10.0,
     ) -> dict[str, Any]:
+        if not self.is_connected or self._socket is None:
+            raise ConnectionError("Socket not connected")
         sock = self.sock
         msg = self._make_message(opcode, payload, cmd)
         loop = asyncio.get_running_loop()
@@ -303,11 +314,21 @@ async def _send_and_wait(
                 data.get("opcode"),
             )
             return data
+
+        except (ssl.SSLEOFError, ssl.SSLError, ConnectionError):
+            self.logger.warning("Connection lost, reconnecting...")
+            self.is_connected = False
+            try:
+                await self._connect(self.user_agent)
+            except Exception:
+                self.logger.error("Reconnect failed", exc_info=True)
+                raise
         except Exception:
             self.logger.exception(
                 "Send and wait failed (opcode=%s, seq=%s)", opcode, msg["seq"]
             )
             raise RuntimeError("Send and wait failed (socket)")
+
         finally:
             self._pending.pop(msg["seq"], None)