pymammotion 0.2.8__py3-none-any.whl → 0.2.9__py3-none-any.whl

pymammotion/aliyun/cloud_gateway.py

@@ -71,7 +71,7 @@ class CloudIOTGateway:
 
     converter = DatatypeConverter()
 
-    def __init__(self, connect_response: ConnectResponse | None, login_by_oauth_response: LoginByOAuthResponse | None, aep_response: AepResponse | None, session_by_authcode_response: SessionByAuthCodeResponse | None, region_response: RegionResponse | None, dev_by_account: ListingDevByAccountResponse | None):
+    def __init__(self, connect_response: ConnectResponse | None = None, login_by_oauth_response: LoginByOAuthResponse | None = None, aep_response: AepResponse | None = None, session_by_authcode_response: SessionByAuthCodeResponse | None = None, region_response: RegionResponse | None = None, dev_by_account: ListingDevByAccountResponse | None = None):
         """Initialize the CloudIOTGateway."""
         self._app_key = APP_KEY
         self._app_secret = APP_SECRET

pymammotion/mammotion/devices/mammotion.py

@@ -235,7 +235,7 @@ class Mammotion(object):
                 return
 
         self.cloud_client = cloud_client
-        self.mqtt = MammotionMQTT(region_id=cloud_client._region.data.regionId,
+        self.mqtt = MammotionMQTT(region_id=cloud_client._region_response.data.regionId,
                                         product_key=cloud_client._aep_response.data.productKey,
                                         device_name=cloud_client._aep_response.data.deviceName,
                                         device_secret=cloud_client._aep_response.data.deviceSecret,
@@ -332,8 +332,9 @@ class MammotionBaseDevice:
 
     _mower: MowingDevice
     _state_manager: StateManager
+    _cloud_device: Device | None = None
 
-    def __init__(self, device: MowingDevice) -> None:
+    def __init__(self, device: MowingDevice, cloud_device: Device | None = None) -> None:
         """Initialize MammotionBaseDevice."""
         self.loop = asyncio.get_event_loop()
         self._raw_data = LubaMsg().to_dict(casing=betterproto.Casing.SNAKE)
@@ -342,12 +343,13 @@ class MammotionBaseDevice:
         self._state_manager.gethash_ack_callback = self.datahash_response
         self._state_manager.get_commondata_ack_callback = self.commdata_response
         self._notify_future: asyncio.Future[bytes] | None = None
+        self._cloud_device = cloud_device
 
     async def datahash_response(self, hash_ack: NavGetHashListAck):
         """Handle datahash responses."""
         result_hash = 0
         while hash_ack.data_couple[0] != result_hash:
-            data = await self._send_command_with_args("synchronize_hash_data", hash_num=hash_ack.data_couple[0])
+            data = await self.queue_command("synchronize_hash_data", hash_num=hash_ack.data_couple[0])
             msg = LubaMsg().parse(data)
             if betterproto.serialized_on_wire(msg.nav.toapp_get_commondata_ack):
                 result_hash = msg.nav.toapp_get_commondata_ack.hash
@@ -365,7 +367,7 @@ class MammotionBaseDevice:
                 return
             result_hash = 0
             while data_hash != result_hash:
-                data = await self._send_command_with_args("synchronize_hash_data", hash_num=data_hash)
+                data = await self.queue_command("synchronize_hash_data", hash_num=data_hash)
                 msg = LubaMsg().parse(data)
                 if betterproto.serialized_on_wire(msg.nav.toapp_get_commondata_ack):
                     result_hash = msg.nav.toapp_get_commondata_ack.hash
@@ -377,7 +379,7 @@ class MammotionBaseDevice:
             region_data.type = common_data.type
             region_data.total_frame = total_frame
             region_data.current_frame = current_frame
-            await self._send_command_with_args("get_regional_data", regional_data=region_data)
+            await self.queue_command("get_regional_data", regional_data=region_data)
 
     def _update_raw_data(self, data: bytes) -> None:
         """Update raw and model data from notifications."""
@@ -475,6 +477,10 @@ class MammotionBaseDevice:
         """Get the LubaMsg of the device."""
         return self._mower
 
+    @abstractmethod
+    async def queue_command(self, key: str,  **kwargs: any) -> bytes:
+        """Queue commands to mower."""
+
     @abstractmethod
     async def _send_command(self, key: str, retry: int | None = None) -> bytes | None:
         """Send command to device and read response."""
@@ -505,9 +511,15 @@ class MammotionBaseDevice:
 
         await self._send_command_with_args("get_hash_response", total_frame=1, current_frame=1)
 
-        await self._send_command_with_args(
-            "get_area_name_list", device_id=self._mower.device.net.toapp_wifi_iot_status.devicename
-        )
+        if self._cloud_device:
+            await self._send_command_with_args(
+                "get_area_name_list", device_id=self._cloud_device.deviceName
+            )
+        if has_field(self._mower.net.toapp_wifi_iot_status):
+            await self._send_command_with_args(
+                "get_area_name_list", device_id=self._mower.net.toapp_wifi_iot_status.devicename
+            )
+
 
         # sub_cmd 3 is job hashes??
         # sub_cmd 4 is dump location (yuka)
@@ -586,6 +598,9 @@ class MammotionBaseBLEDevice(MammotionBaseDevice):
                 130, lambda: asyncio.ensure_future(self.run_periodic_sync_task())
             )
 
+    async def queue_command(self, key: str, **kwargs: any) -> bytes | None:
+        return await self._send_command_with_args(key, **kwargs)
+
     async def _send_command_with_args(self, key: str, **kwargs) -> bytes | None:
         """Send command to device and read response."""
         if self._operation_lock.locked():
@@ -655,7 +670,7 @@ class MammotionBaseBLEDevice(MammotionBaseDevice):
     def rssi(self) -> int:
         """Return RSSI of device."""
         try:
-            return self._mower.device.sys.toapp_report_data.connect.ble_rssi
+            return self._mower.sys.toapp_report_data.connect.ble_rssi
         finally:
             return 0
 
@@ -938,9 +953,11 @@ class MammotionBaseCloudDevice(MammotionBaseDevice):
         mowing_state: MowingDevice
     ) -> None:
         """Initialize MammotionBaseCloudDevice."""
-        super().__init__(mowing_state)
+        super().__init__(mowing_state, cloud_device)
         self._ble_sync_task = None
         self.is_ready = False
+        self.command_queue = asyncio.Queue()
+        self.processing_task = asyncio.create_task(self._process_queue())
         self._mqtt_client = mqtt_client
         self.iot_id = cloud_device.iotId
         self.device = cloud_device
@@ -966,7 +983,7 @@ class MammotionBaseCloudDevice(MammotionBaseDevice):
     async def on_ready(self):
         """Callback for when MQTT is subscribed to events."""
         if self.on_ready_callback:
-            self.on_ready_callback()
+            await self.on_ready_callback()
 
         await self._ble_sync()
         await self.run_periodic_sync_task()
@@ -998,6 +1015,32 @@ class MammotionBaseCloudDevice(MammotionBaseDevice):
                 160, lambda: asyncio.ensure_future(self.run_periodic_sync_task())
             )
 
+    async def queue_command(self, key: str, **kwargs: any) -> bytes:
+        # Create a future to hold the result
+        _LOGGER.debug("Queueing command: %s", key)
+        future = asyncio.Future()
+        # Put the command in the queue as a tuple (key, command, future)
+        command_bytes = getattr(self._commands, key)(**kwargs)
+        await self.command_queue.put((key, command_bytes, future))
+        # Wait for the future to be resolved
+        return await future
+
+    async def _process_queue(self):
+        while True:
+            # Get the next item from the queue
+            key, command, future = await self.command_queue.get()
+            try:
+                # Process the command using _execute_command_locked
+                result = await self._execute_command_locked(key, command)
+                # Set the result on the future
+                future.set_result(result)
+            except Exception as ex:
+                # Set the exception on the future if something goes wrong
+                future.set_exception(ex)
+            finally:
+                # Mark the task as done
+                self.command_queue.task_done()
+
     async def _on_mqtt_message(self, topic: str, payload: str, iot_id: str) -> None:
         """Handle incoming MQTT messages."""
         _LOGGER.debug("MQTT message received on topic %s: %s, iot_id: %s", topic, payload, iot_id)

pymammotion/mqtt/mammotion_mqtt.py

@@ -57,7 +57,7 @@ class MammotionMQTT:
         ).hexdigest()
 
         self._client_id = client_id
-        self.loop = asyncio.get_event_loop()
+        self.loop = asyncio.get_running_loop()
 
         self._linkkit_client = LinkKit(
             region_id,
@@ -139,7 +139,8 @@ class MammotionMQTT:
         payload = json.loads(payload)
         iot_id = payload.get("params", {}).get("iotId", "")
         if iot_id != "" and self.on_message:
-            asyncio.run_coroutine_threadsafe(self.on_message(topic, payload, iot_id), self.loop).result()
+            future = asyncio.run_coroutine_threadsafe(self.on_message(topic, payload, iot_id), self.loop)
+            asyncio.wrap_future(future, loop=self.loop)
 
 
     def _thing_on_connect(self, session_flag, rc, user_data):

{pymammotion-0.2.8.dist-info → pymammotion-0.2.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pymammotion
-Version: 0.2.8
+Version: 0.2.9
 Summary: 
 License: GNU-3.0
 Author: Michael Arthur

{pymammotion-0.2.8.dist-info → pymammotion-0.2.9.dist-info}/RECORD

@@ -1,6 +1,6 @@
 pymammotion/__init__.py,sha256=kmnjdt3AEMejIz5JK7h1tTJj5ZriAgKwZBa3ScA4-Ao,1516
 pymammotion/aliyun/__init__.py,sha256=T1lkX7TRYiL4nqYanG4l4MImV-SlavSbuooC-W-uUGw,29
-pymammotion/aliyun/cloud_gateway.py,sha256=sOqewx_aRxlRGj9FvPYJ0VPmheSTDpH3e1mCAX-HGko,21788
+pymammotion/aliyun/cloud_gateway.py,sha256=IqmxgjDIgRoZvJvQnwYGOSl4XaCV4MoMKTlfhC9-6Ys,21830
 pymammotion/aliyun/cloud_service.py,sha256=YWcKuKK6iRWy5mTnBYgHxcCusiRGGzQt3spSf7dGDss,2183
 pymammotion/aliyun/dataclass/aep_response.py,sha256=8f6GIP58ve8gd6AL3HBoXxsy0n2q4ygWvjELGnoOnVc,452
 pymammotion/aliyun/dataclass/connect_response.py,sha256=Yz-fEbDzgGPTo5Of2oAjmFkSv08T7ze80pQU4k-gKIU,824
@@ -58,10 +58,10 @@ pymammotion/mammotion/commands/messages/video.py,sha256=_8lJsU4sLm2CGnc7RDkueA0A
 pymammotion/mammotion/control/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pymammotion/mammotion/control/joystick.py,sha256=EWV20MMzQuhbLlNlXbsyZKSEpeM7x1CQL7saU4Pn0-g,6165
 pymammotion/mammotion/devices/__init__.py,sha256=T72jt0ejtMjo1rPmn_FeMF3pmp0LLeRRpc9WcDKEYYY,126
-pymammotion/mammotion/devices/mammotion.py,sha256=akUzxCSKPVgg2mG2-UQ6kjDHcZEz9Irs6p1Xh-_C51w,45017
+pymammotion/mammotion/devices/mammotion.py,sha256=IqOPXLVDcq8VQ2mm33qHrauzkS9XAKOVFE0-TOk-q4k,46909
 pymammotion/mqtt/__init__.py,sha256=Ocs5e-HLJvTuDpVXyECEsWIvwsUaxzj7lZ9mSYutNDY,105
 pymammotion/mqtt/mammotion_future.py,sha256=WKnHqeHiS2Ut-SaDBNOxqh1jDLeTiyLTsJ7PNUexrjk,687
-pymammotion/mqtt/mammotion_mqtt.py,sha256=cBx-T_gsck8bEGaaE-oxdV1ADoPxJi8FKbSntRCSlVQ,8013
+pymammotion/mqtt/mammotion_mqtt.py,sha256=K9TokiQWYJloWu5Hom00g7cfGhWHDSWqbcU9AqW4C9Q,8071
 pymammotion/proto/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pymammotion/proto/basestation.proto,sha256=_x5gAz3FkZXS1jtq4GgZgaDCuRU-UV-7HTFdsfQ3zbo,1034
 pymammotion/proto/basestation.py,sha256=js64_N2xQYRxWPRdVNEapO0qe7vBlfYnjW5sE8hi7hw,2026
@@ -111,7 +111,7 @@ pymammotion/utility/device_type.py,sha256=KYawu2glZMVlPmxRbA4kVFujXz3miHp3rJiOWR
 pymammotion/utility/map.py,sha256=aoi-Luzuph02hKynTofMoq3mnPstanx75MDAVv49CuY,2211
 pymammotion/utility/periodic.py,sha256=9wJMfwXPlx6Mbp3Fws7LLTI34ZDKphH1bva_Ggyk32g,3281
 pymammotion/utility/rocker_util.py,sha256=syPL0QN4zMzHiTIkUKS7RXBBptjdbkfNlPddwUD5V3A,7171
-pymammotion-0.2.8.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-pymammotion-0.2.8.dist-info/METADATA,sha256=oPDS7unJxPTKajrvuT1s7MIK7c6fzKy6t-tKcLT2JP0,3968
-pymammotion-0.2.8.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-pymammotion-0.2.8.dist-info/RECORD,,
+pymammotion-0.2.9.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+pymammotion-0.2.9.dist-info/METADATA,sha256=sUQS-JHi__zyOgzpZzSQDKZpkCJ4yzl4UjNQdGE73M0,3968
+pymammotion-0.2.9.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+pymammotion-0.2.9.dist-info/RECORD,,