pymammotion 0.5.25__tar.gz → 0.5.26__tar.gz

{pymammotion-0.5.25 → pymammotion-0.5.26}/PKG-INFO

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

{pymammotion-0.5.25 → pymammotion-0.5.26}/pymammotion/data/model/device_info.py

@@ -19,18 +19,22 @@ class DeviceNonWorkingHours(DataClassORJSONMixin):
     start_time: str = ""
     end_time: str = ""
 
+
 @dataclass
 class LampInfo(DataClassORJSONMixin):
     lamp_bright: int = 0
     manual_light: bool = False
     night_light: bool = False
 
+
 @dataclass
 class MowerInfo(DataClassORJSONMixin):
     blade_status: bool = False
     rain_detection: bool = False
     traversal_mode: int = 0
     turning_mode: int = 0
+    blade_mode: int = 0
+    blade_rpm: int = 0
     side_led: SideLight = field(default_factory=SideLight)
     collector_installation_status: bool = False
     model: str = ""

{pymammotion-0.5.25 → pymammotion-0.5.26}/pymammotion/data/state_manager.py

@@ -18,6 +18,7 @@ from pymammotion.data.mqtt.status import ThingStatusMessage
 from pymammotion.event.event import DataEvent
 from pymammotion.proto import (
     AppGetAllAreaHashName,
+    AppGetCutterWorkMode,
     DeviceFwInfo,
     DeviceProductTypeInfoT,
     DrvDevInfoResp,
@@ -254,6 +255,12 @@ class StateManager:
 
     def _update_driver_data(self, message) -> None:
         """Update driver data."""
+        driver_msg = betterproto2.which_one_of(message.driver, "SubDrvMsg")
+        match driver_msg[0]:
+            case "current_cutter_mode":
+                cutter_work_mode: AppGetCutterWorkMode = driver_msg[1]
+                self._device.mower_state.cutter_mode = cutter_work_mode.current_cutter_mode
+                self._device.mower_state.cutter_rpm = cutter_work_mode.current_cutter_rpm
 
     def _update_net_data(self, message) -> None:
         """Update network data."""
@@ -276,7 +283,7 @@ class StateManager:
         """Media and video states."""
         mul_msg = betterproto2.which_one_of(message.mul, "SubMul")
         match mul_msg[0]:
-            case "Getlamprsp":
+            case "get_lamp_rsp":
                 lamp_resp: Getlamprsp = mul_msg[1]
                 self._device.mower_state.lamp_info.lamp_bright = lamp_resp.lamp_bright
                 if lamp_resp.get_ids in (1126, 1127):

{pymammotion-0.5.25 → pymammotion-0.5.26}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [project]
 name        = "pymammotion"
-version     = "0.5.25"
+version     = "0.5.26"
 license     = "GPL-3.0"
 description = ""
 readme      = "README.md"
@@ -66,7 +66,7 @@ autotyping = "^24.3.0"
 ipython = "^9.0.2"
 
 [tool.bumpver]
-current_version = "0.5.25"
+current_version = "0.5.26"
 version_pattern = "MAJOR.MINOR.PATCH"
 commit_message  = "Bump version {old_version} -> {new_version}"
 commit          = true