fix: using cache only a single time

Author: humbertogontijo

roborock/containers.py

@@ -76,6 +76,7 @@ def decamelize_obj(d: dict | list, ignore_keys: list[str]):
 @dataclass
 class RoborockBase:
     _ignore_keys = []  # type: ignore
+    is_cached = False
 
     @classmethod
     def from_dict(cls, data: dict[str, Any]):

roborock/local_api.py

@@ -7,7 +7,7 @@
 
 import async_timeout
 
-from . import DeviceData
+from . import DeviceData, RoborockBase
 from .api import COMMANDS_SECURED, QUEUE_TIMEOUT, RoborockClient
 from .exceptions import CommandVacuumError, RoborockConnectionException, RoborockException
 from .protocol import MessageParser
@@ -135,4 +135,9 @@ async def send_message(self, roborock_messages: list[RoborockMessage] | Roborock
         if exception:
             await self.async_disconnect()
             raise exception
+        is_cached = next(
+            (response for response in responses if isinstance(response, RoborockBase) and response.is_cached), None
+        )
+        if is_cached:
+            await self.async_disconnect()
         return responses

roborock/util.py

@@ -36,6 +36,7 @@ def wrapped(*args, **kwargs):
 
 class CacheableResult:
     last_run_result = None
+    use_count = 0
 
 
 RT = TypeVar("RT", bound=Callable[..., Coroutine])
@@ -48,9 +49,12 @@ def fallback_cache(func: RT) -> RT:
     async def wrapped(*args, **kwargs):
         try:
             cache.last_run_result = await func(*args, **kwargs)
+            cache.use_count = 0
         except Exception as e:
-            if cache.last_run_result is None:
+            if cache.last_run_result is None or cache.use_count > 0:
                 raise e
+            cache.last_run_result.is_cached = True
+            cache.use_count += 1
         return cache.last_run_result
 
     return wrapped  # type: ignore