|
4 | 4 | import socket |
5 | 5 | import ssl |
6 | 6 | import sys |
7 | | -import time |
8 | | -import traceback |
9 | | -from collections.abc import Callable |
10 | | -from typing import Any, Literal |
| 7 | +from typing import Any |
11 | 8 | from urllib.parse import urlparse |
12 | 9 |
|
13 | 10 | import lz4.block |
|
17 | 14 | from pymax.exceptions import Error, SocketNotConnectedError, SocketSendError |
18 | 15 | from pymax.filters import BaseFilter |
19 | 16 | from pymax.interfaces import BaseTransport |
20 | | -from pymax.payloads import BaseWebSocketMessage, SyncPayload, UserAgentPayload |
21 | | -from pymax.protocols import ClientProtocol |
| 17 | +from pymax.payloads import UserAgentPayload |
22 | 18 | from pymax.static.constant import ( |
23 | | - DEFAULT_PING_INTERVAL, |
24 | 19 | DEFAULT_TIMEOUT, |
25 | 20 | RECV_LOOP_BACKOFF_DELAY, |
26 | 21 | ) |
27 | | -from pymax.static.enum import ChatType, MessageStatus, Opcode |
| 22 | +from pymax.static.enum import Opcode |
28 | 23 | from pymax.types import ( |
29 | | - Channel, |
30 | 24 | Chat, |
31 | | - Dialog, |
32 | | - Me, |
33 | | - Message, |
34 | | - ReactionCounter, |
35 | | - ReactionInfo, |
36 | | - User, |
37 | 25 | ) |
38 | 26 |
|
39 | 27 |
|
40 | 28 | class SocketMixin(BaseTransport): |
41 | 29 | def _close_socket_safely(self) -> None: |
42 | 30 | if self._socket is not None: |
43 | 31 | sock = self._socket |
44 | | - self._socket = None # Nullify reference first to prevent double-closes |
| 32 | + self._socket = None |
45 | 33 | with contextlib.suppress(ssl.SSLError, Exception): |
46 | 34 | sock.close() |
47 | 35 |
|
@@ -365,13 +353,15 @@ async def _recv_loop(self) -> None: |
365 | 353 |
|
366 | 354 | while True: |
367 | 355 | try: |
368 | | - sock = self._socket # Get fresh reference on each iteration |
| 356 | + sock = self._socket |
369 | 357 | if sock is None: |
370 | 358 | self.logger.warning("Socket became None, exiting recv loop") |
371 | 359 | break |
372 | 360 |
|
373 | 361 | header = await self._parse_header(loop, sock) |
374 | 362 |
|
| 363 | + self.logger.debug("Received header: %s", header) |
| 364 | + |
375 | 365 | if not header: |
376 | 366 | break |
377 | 367 |
|
|
0 commit comments