pymammotion 0.4.0a9__py3-none-any.whl → 0.4.0b1__py3-none-any.whl

pymammotion/data/state_manager.py

@@ -21,127 +21,153 @@ logger = logging.getLogger(__name__)
 
 
 class StateManager:
-    """Manage state."""
-
-    _device: MowingDevice
-    last_updated_at: datetime = datetime.now()
-
-    def __init__(self, device: MowingDevice) -> None:
-        self._device = device
-        self.gethash_ack_callback: Callable[[NavGetHashListAck], Awaitable[None]] | None = None
-        self.get_commondata_ack_callback: Callable[[NavGetCommDataAck | SvgMessageAckT], Awaitable[None]] | None = None
-        self.on_notification_callback: Callable[[tuple[str, Any | None]], Awaitable[None]] | None = None
-        self.queue_command_callback: Callable[[str, dict[str, Any]], Awaitable[bytes]] | None = None
-        self.last_updated_at = datetime.now()
-
-    def get_device(self) -> MowingDevice:
-        """Get device."""
-        return self._device
-
-    def set_device(self, device: MowingDevice) -> None:
-        """Set device."""
-        self._device = device
-
-    async def properties(self, properties: ThingPropertiesMessage) -> None:
-        self._device.mqtt_properties = properties
-
-    async def status(self, status: ThingStatusMessage) -> None:
-        if not self._device.online:
-            self._device.online = True
-        self._device.status_properties = status
-
-    @property
-    def online(self) -> bool:
-        return self._device.online
-
-    @online.setter
-    def online(self, value: bool) -> None:
-        self._device.online = value
-
-    async def notification(self, message: LubaMsg) -> None:
-        """Handle protobuf notifications."""
-        res = betterproto.which_one_of(message, "LubaSubMsg")
-        self.last_updated_at = datetime.now()
-
-        match res[0]:
-            case "nav":
-                await self._update_nav_data(message)
-            case "sys":
-                await self._update_sys_data(message)
-            case "driver":
-                self._update_driver_data(message)
-            case "net":
-                self._update_net_data(message)
-            case "mul":
-                self._update_mul_data(message)
-            case "ota":
-                self._update_ota_data(message)
-
-        if self.on_notification_callback:
-            await self.on_notification_callback(res)
-
-    async def _update_nav_data(self, message) -> None:
-        """Update nav data."""
-        nav_msg = betterproto.which_one_of(message.nav, "SubNavMsg")
-        match nav_msg[0]:
-            case "toapp_gethash_ack":
-                hashlist_ack: NavGetHashListAck = nav_msg[1]
-                self._device.map.update_root_hash_list(hashlist_ack)
-                await self.gethash_ack_callback(nav_msg[1])
-            case "toapp_get_commondata_ack":
-                common_data: NavGetCommDataAck = nav_msg[1]
-                updated = self._device.map.update(common_data)
-                if updated:
-                    await self.get_commondata_ack_callback(common_data)
-            case "toapp_svg_msg":
-                common_data: SvgMessageAckT = nav_msg[1]
-                updated = self._device.map.update(common_data)
-                if updated:
-                    await self.get_commondata_ack_callback(common_data)
-
-            case "toapp_all_hash_name":
-                hash_names: AppGetAllAreaHashName = nav_msg[1]
-                converted_list = [AreaHashNameList(name=item.name, hash=item.hash) for item in hash_names.hashnames]
-                self._device.map.area_name = converted_list
-
-    async def _update_sys_data(self, message) -> None:
-        """Update system."""
-        sys_msg = betterproto.which_one_of(message.sys, "SubSysMsg")
-        match sys_msg[0]:
-            case "system_update_buf":
-                self._device.buffer(sys_msg[1])
-            case "toapp_report_data":
-                self._device.update_report_data(sys_msg[1])
-            case "mow_to_app_info":
-                self._device.mow_info(sys_msg[1])
-            case "system_tard_state_tunnel":
-                self._device.run_state_update(sys_msg[1])
-            case "todev_time_ctrl_light":
-                ctrl_light: TimeCtrlLight = sys_msg[1]
-                side_led: SideLight = SideLight.from_dict(ctrl_light.to_dict(casing=betterproto.Casing.SNAKE))
-                self._device.mower_state.side_led = side_led
-            case "device_product_type_info":
-                device_product_type: DeviceProductTypeInfoT = sys_msg[1]
-                self._device.mower_state.model_id = device_product_type.main_product_type
-
-    def _update_driver_data(self, message) -> None:
-        pass
-
-    def _update_net_data(self, message) -> None:
-        net_msg = betterproto.which_one_of(message.net, "NetSubType")
-        match net_msg[0]:
-            case "toapp_wifi_iot_status":
-                wifi_iot_status: WifiIotStatusReport = net_msg[1]
-                self._device.mower_state.product_key = wifi_iot_status.productkey
-            case "toapp_devinfo_resp":
-                toapp_devinfo_resp: DrvDevInfoResp = net_msg[1]
-                for resp in toapp_devinfo_resp.resp_ids:
-                    if resp.res == "DRV_RESULT_SUC":
-                        self._device.mower_state.swversion = resp.info
-                        self._device.device_firmwares.device_version = resp.info
-
-    def _update_mul_data(self, message) -> None:
-        pass
-
-    def _update_ota_data(self, message) -> None:
-        pass
+"""Manage state."""
+
+_device: MowingDevice
+last_updated_at: datetime = datetime.now()
+
+def __init__(self, device: MowingDevice) -> None:
+    self._device = device
+    self.cloud_gethash_ack_callback: Callable[[NavGetHashListAck], Awaitable[None]] | None = None
+    self.cloud_get_commondata_ack_callback: Callable[[NavGetCommDataAck | SvgMessageAckT], Awaitable[None]] | None = None
+    self.cloud_on_notification_callback: Callable[[tuple[str, Any | None]], Awaitable[None]] | None = None
+
+    # possibly don't need anymore
+    self.cloud_queue_command_callback: Callable[[str, dict[str, Any]], Awaitable[bytes]] | None = None
+
+    self.ble_gethash_ack_callback: Callable[[NavGetHashListAck], Awaitable[None]] | None = None
+    self.ble_get_commondata_ack_callback: Callable[[NavGetCommDataAck | SvgMessageAckT], Awaitable[None]] | None = None
+    self.ble_on_notification_callback: Callable[[tuple[str, Any | None]], Awaitable[None]] | None = None
+
+    # possibly don't need anymore
+    self.ble_queue_command_callback: Callable[[str, dict[str, Any]], Awaitable[bytes]] | None = None
+    self.last_updated_at = datetime.now()
+
+def get_device(self) -> MowingDevice:
+    """Get device."""
+    return self._device
+
+def set_device(self, device: MowingDevice) -> None:
+    """Set device."""
+    self._device = device
+
+async def properties(self, properties: ThingPropertiesMessage) -> None:
+    self._device.mqtt_properties = properties
+
+async def status(self, status: ThingStatusMessage) -> None:
+    if not self._device.online:
+        self._device.online = True
+    self._device.status_properties = status
+
+@property
+def online(self) -> bool:
+    return self._device.online
+
+@online.setter
+def online(self, value: bool) -> None:
+    self._device.online = value
+
+async def gethash_ack_callback(self, msg: NavGetHashListAck):
+    if self.cloud_gethash_ack_callback:
+        await self.cloud_gethash_ack_callback(msg)
+    if self.ble_gethash_ack_callback:
+        await self.ble_gethash_ack_callback(msg)
+
+async def on_notification_callback(self, res: tuple[str, Any | None]):
+    if self.cloud_on_notification_callback:
+        await self.cloud_on_notification_callback(res)
+    if self.ble_on_notification_callback:
+        await self.ble_on_notification_callback(res)
+
+async def get_commondata_ack_callback(self, comm_data: NavGetCommDataAck | SvgMessageAckT):
+    if self.cloud_get_commondata_ack_callback:
+        await self.cloud_get_commondata_ack_callback(comm_data)
+    if self.ble_get_commondata_ack_callback:
+        await self.ble_get_commondata_ack_callback(comm_data)
+
+async def notification(self, message: LubaMsg) -> None:
+    """Handle protobuf notifications."""
+    res = betterproto.which_one_of(message, "LubaSubMsg")
+    self.last_updated_at = datetime.now()
+
+    match res[0]:
+        case "nav":
+            await self._update_nav_data(message)
+        case "sys":
+            await self._update_sys_data(message)
+        case "driver":
+            self._update_driver_data(message)
+        case "net":
+            self._update_net_data(message)
+        case "mul":
+            self._update_mul_data(message)
+        case "ota":
+            self._update_ota_data(message)
+
+    await self.on_notification_callback(res)
+
+async def _update_nav_data(self, message) -> None:
+    """Update nav data."""
+    nav_msg = betterproto.which_one_of(message.nav, "SubNavMsg")
+    match nav_msg[0]:
+        case "toapp_gethash_ack":
+            hashlist_ack: NavGetHashListAck = nav_msg[1]
+            self._device.map.update_root_hash_list(hashlist_ack)
+            await self.gethash_ack_callback(nav_msg[1])
+        case "toapp_get_commondata_ack":
+            common_data: NavGetCommDataAck = nav_msg[1]
+            updated = self._device.map.update(common_data)
+            if updated:
+                await self.get_commondata_ack_callback(common_data)
+        case "toapp_svg_msg":
+            common_data: SvgMessageAckT = nav_msg[1]
+            updated = self._device.map.update(common_data)
+            if updated:
+                await self.get_commondata_ack_callback(common_data)
+
+        case "toapp_all_hash_name":
+            hash_names: AppGetAllAreaHashName = nav_msg[1]
+            converted_list = [AreaHashNameList(name=item.name, hash=item.hash) for item in hash_names.hashnames]
+            self._device.map.area_name = converted_list
+
+async def _update_sys_data(self, message) -> None:
+    """Update system."""
+    sys_msg = betterproto.which_one_of(message.sys, "SubSysMsg")
+    match sys_msg[0]:
+        case "system_update_buf":
+            self._device.buffer(sys_msg[1])
+        case "toapp_report_data":
+            self._device.update_report_data(sys_msg[1])
+        case "mow_to_app_info":
+            self._device.mow_info(sys_msg[1])
+        case "system_tard_state_tunnel":
+            self._device.run_state_update(sys_msg[1])
+        case "todev_time_ctrl_light":
+            ctrl_light: TimeCtrlLight = sys_msg[1]
+            side_led: SideLight = SideLight.from_dict(ctrl_light.to_dict(casing=betterproto.Casing.SNAKE))
+            self._device.mower_state.side_led = side_led
+        case "device_product_type_info":
+            device_product_type: DeviceProductTypeInfoT = sys_msg[1]
+            self._device.mower_state.model_id = device_product_type.main_product_type
+
+def _update_driver_data(self, message) -> None:
+    pass
+
+def _update_net_data(self, message) -> None:
+    net_msg = betterproto.which_one_of(message.net, "NetSubType")
+    match net_msg[0]:
+        case "toapp_wifi_iot_status":
+            wifi_iot_status: WifiIotStatusReport = net_msg[1]
+            self._device.mower_state.product_key = wifi_iot_status.productkey
+        case "toapp_devinfo_resp":
+            toapp_devinfo_resp: DrvDevInfoResp = net_msg[1]
+            for resp in toapp_devinfo_resp.resp_ids:
+                if resp.res == "DRV_RESULT_SUC":
+                    self._device.mower_state.swversion = resp.info
+                    self._device.device_firmwares.device_version = resp.info
+
+def _update_mul_data(self, message) -> None:
+    pass
+
+def _update_ota_data(self, message) -> None:
+    pass

pymammotion/mammotion/commands/messages/video.py

@@ -5,6 +5,7 @@ import time
 from pymammotion.mammotion.commands.abstract_message import AbstractMessage
 from pymammotion.proto import luba_msg_pb2, luba_mul_pb2
 from pymammotion.proto.luba_msg import MsgAttr, MsgCmdType, MsgDevice
+from pymammotion.proto.luba_mul import MUL_CAMERA_POSITION
 from pymammotion.utility.device_type import DeviceType
 
 
@@ -25,10 +26,6 @@ class MessageVideo(AbstractMessage, ABC):
         return luba_msg.SerializeToString()
 
     def device_agora_join_channel_with_position(self, enter_state: int):
-        position = (
-            luba_mul_pb2.MUL_CAMERA_POSITION.ALL
-            if DeviceType.is_yuka(self.get_device_name())
-            else luba_mul_pb2.MUL_CAMERA_POSITION.LEFT
-        )
+        position = MUL_CAMERA_POSITION.ALL if DeviceType.is_yuka(self.get_device_name()) else MUL_CAMERA_POSITION.LEFT
         mctl_sys = luba_mul_pb2.SocMul(set_video=luba_mul_pb2.MulSetVideo(position=position, vi_switch=enter_state))
         return self.send_order_msg_video(mctl_sys)

pymammotion/mammotion/devices/base.py

@@ -1,6 +1,5 @@
 from abc import abstractmethod
 import asyncio
-from collections.abc import Awaitable, Callable
 import logging
 from typing import Any
 
@@ -41,17 +40,9 @@ class MammotionBaseDevice:
         self._state_manager = state_manager
         self._raw_data = dict()
         self._raw_mower_data: RawMowerData = RawMowerData()
-        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
 
-    def set_notification_callback(self, func: Callable[[tuple[str, Any | None]], Awaitable[None]]) -> None:
-        self._state_manager.on_notification_callback = func
-
-    def set_queue_callback(self, func: Callable[[str, dict[str, Any]], Awaitable[bytes]]) -> None:
-        self._state_manager.queue_command_callback = func
-
     async def datahash_response(self, hash_ack: NavGetHashListAck) -> None:
         """Handle datahash responses."""
         current_frame = hash_ack.current_frame

pymammotion/mammotion/devices/mammotion_bluetooth.py

@@ -1,4 +1,5 @@
 import asyncio
+from collections.abc import Awaitable, Callable
 import logging
 from typing import Any, cast
 from uuid import UUID
@@ -89,9 +90,17 @@ class MammotionBaseBLEDevice(MammotionBaseDevice):
         self._operation_lock = asyncio.Lock()
         self._key: str | None = None
         self.set_queue_callback(self.queue_command)
+        self._state_manager.ble_gethash_ack_callback = self.datahash_response
+        self._state_manager.ble_get_commondata_ack_callback = self.commdata_response
         loop = asyncio.get_event_loop()
         loop.create_task(self.process_queue())
 
+    def set_notification_callback(self, func: Callable[[tuple[str, Any | None]], Awaitable[None]]) -> None:
+        self._state_manager.ble_on_notification_callback = func
+
+    def set_queue_callback(self, func: Callable[[str, dict[str, Any]], Awaitable[bytes]]) -> None:
+        self._state_manager.ble_queue_command_callback = func
+
     def update_device(self, device: BLEDevice) -> None:
         """Update the BLE device."""
         self.ble_device = device

pymammotion/mammotion/devices/mammotion_cloud.py

@@ -172,6 +172,8 @@ class MammotionBaseCloudDevice(MammotionBaseDevice):
         self._mqtt.on_ready_event.add_subscribers(self.on_ready)
         self._mqtt.on_disconnected_event.add_subscribers(self.on_disconnect)
         self._mqtt.on_connected_event.add_subscribers(self.on_connect)
+        self._state_manager.cloud_gethash_ack_callback = self.datahash_response
+        self._state_manager.cloud_get_commondata_ack_callback = self.commdata_response
         self.set_queue_callback(self.queue_command)
 
         if self._mqtt.is_ready:
@@ -185,6 +187,12 @@ class MammotionBaseCloudDevice(MammotionBaseDevice):
         if self._ble_sync_task:
             self._ble_sync_task.cancel()
 
+    def set_notification_callback(self, func: Callable[[tuple[str, Any | None]], Awaitable[None]]) -> None:
+        self._state_manager.cloud_on_notification_callback = func
+
+    def set_queue_callback(self, func: Callable[[str, dict[str, Any]], Awaitable[bytes]]) -> None:
+        self._state_manager.cloud_queue_command_callback = func
+
     async def on_ready(self) -> None:
         """Callback for when MQTT is subscribed to events."""
         if self.stopped:

{pymammotion-0.4.0a9.dist-info → pymammotion-0.4.0b1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: pymammotion
-Version: 0.4.0a9
+Version: 0.4.0b1
 Summary: 
 License: GPL-3.0
 Author: Michael Arthur

{pymammotion-0.4.0a9.dist-info → pymammotion-0.4.0b1.dist-info}/RECORD

@@ -43,7 +43,7 @@ pymammotion/data/mqtt/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdr
 pymammotion/data/mqtt/event.py,sha256=r14gzZVxmlGVAwFdZQ1CUsMZFHHwRKnbt2VHnjugP28,5123
 pymammotion/data/mqtt/properties.py,sha256=pX5JRVmmpVO04CSPm5xAGcSWA_OeLd0JnBagLsfiSEc,3755
 pymammotion/data/mqtt/status.py,sha256=DuNC3JdewLPKNqNHx76_FPXRvheYSiM-CdiVTCCYY8s,1079
-pymammotion/data/state_manager.py,sha256=GrzKjBuz2CB70FGqnBdExgd3IHUheBTWH-FSR1N14nI,6140
+pymammotion/data/state_manager.py,sha256=4NLSzJIfYi2fS_Y8qRgCyAo0eMd3qpmjJ961L_5Iy4o,7009
 pymammotion/event/__init__.py,sha256=mgATR6vPHACNQ-0zH5fi7NdzeTCDV1CZyaWPmtUusi8,115
 pymammotion/event/event.py,sha256=bj2RirSIRyBs0QvkcrOtwZWUX_8F3m1sySuHVyKmZLs,2143
 pymammotion/http/_init_.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -61,14 +61,14 @@ pymammotion/mammotion/commands/messages/navigation.py,sha256=BE5h8xN2XCkcfynb5cQ
 pymammotion/mammotion/commands/messages/network.py,sha256=vUjBvuiBhRkOA4nftqTFw6rFdXNjho-ZA9Y37hL2q5Q,7623
 pymammotion/mammotion/commands/messages/ota.py,sha256=g937HT_-OQXV6A3zUiZ53b45cOX6y-rzs5m-4b0IcTk,1473
 pymammotion/mammotion/commands/messages/system.py,sha256=CYf_Ju5nMK9cFd0r5jOzEJ_w4k70PzassdhE8xLeSJU,14371
-pymammotion/mammotion/commands/messages/video.py,sha256=OtuSCy_5TQQx460etLvIu30FrjlnCug5tseicgAsNS8,1319
+pymammotion/mammotion/commands/messages/video.py,sha256=JBPyK1C9gZ6B3QcNe3IUTUnn0QWNXR4Pwj9W0qNO9Ps,1304
 pymammotion/mammotion/control/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pymammotion/mammotion/control/joystick.py,sha256=QfBVxM_gxpWsZAGO90whtgxCI2tIZ3TTad9wHIPsU9s,5640
 pymammotion/mammotion/devices/__init__.py,sha256=f2qQFPgLGmV85W2hSlMUh5BYuht9o_Ar_JEAAMD4fsE,102
-pymammotion/mammotion/devices/base.py,sha256=ZWQVOZu4R_ERXubE3kNx7-zAUxeuaAVX1VsdivlWhUg,10790
+pymammotion/mammotion/devices/base.py,sha256=evNKt1PDDGPkm4wleIEC2dtHuyDP_nHRCg35nrQg6qg,10260
 pymammotion/mammotion/devices/mammotion.py,sha256=-x9hf_M-_3fp5srHGBuMmO5DBsSDnwVGMyEiAEz5dW8,12685
-pymammotion/mammotion/devices/mammotion_bluetooth.py,sha256=HUOKnjYUWTGjksyQDCPKF_u3dWo2ddgrWCBXnJ2VkwA,18963
-pymammotion/mammotion/devices/mammotion_cloud.py,sha256=8DrVNs9QbUuH7VMs-Wi1HgHkyMLzLNIV_emnqjVSjXE,13329
+pymammotion/mammotion/devices/mammotion_bluetooth.py,sha256=tfD8TIgKVd57Hrr-totXP0-WFSRFYBuoV5Dw0EpvOi4,19509
+pymammotion/mammotion/devices/mammotion_cloud.py,sha256=nd1Uj6rF8UKHJbHC6nlPRrNQPggPXxfpydT0o0cWbMM,13835
 pymammotion/mqtt/__init__.py,sha256=Ocs5e-HLJvTuDpVXyECEsWIvwsUaxzj7lZ9mSYutNDY,105
 pymammotion/mqtt/linkkit/__init__.py,sha256=ENgc3ynd2kd9gMQR3-kgmCu6Ed9Y6XCIzU0zFReUlkk,80
 pymammotion/mqtt/linkkit/h2client.py,sha256=w9Nvi_nY4CLD_fw-pHtYChwQf7e2TiAGeqkY_sF4cf0,19659
@@ -127,7 +127,7 @@ pymammotion/utility/map.py,sha256=GYscVMg2cX3IPlNpCBNHDW0S55yS1WGRf1iHnNZ7TfQ,22
 pymammotion/utility/movement.py,sha256=N75oAoAgFydqoaOedYIxGUHmuTCtPzAOtb-d_29tpfI,615
 pymammotion/utility/periodic.py,sha256=MbeSb9cfhxzYmdT_RiE0dZe3H9IfbQW_zSqhmSX2RUc,3321
 pymammotion/utility/rocker_util.py,sha256=6tX7sS87qoQC_tsxbx3NLL-HgS08wtzXiZkhDiz7uo0,7179
-pymammotion-0.4.0a9.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-pymammotion-0.4.0a9.dist-info/METADATA,sha256=-U5a_GhF8fugM_IcnsvR7zsHyH5oX55wduoZIvKum-s,3886
-pymammotion-0.4.0a9.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-pymammotion-0.4.0a9.dist-info/RECORD,,
+pymammotion-0.4.0b1.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+pymammotion-0.4.0b1.dist-info/METADATA,sha256=QzBQE6mluKSxxyVSZan9r9lsbv9EYV57h3mw789NUdE,3886
+pymammotion-0.4.0b1.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+pymammotion-0.4.0b1.dist-info/RECORD,,