fix: typing and versioning (#134)

* fix: change some typing

* fix: include poetry lock

* fix: linting

* fix: linting

.gitignore

@@ -3,7 +3,6 @@ venv
 .venv
 .idea
 roborock/__pycache__
-poetry.lock
 *.pyc
 .coverage
 

pyproject.toml

@@ -31,7 +31,7 @@ construct = "^2.10.56"
 
 
 [build-system]
-requires = ["poetry-core==1.6.1"]
+requires = ["poetry-core==1.7.0"]
 build-backend = "poetry.core.masonry.api"
 
 [tool.poetry.group.dev.dependencies]

roborock/api.py

@@ -348,7 +348,7 @@ def _async_response(
 
     def _get_payload(
         self,
-        method: RoborockCommand,
+        method: RoborockCommand | str,
         params: list | dict | None = None,
         secured=False,
     ):
@@ -380,15 +380,15 @@ async def send_message(self, roborock_message: RoborockMessage):
 
     async def _send_command(
         self,
-        method: RoborockCommand,
+        method: RoborockCommand | str,
         params: list | dict | None = None,
     ):
         raise NotImplementedError
 
     @final
     async def send_command(
         self,
-        method: RoborockCommand,
+        method: RoborockCommand | str,
         params: list | dict | None = None,
         return_type: type[RT] | None = None,
     ) -> RT:

roborock/cloud_api.py

@@ -198,7 +198,7 @@ async def send_message(self, roborock_message: RoborockMessage):
 
     async def _send_command(
         self,
-        method: RoborockCommand,
+        method: RoborockCommand | str,
         params: list | dict | None = None,
     ):
         request_id, timestamp, payload = super()._get_payload(method, params, True)

roborock/command_cache.py

@@ -168,11 +168,11 @@ def get_change_commands(attr: RoborockAttribute) -> list[RoborockCommand]:
     return [command for command in commands if command is not None]
 
 
-cache_map_by_get_command: dict[RoborockCommand, CacheableAttribute] = {
+cache_map_by_get_command: dict[RoborockCommand | str, CacheableAttribute] = {
     attribute.get_command: cacheable_attribute for cacheable_attribute, attribute in cache_map.items()
 }
 
-cache_map_by_change_command: dict[RoborockCommand, CacheableAttribute] = {
+cache_map_by_change_command: dict[RoborockCommand | str, CacheableAttribute] = {
     command: cacheable_attribute
     for cacheable_attribute, attribute in cache_map.items()
     for command in get_change_commands(attribute)
@@ -195,7 +195,7 @@ class CacheableAttributeResult:
     type: CommandType
 
 
-def find_cacheable_attribute(method: RoborockCommand) -> CacheableAttributeResult | None:
+def find_cacheable_attribute(method: RoborockCommand | str) -> CacheableAttributeResult | None:
     if method is None:
         return None
 

roborock/containers.py

@@ -3,7 +3,7 @@
 import datetime
 import logging
 import re
-from dataclasses import asdict, dataclass
+from dataclasses import asdict, dataclass, field
 from datetime import timezone
 from enum import Enum
 from typing import Any, NamedTuple
@@ -168,10 +168,10 @@ class HomeDataProductSchema(RoborockBase):
 
 @dataclass
 class HomeDataProduct(RoborockBase):
-    id: str | None = None
-    name: str | None = None
+    id: str
+    name: str
+    model: str
     code: str | None = None
-    model: str | None = None
     iconurl: str | None = None
     attribute: Any | None = None
     capability: int | None = None
@@ -216,14 +216,14 @@ class HomeDataRoom(RoborockBase):
 
 @dataclass
 class HomeData(RoborockBase):
-    id: int | None = None
-    name: str | None = None
+    id: int
+    name: str
+    products: list[HomeDataProduct] = field(default_factory=lambda: [])
+    devices: list[HomeDataDevice] = field(default_factory=lambda: [])
+    received_devices: list[HomeDataDevice] = field(default_factory=lambda: [])
     lon: Any | None = None
     lat: Any | None = None
     geo_name: Any | None = None
-    products: list[HomeDataProduct] | None = None
-    devices: list[HomeDataDevice] | None = None
-    received_devices: list[HomeDataDevice] | None = None
     rooms: list[HomeDataRoom] | None = None
 
     def get_all_devices(self) -> list[HomeDataDevice]:

roborock/local_api.py

@@ -82,7 +82,9 @@ async def async_disconnect(self) -> None:
         async with self._mutex:
             self.sync_disconnect()
 
-    def build_roborock_message(self, method: RoborockCommand, params: list | dict | None = None) -> RoborockMessage:
+    def build_roborock_message(
+        self, method: RoborockCommand | str, params: list | dict | None = None
+    ) -> RoborockMessage:
         secured = True if method in COMMANDS_SECURED else False
         request_id, timestamp, payload = self._get_payload(method, params, secured)
         request_protocol = RoborockMessageProtocol.GENERAL_REQUEST
@@ -120,7 +122,7 @@ async def ping(self):
 
     async def _send_command(
         self,
-        method: RoborockCommand,
+        method: RoborockCommand | str,
         params: list | dict | None = None,
     ):
         roborock_message = self.build_roborock_message(method, params)