Enable trace reporting from the bot.

Author: nanomad

deebot_client/commands/xml/__init__.py

@@ -14,7 +14,7 @@
 from .error import GetError
 from .fan_speed import GetCleanSpeed, SetCleanSpeed
 from .life_span import GetLifeSpan
-from .map import GetMapM, GetMapSet, GetMapSt, PullM, PullMP
+from .map import GetMapM, GetMapSet, GetMapSt, GetTrM, PullM, PullMP
 from .play_sound import PlaySound
 from .pos import GetPos
 from .stats import GetCleanSum
@@ -39,6 +39,7 @@
     "GetMapSet",
     "GetMapSt",
     "GetPos",
+    "GetTrM",
     "GetWaterBoxInfo",
     "GetWaterPermeability",
     "PlaySound",
@@ -65,6 +66,7 @@
     GetMapSet,
     GetMapSt,
     GetPos,
+    GetTrM,
     GetWaterBoxInfo,
     GetWaterPermeability,
     PlaySound,

deebot_client/commands/xml/map.py

@@ -260,3 +260,22 @@ def _handle_response(
             return CommandResult(result.state, result.args)
 
         return result
+
+
+class GetTrM(XmlCommandWithMessageHandling):
+    """GetTrM command.
+
+    Enables trace reporting from the bot.
+    """
+
+    NAME = "GetTrM"
+
+    @classmethod
+    def _handle_xml(cls, _event_bus: EventBus, xml: Element) -> HandlingResult:
+        """Handle xml message and notify the correct event subscribers.
+
+        :return: A message response
+        """
+        if xml.attrib.get("ret") != "ok":
+            return HandlingResult.analyse()
+        return HandlingResult(HandlingState.SUCCESS)

deebot_client/hardware/deebot/2pv572.py

@@ -33,7 +33,7 @@
 )
 from deebot_client.commands.xml.charge_state import GetChargeState
 from deebot_client.commands.xml.error import GetError
-from deebot_client.commands.xml.map import GetMapM, GetMapSt
+from deebot_client.commands.xml.map import GetMapM, GetMapSt, GetTrM
 from deebot_client.commands.xml.pos import GetPos
 from deebot_client.commands.xml.stats import GetCleanSum
 from deebot_client.commands.xml.water_info import GetWaterBoxInfo, SetWaterPermeability
@@ -109,7 +109,7 @@
             major=CapabilityEvent(MajorMapEvent, [GetMapM()]),
             position=CapabilityEvent(PositionsEvent, [GetPos()]),
             rooms=CapabilityEvent(RoomsEvent, [GetMapSt()]),
-            trace=CapabilityEvent(MapTraceEvent, []),
+            trace=CapabilityEvent(MapTraceEvent, [GetTrM()]),
         ),
         network=CapabilityEvent(NetworkInfoEvent, [GetNetInfoLegacy()]),
         play_sound=CapabilityExecute(PlaySound),