pymammotion 0.5.15__py3-none-any.whl → 0.5.17__py3-none-any.whl

pymammotion/data/model/device_info.py

@@ -19,6 +19,11 @@ 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):
@@ -35,6 +40,7 @@ class MowerInfo(DataClassORJSONMixin):
     sub_model_id: str = ""
     ble_mac: str = ""
     wifi_mac: str = ""
+    lamp_info: LampInfo = field(default_factory=LampInfo)
 
 
 @dataclass

pymammotion/data/state_manager.py

@@ -32,7 +32,7 @@ from pymammotion.proto import (
     NavUnableTimeSet,
     SvgMessageAckT,
     TimeCtrlLight,
-    WifiIotStatusReport,
+    WifiIotStatusReport, Getlamprsp,
 )
 
 logger = logging.getLogger(__name__)
@@ -263,7 +263,14 @@ class StateManager:
                 self._device.mower_state.wifi_mac = get_network_info_resp.wifi_mac
 
     def _update_mul_data(self, message) -> None:
-        pass
+        """Media and video states."""
+        mul_msg = betterproto.which_one_of(message.net, "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
+                self._device.mower_state.lamp_info.night_light = True if lamp_resp.lamp_ctrl.value > 0 else False
+                self._device.mower_state.lamp_info.manual_light = True if lamp_resp.lamp_manual_ctrl.value > 0 else False
 
     def _update_ota_data(self, message) -> None:
         pass

pymammotion/mammotion/commands/messages/basestation.py

@@ -0,0 +1,43 @@
+"""RTK protobuf commands."""
+
+from abc import ABC
+from logging import getLogger
+import time
+
+from pymammotion.mammotion.commands.abstract_message import AbstractMessage
+from pymammotion.proto import (
+    AppToBaseMqttRtkT,
+    BaseStation,
+    LubaMsg,
+    MsgAttr,
+    MsgCmdType,
+    MsgDevice,
+    RequestBasestationInfoT,
+)
+
+logger = getLogger(__name__)
+
+
+class MessageBasestation(AbstractMessage, ABC):
+    def send_order_msg_basestation(self, driver) -> bytes:
+        return LubaMsg(
+            msgtype=MsgCmdType.BASESTATION,
+            sender=MsgDevice.DEV_MOBILEAPP,
+            rcver=self.get_msg_device(MsgCmdType.BASESTATION, MsgDevice.DEV_MAINCTL),
+            msgattr=MsgAttr.REQ,
+            timestamp=round(time.time() * 1000),
+            seqs=self.seqs.increment_and_get() & 255,
+            version=1,
+            subtype=self.user_account,
+            driver=driver,
+        ).SerializeToString()
+
+    def basestation_info(self) -> bytes:
+        """Build and send a request to get basestation info (request_type=1)."""
+        base = BaseStation(to_dev=RequestBasestationInfoT(request_type=1))
+        return self.send_order_msg_basestation(base)
+
+    def set_base_net_rtk_switch(self, rtk_switch: int) -> bytes:
+        """Set RTK switch via app_to_base_mqtt_rtk_t."""
+        base = BaseStation(app_to_base_mqtt_rtk_msg=AppToBaseMqttRtkT(rtk_switch=rtk_switch))
+        return self.send_order_msg_basestation(base)

pymammotion/mammotion/commands/messages/media.py

@@ -81,7 +81,7 @@ class MessageMedia(AbstractMessage, ABC):
         manual off: false, id=1127, power_ctrl=2
         """
         ids = 1125 if manual_ctrl else 1127
-        manual_light_ctrl = LampManualCtrlSta.manual_power_off if manual_ctrl else LampManualCtrlSta.manual_power_on
+        manual_light_ctrl = LampManualCtrlSta.manual_power_on if manual_ctrl else LampManualCtrlSta.manual_power_off
         return self.send_order_msg_media(
             SocMul(set_lamp=SetHeadlamp(set_ids=ids, lamp_power_ctrl=2, lamp_manual_ctrl=manual_light_ctrl))
         )

pymammotion/proto/__init__.py

@@ -235,6 +235,7 @@ class MsgCmdType(betterproto.Enum):
     ESP = 248
     MUL = 249
     PEPT = 250
+    BASESTATION = 251
 
 
 

{pymammotion-0.5.15.dist-info → pymammotion-0.5.17.dist-info}/METADATA

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

{pymammotion-0.5.15.dist-info → pymammotion-0.5.17.dist-info}/RECORD

@@ -28,7 +28,7 @@ pymammotion/data/model/__init__.py,sha256=UVRbSXGOjYnWv30ZEvzT5QRpdVqAbyeToo-t0Q
 pymammotion/data/model/account.py,sha256=vJM-KTf2q6eBfVC-UlNHBSmJvqHiCawZ40vnuhXhaz8,140
 pymammotion/data/model/device.py,sha256=4e5q3slI7fjpizr0DsZIFcJoiCnQtRv0ty_dliDScH8,8288
 pymammotion/data/model/device_config.py,sha256=DTnoePsMwJrKVXqGeKvbZQjs8Zab61U5T3b-c-w0WlM,2654
-pymammotion/data/model/device_info.py,sha256=Q5PbPdTfd948ZWc8EHjN5zwU6pha5MCr47ZEXzCYKNY,1148
+pymammotion/data/model/device_info.py,sha256=Et7WtUnoDEUOGKvHSoPyJJHOWKoTYnKMsvak36Of_1w,1341
 pymammotion/data/model/device_limits.py,sha256=m8HdxD-RaAkPm7jHYb9GLxMEH9IfzBPz0ZypmsLnId4,1946
 pymammotion/data/model/enums.py,sha256=NLImBbeqwbwBLbABC_NbBzk4M8OxhUTcto6QcGZVNOc,1707
 pymammotion/data/model/errors.py,sha256=lBHq2cE8P5fc6Q4JXgrkJXzFKTWgxsoPOyMlTaJWbWk,396
@@ -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=AyrXwcU7SxliQ1r8b_QqFMkMoS6l0VE-XfxLq7NZ5go,12107
+pymammotion/data/state_manager.py,sha256=ChT4k_Mjags7z3EMk4XWd1MZeL3Cvep_1aTdJpvWJFo,12641
 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
@@ -64,8 +64,9 @@ pymammotion/mammotion/commands/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5
 pymammotion/mammotion/commands/abstract_message.py,sha256=M4qL-Yw5g3fTgS9jB9LUSgJSAqpUf7MME-2mfv471Sk,808
 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/media.py,sha256=X4LQG9SFNZXyqeaedjuEM10Jx9Bc6SN7kQFpKxrSFrQ,2934
+pymammotion/mammotion/commands/messages/media.py,sha256=ZKATEvqSgjTGTPrVGjDWCAnI1aq_d4TMXyl4dFzgjhE,2934
 pymammotion/mammotion/commands/messages/navigation.py,sha256=yI-B5HHGuZJXn9iYrxhYZ5BArde6wB_j9bUc5IcSIJE,23261
 pymammotion/mammotion/commands/messages/network.py,sha256=OrmW4-BP_o3piBjwXZ12WY0HqbXHQqeDXC0YaDPeBM8,7409
 pymammotion/mammotion/commands/messages/ota.py,sha256=Nk3Tlp6n7hkbkuy355BlqbozHIHzsYnrH2cDO8QGaLk,1446
@@ -84,7 +85,7 @@ pymammotion/mqtt/linkkit/h2client.py,sha256=w9Nvi_nY4CLD_fw-pHtYChwQf7e2TiAGeqkY
 pymammotion/mqtt/linkkit/linkkit.py,sha256=SGWny2_LKa8qg8RD0Bgd3a4S_fMqpYxJ7a9eFRvgcN0,132923
 pymammotion/mqtt/mammotion_future.py,sha256=_OWqKOlUGl2yT1xOsXFQYpGd-1zQ63OxqXgy7KRQgYc,710
 pymammotion/mqtt/mammotion_mqtt.py,sha256=OtLyTxow8xJc1RK1tv3ZCUBleTLZ3QABKmVYIcynoxU,10208
-pymammotion/proto/__init__.py,sha256=dUl6HZ0NMRQUVOWe5Kf3fOJFVC5YJ98I9xEfd1R9Wdg,73487
+pymammotion/proto/__init__.py,sha256=9WLHpBnpdJNWHBstuDpegq_O9KKOtxfSc0AfSd9hfmg,73509
 pymammotion/proto/basestation.proto,sha256=YiSsDmT0DY_ep6DHay13SbhxGMhentYt0BmJrVQrwLQ,1198
 pymammotion/proto/basestation_pb2.py,sha256=suenbpMz9JZCXdGJGdiPaapppRz9Cf4IDzAXUfdIG3w,3083
 pymammotion/proto/basestation_pb2.pyi,sha256=lGcTPlGaLdHEQ1A39YITbMG_vs1cVaqHAg5TsPvzexc,4652
@@ -127,7 +128,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.15.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-pymammotion-0.5.15.dist-info/METADATA,sha256=LsO3ACCgRJjuuc8XSeSKCvphDUOilr5f9jCjKoAeuro,3871
-pymammotion-0.5.15.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-pymammotion-0.5.15.dist-info/RECORD,,
+pymammotion-0.5.17.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+pymammotion-0.5.17.dist-info/METADATA,sha256=xoN2mPqAqKJRmPYuPKqdLtcK7lx6ZMbDZBH-2cYxOmU,3871
+pymammotion-0.5.17.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+pymammotion-0.5.17.dist-info/RECORD,,