pymammotion 0.5.22__py3-none-any.whl → 0.5.24__py3-none-any.whl

pymammotion/aliyun/client.py

@@ -81,7 +81,7 @@ class Client:
                 "maxAttempts": UtilClient.default_number(runtime.max_attempts, 3),
             },
             "backoff": {
-                "policy": UtilClient.default_string(runtime.backoff_policy, "no"),
+                "policy": UtilClient.default_string(runtime.backoff_policy, "yes"),
                 "period": UtilClient.default_number(runtime.backoff_period, 1),
             },
             "ignoreSSL": runtime.ignore_ssl,
@@ -171,7 +171,7 @@ class Client:
                 "maxAttempts": UtilClient.default_number(runtime.max_attempts, 3),
             },
             "backoff": {
-                "policy": UtilClient.default_string(runtime.backoff_policy, "no"),
+                "policy": UtilClient.default_string(runtime.backoff_policy, "yes"),
                 "period": UtilClient.default_number(runtime.backoff_period, 1),
             },
             "ignoreSSL": runtime.ignore_ssl,

pymammotion/aliyun/cloud_gateway.py

@@ -1,5 +1,6 @@
 """Module for interacting with Aliyun Cloud IoT Gateway."""
 
+import asyncio
 import base64
 import hashlib
 import hmac
@@ -86,6 +87,14 @@ class GatewayTimeoutException(Exception):
         self.iot_id = args[1]
 
 
+class TooManyRequestsException(Exception):
+    """Raise exception when the gateway times out."""
+
+    def __init__(self, *args: object) -> None:
+        super().__init__(args)
+        self.iot_id = args[1]
+
+
 class LoginException(Exception):
     """Raise exception when library cannot log in."""
 
@@ -118,7 +127,7 @@ class CloudIOTGateway:
         self._app_key = APP_KEY
         self._app_secret = APP_SECRET
         self.domain = ALIYUN_DOMAIN
-
+        self.message_delay = 1
         self._client_id = self.generate_hardware_string(8)  # 8 characters
         self._device_sn = self.generate_hardware_string(32)  # 32 characters
         self._utdid = self.generate_hardware_string(32)  # 32 characters
@@ -730,7 +739,6 @@ class CloudIOTGateway:
         )
 
         client = Client(config)
-
         # build request
         request = CommonParams(
             api_ver="1.0.5",
@@ -762,6 +770,14 @@ class CloudIOTGateway:
         logger.debug(response.body)
         logger.debug(iot_id)
 
+        if response.status_code == 429:
+            logger.debug("too many requests.")
+            if self.message_delay > 8:
+                raise TooManyRequestsException(response.status_message, iot_id)
+            asyncio.get_event_loop().call_later(self.message_delay, self.send_cloud_command, iot_id, command)
+            self.message_delay = self.message_delay * 2
+            return message_id
+
         response_body_str = response.body.decode("utf-8")
         response_body_dict = self.parse_json_response(response_body_str)
 
@@ -785,6 +801,9 @@ class CloudIOTGateway:
             if response_body_dict.get("code") == 6205:
                 raise DeviceOfflineException(response_body_dict.get("code"), iot_id)
 
+        if self.message_delay != 1:
+            self.message_delay = 1
+
         return message_id
 
     async def get_device_properties(self, iot_id: str) -> ThingPropertiesResponse:

pymammotion/data/state_manager.py

@@ -1,8 +1,9 @@
 """Manage state from notifications into MowingDevice."""
 
+from collections.abc import Awaitable, Callable
 from datetime import UTC, datetime
 import logging
-from typing import Any, Awaitable, Callable
+from typing import Any
 
 import betterproto2
 
@@ -42,6 +43,7 @@ class StateManager:
     """Manage state."""
 
     def __init__(self, device: MowingDevice) -> None:
+        """Initialize state manager with a device."""
         self._device: MowingDevice = device
         self.last_updated_at = datetime.now(UTC)
         self.preference = ConnectionPreference.WIFI
@@ -74,11 +76,13 @@ class StateManager:
         self._device = device
 
     async def properties(self, thing_properties: ThingPropertiesMessage) -> None:
+        """Update device properties and invoke callback."""
         # TODO update device based off thing properties
         self._device.mqtt_properties = thing_properties
         await self.on_properties_callback(thing_properties)
 
     async def status(self, thing_status: ThingStatusMessage) -> None:
+        """Update device status and invoke callback."""
         if not self._device.online:
             self._device.online = True
         self._device.status_properties = thing_status
@@ -93,19 +97,23 @@ class StateManager:
 
     @property
     def online(self) -> bool:
+        """Return online status."""
         return self._device.online
 
     @online.setter
     def online(self, value: bool) -> None:
+        """Set online status."""
         self._device.online = value
 
     async def gethash_ack_callback(self, msg: NavGetHashListAck) -> None:
+        """Dispatch hash list acknowledgment to available callback."""
         if self.cloud_gethash_ack_callback:
             await self.cloud_gethash_ack_callback(msg)
         elif self.ble_gethash_ack_callback:
             await self.ble_gethash_ack_callback(msg)
 
     async def on_notification_callback(self, res: tuple[str, Any | None]) -> None:
+        """Dispatch notification to available callback."""
         if self.cloud_on_notification_callback:
             await self.cloud_on_notification_callback.data_event(res)
         elif self.ble_on_notification_callback:
@@ -134,6 +142,7 @@ class StateManager:
             await self.ble_get_commondata_ack_callback(comm_data)
 
     async def get_plan_callback(self, planjob: NavPlanJobSet) -> None:
+        """Dispatch plan job to available callback."""
         if self.cloud_get_plan_callback:
             await self.cloud_get_plan_callback(planjob)
         elif self.ble_get_plan_callback:
@@ -244,9 +253,10 @@ class StateManager:
                 self._device.mower_state.swversion = device_fw_info.version
 
     def _update_driver_data(self, message) -> None:
-        pass
+        """Update driver data."""
 
     def _update_net_data(self, message) -> None:
+        """Update network data."""
         net_msg = betterproto2.which_one_of(message.net, "NetSubType")
         match net_msg[0]:
             case "toapp_wifi_iot_status":
@@ -264,15 +274,21 @@ class StateManager:
 
     def _update_mul_data(self, message) -> None:
         """Media and video states."""
-        mul_msg = betterproto2.which_one_of(message.net, "SubMul")
+        mul_msg = betterproto2.which_one_of(message.mul, "SubMul")
         match mul_msg[0]:
             case "Getlamprsp":
                 lamp_resp: Getlamprsp = mul_msg[1]
                 self._device.mower_state.lamp_info.lamp_bright = lamp_resp.lamp_bright
-                if lamp_resp.get_ids == 1126:
-                    self._device.mower_state.lamp_info.manual_light = bool(lamp_resp.lamp_manual_ctrl.value)
+                if lamp_resp.get_ids in (1126, 1127):
+                    self._device.mower_state.lamp_info.lamp_bright = lamp_resp.lamp_bright
+                    self._device.mower_state.lamp_info.manual_light = bool(lamp_resp.lamp_manual_ctrl.value) or bool(
+                        lamp_resp.lamp_bright
+                    )
                 if lamp_resp.get_ids == 1123:
-                    self._device.mower_state.lamp_info.night_light = bool(lamp_resp.lamp_ctrl.value)
+                    self._device.mower_state.lamp_info.lamp_bright = lamp_resp.lamp_bright
+                    self._device.mower_state.lamp_info.night_light = bool(lamp_resp.lamp_ctrl.value) or bool(
+                        lamp_resp.lamp_bright
+                    )
 
     def _update_ota_data(self, message) -> None:
-        pass
+        """Update OTA data."""

pymammotion/mammotion/commands/messages/driver.py

@@ -5,6 +5,8 @@ import time
 
 from pymammotion.mammotion.commands.abstract_message import AbstractMessage
 from pymammotion.proto import (
+    AppGetCutterWorkMode,
+    AppSetCutterWorkMode,
     DrvKnifeHeight,
     DrvMotionCtrl,
     DrvMowCtrlByHand,
@@ -23,6 +25,7 @@ logger = getLogger(__name__)
 
 class MessageDriver(AbstractMessage, ABC):
     def send_order_msg_driver(self, driver) -> bytes:
+        """Build and serialize a driver command message."""
         return LubaMsg(
             msgtype=MsgCmdType.EMBED_DRIVER,
             sender=MsgDevice.DEV_MOBILEAPP,
@@ -36,23 +39,42 @@ class MessageDriver(AbstractMessage, ABC):
         ).SerializeToString()
 
     def set_blade_height(self, height: int) -> bytes:
+        """Set mower blade height."""
         logger.debug(f"Send knife height height={height}")
         build = MctlDriver(todev_knife_height_set=DrvKnifeHeight(knife_height=height))
         logger.debug(f"Send command--Knife motor height setting height={height}")
         return self.send_order_msg_driver(build)
 
     def set_speed(self, speed: float) -> bytes:
+        """Set the device speed."""
         logger.debug(f"{self.get_device_name()} set speed, {speed}")
         build = MctlDriver(bidire_speed_read_set=DrvSrSpeed(speed=speed, rw=1))
         logger.debug(f"Send command--Speed setting speed={speed}")
         return self.send_order_msg_driver(build)
 
+    def get_cutter_mode(self) -> bytes:
+        """Request the current cutter mode."""
+        build = MctlDriver(current_cutter_mode=AppGetCutterWorkMode())
+        return self.send_order_msg_driver(build)
+
+    def set_cutter_mode(self, cutter_mode: int) -> bytes:
+        """Set blade speed."""
+        """
+        1 slow
+        0 normal
+        2 fast
+        """
+        build = MctlDriver(cutter_mode_ctrl_by_hand=AppSetCutterWorkMode(cutter_mode=cutter_mode))
+        return self.send_order_msg_driver(build)
+
     def syn_nav_star_point_data(self, sat_system: int) -> bytes:
+        """Synchronize navigation satellite frequency points."""
         build = MctlDriver(rtk_sys_mask_query=RtkSysMaskQueryT(sat_system=sat_system))
         logger.debug(f"Send command--Navigation satellite frequency point synchronization={sat_system}")
         return self.send_order_msg_driver(build)
 
     def set_nav_star_point(self, cmd_req: str) -> bytes:
+        """Configure navigation satellite frequency points."""
         build = MctlDriver(rtk_cfg_req=RtkCfgReqT(cmd_req=cmd_req, cmd_length=len(cmd_req) - 1))
         logger.debug(f"Send command--Navigation satellite frequency point setting={cmd_req}")
         logger.debug(
@@ -61,6 +83,7 @@ class MessageDriver(AbstractMessage, ABC):
         return self.send_order_msg_driver(build)
 
     def get_speed(self) -> bytes:
+        """Request the current speed value."""
         build = MctlDriver(bidire_speed_read_set=DrvSrSpeed(rw=0))
         logger.debug("Send command--Get speed value")
         return self.send_order_msg_driver(build)
@@ -72,6 +95,7 @@ class MessageDriver(AbstractMessage, ABC):
         cut_knife_height: int,
         max_run_speed: float,
     ) -> bytes:
+        """Send manual mowing control command."""
         build = MctlDriver(
             mow_ctrl_by_hand=DrvMowCtrlByHand(
                 main_ctrl=main_ctrl,
@@ -88,6 +112,7 @@ class MessageDriver(AbstractMessage, ABC):
         return self.send_order_msg_driver(build)
 
     def send_movement(self, linear_speed: int, angular_speed: int) -> bytes:
+        """Send motion command with linear and angular speeds."""
         logger.debug(f"Control command print, linearSpeed={
         linear_speed} // angularSpeed={angular_speed}")
         return self.send_order_msg_driver(

{pymammotion-0.5.22.dist-info → pymammotion-0.5.24.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: pymammotion
-Version: 0.5.22
+Version: 0.5.24
 Summary: 
 License: GPL-3.0
 Author: Michael Arthur

{pymammotion-0.5.22.dist-info → pymammotion-0.5.24.dist-info}/RECORD

@@ -1,7 +1,7 @@
 pymammotion/__init__.py,sha256=H-U94bNLp0LPC6hkRopVEUlUsZSR97n7WfKGPjK1GMg,1638
 pymammotion/aliyun/__init__.py,sha256=T1lkX7TRYiL4nqYanG4l4MImV-SlavSbuooC-W-uUGw,29
-pymammotion/aliyun/client.py,sha256=GCpAnLOVWW_W0c8UvdRpWVzZH5oLto8JZnb82stJi-8,9662
-pymammotion/aliyun/cloud_gateway.py,sha256=Qhzs3UtkAnVQljf4qtFiOdd4bz04oHzJT-ZPvnM2CdQ,31250
+pymammotion/aliyun/client.py,sha256=z-dCR6cD5-8MuxjFaXC5U8IsiKcsGQ4EzzexG7X78B8,9664
+pymammotion/aliyun/cloud_gateway.py,sha256=rnOV5v6V1jIbZ8nYXTwvJTEfbAIrmbNBPG8eIbqVyZ0,31977
 pymammotion/aliyun/model/aep_response.py,sha256=EY4uMTJ4F9rvbcXnAOc5YKi7q__9kIVgfDwfyr65Gk0,421
 pymammotion/aliyun/model/connect_response.py,sha256=Yz-fEbDzgGPTo5Of2oAjmFkSv08T7ze80pQU4k-gKIU,824
 pymammotion/aliyun/model/dev_by_account_response.py,sha256=P9yYy4Z2tLkJSqXA_5XGaCUliSSVa5ILl7VoMtL_tCA,977
@@ -47,7 +47,7 @@ pymammotion/data/mqtt/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdr
 pymammotion/data/mqtt/event.py,sha256=pj65y93ACcTzUIJkHZC6s_pRFBYaNMSU1wyYRhiGvw4,5571
 pymammotion/data/mqtt/properties.py,sha256=laud9rE-JqBHWKZ44Dov2yMkp3Ld8ghpBwlZ4_3g9uQ,4321
 pymammotion/data/mqtt/status.py,sha256=qwZPevIzScePA_25nbRWMN7IhU-MhhtIl7fcZKAJfIc,1102
-pymammotion/data/state_manager.py,sha256=3EXOul7lorlKSnaSWqswBEXIlY9KC2D33VGNj_BBybE,12696
+pymammotion/data/state_manager.py,sha256=i9TSTGQeDtUCgUWuunQI_tbm76x0F5L8CkCZ0E_ixms,13565
 pymammotion/event/__init__.py,sha256=mgATR6vPHACNQ-0zH5fi7NdzeTCDV1CZyaWPmtUusi8,115
 pymammotion/event/event.py,sha256=Z8WYxv_-5khEqKjL1w4c_Et24G1Kdm8QFuIBylD3h3U,3021
 pymammotion/http/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -65,7 +65,7 @@ pymammotion/mammotion/commands/abstract_message.py,sha256=M4qL-Yw5g3fTgS9jB9LUSg
 pymammotion/mammotion/commands/mammotion_command.py,sha256=AMOx6nEbPa0HijbENomPLLuJummS1q-23BzgetBAuOI,2991
 pymammotion/mammotion/commands/messages/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pymammotion/mammotion/commands/messages/basestation.py,sha256=a1LEF4C25ynILeQLx-FOOU4d-N_vMkXSh_etojJjIDM,1442
-pymammotion/mammotion/commands/messages/driver.py,sha256=LAYGvAcDDXwwliQ1OfRwNBk1VpIdpURo1TlA64zpLA4,3701
+pymammotion/mammotion/commands/messages/driver.py,sha256=tJUM1WhlWFs92PbZW5QSG-TMo6nemAUkcUDzI_6gETU,4695
 pymammotion/mammotion/commands/messages/media.py,sha256=KJyMzSZkwQrHWjAWfAI88Y7-jmfN3-WYtY193KaomT4,2930
 pymammotion/mammotion/commands/messages/navigation.py,sha256=yI-B5HHGuZJXn9iYrxhYZ5BArde6wB_j9bUc5IcSIJE,23261
 pymammotion/mammotion/commands/messages/network.py,sha256=7K6aqt29ymTSUG0iEv4kIRD0Dv6rfHNctE00UvuMpG4,7264
@@ -130,7 +130,7 @@ pymammotion/utility/movement.py,sha256=N75oAoAgFydqoaOedYIxGUHmuTCtPzAOtb-d_29tp
 pymammotion/utility/mur_mur_hash.py,sha256=xEfOZVbqRawJj66eLgtnZ85OauDR47oIPr29OHelzPI,4468
 pymammotion/utility/periodic.py,sha256=MbeSb9cfhxzYmdT_RiE0dZe3H9IfbQW_zSqhmSX2RUc,3321
 pymammotion/utility/rocker_util.py,sha256=6tX7sS87qoQC_tsxbx3NLL-HgS08wtzXiZkhDiz7uo0,7179
-pymammotion-0.5.22.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-pymammotion-0.5.22.dist-info/METADATA,sha256=Kk3wTPnCon-TxjgqtRIJvn8XXBu4YBcVxcuv2HwfY3E,3872
-pymammotion-0.5.22.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-pymammotion-0.5.22.dist-info/RECORD,,
+pymammotion-0.5.24.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+pymammotion-0.5.24.dist-info/METADATA,sha256=Wj7OG5shjN_rQ6QoBC3LgH-wnAr0PsLIjEhj6uQ5Sb4,3872
+pymammotion-0.5.24.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+pymammotion-0.5.24.dist-info/RECORD,,