Refactor device_id type to UUID in multiple files and update related imports

Author: ink-developer

redocs/source/examples.rst

@@ -63,6 +63,7 @@ Command Handler
 
     import asyncio
     from pymax import MaxClient
+    from datetime import datetime
     from pymax.filters import Filters
 
     client = MaxClient(phone="+79001234567")

src/pymax/core.py

@@ -8,11 +8,10 @@
 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 pymax.filters import BaseFilter
-
 from .crud import Database
 from .exceptions import (
     InvalidPhoneError,
@@ -29,13 +28,13 @@
 )
 
 if TYPE_CHECKING:
-    from collections.abc import Awaitable, Callable
-    from typing import Any
-    from uuid import UUID
+    from collections.abc import Callable
 
     import websockets
 
-    from .filters import filters
+    from pymax.filters import BaseFilter
+
+    from .filters import Filters
     from .types import Channel, Chat, Dialog, Me, Message, ReactionInfo, User
 
 
@@ -92,7 +91,7 @@ def __init__(
         registration: bool = False,
         first_name: str = "",
         last_name: str | None = None,
-        device_id: str | None = None,
+        device_id: UUID | None = None,
         logger: logging.Logger | None = None,
         reconnect: bool = True,
         reconnect_delay: float = 1.0,

src/pymax/crud.py

@@ -29,7 +29,7 @@ def get_auth_token(self) -> str | None:
             token = cast(str | None, session.exec(select(Auth.token)).first())
             return token
 
-    def get_device_id(self) -> str:
+    def get_device_id(self) -> UUID:
         with self.get_session() as session:
             device_id = session.exec(select(Auth.device_id)).first()
 

src/pymax/interfaces.py

@@ -31,7 +31,7 @@ def __init__(self, logger: Logger) -> None:
         self._users: dict[int, User] = {}
         self.chats: list[Chat] = []
         self._database: Database
-        self._device_id: str
+        self._device_id: UUID
         self.uri: str
         self.is_connected: bool = False
         self.phone: str

src/pymax/mixins/message.py

@@ -1,7 +1,6 @@
 import asyncio
 import time
 from http import HTTPStatus
-from io import BytesIO
 from pathlib import Path
 
 import aiohttp
@@ -40,7 +39,6 @@
     FileRequest,
     Message,
     ReactionInfo,
-    VideoAttach,
     VideoRequest,
 )
 
@@ -257,7 +255,7 @@ async def _upload_video(self, video: Video) -> None | Attach:
                             return None
             except OSError as e:
                 if "malloc failure" in str(e) or "BUF" in str(e):
-                    self.logger.error(
+                    self.logger.exception(
                         "Memory error during video upload. File too large or insufficient memory. Try uploading smaller files or free up memory."
                     )
                     self._file_upload_waiters.pop(int(video_id), None)
@@ -416,9 +414,9 @@ async def send_message(
 
         elements = []
         clean_text = None
-        raw_elements = Formatting.get_elements_from_markdown(text)[0]
+        raw_elements, parsed_text = Formatting.get_elements_from_markdown(text)
         if raw_elements:
-            clean_text = Formatting.get_elements_from_markdown(text)[1]
+            clean_text = parsed_text
         elements = [
             MessageElement(type=e.type, length=e.length, from_=e.from_)
             for e in raw_elements

src/pymax/models.py

@@ -5,4 +5,4 @@
 
 class Auth(SQLModel, table=True):
     token: str | None = None
-    device_id: str = Field(default_factory=lambda: str(uuid4()), primary_key=True)
+    device_id: UUID = Field(default_factory=uuid4, primary_key=True)