@@ -238,86 +238,6 @@ def v1_channel(self) -> V1Channel:
238238 """Returns the real V1Channel bound to the fake channels."""
239239 return self ._v1_channel
240240
241- @property
242- def battery (self ) -> int :
243- """Get battery level."""
244- return self .status .battery or 0
245-
246- @battery .setter
247- def battery (self , value : int ) -> None :
248- """Set battery level."""
249- self .status .battery = value
250-
251- @property
252- def state (self ) -> RoborockStateCode :
253- """Get device state code."""
254- return self .status .state or RoborockStateCode .charging
255-
256- @state .setter
257- def state (self , value : RoborockStateCode | int ) -> None :
258- """Set device state code."""
259- self .status .state = RoborockStateCode (value )
260-
261- @property
262- def fan_power (self ) -> int :
263- """Get fan power speed."""
264- return self .status .fan_power or 0
265-
266- @fan_power .setter
267- def fan_power (self , value : int ) -> None :
268- """Set fan power speed."""
269- self .status .fan_power = value
270-
271- @property
272- def dnd_enabled (self ) -> int :
273- """Get DND enabled state."""
274- return self .status .dnd_enabled or 0
275-
276- @dnd_enabled .setter
277- def dnd_enabled (self , value : int ) -> None :
278- """Set DND enabled state."""
279- self .status .dnd_enabled = value
280-
281- @property
282- def mop_mode (self ) -> int :
283- """Get mop route mode."""
284- return self .status .mop_mode or 0
285-
286- @mop_mode .setter
287- def mop_mode (self , value : int ) -> None :
288- """Set mop route mode."""
289- self .status .mop_mode = value
290-
291- @property
292- def water_box_mode (self ) -> int :
293- """Get water box mode."""
294- return self .status .water_box_mode or 0
295-
296- @water_box_mode .setter
297- def water_box_mode (self , value : int ) -> None :
298- """Set water box mode."""
299- self .status .water_box_mode = value
300-
301- @property
302- def dss (self ) -> int :
303- """Get dock sensor status."""
304- return self .status .dss or 0
305-
306- @dss .setter
307- def dss (self , value : int ) -> None :
308- """Set dock sensor status."""
309- self .status .dss = value
310-
311- @property
312- def dock_type (self ) -> RoborockDockTypeCode :
313- """Get dock type."""
314- return self .status .dock_type or RoborockDockTypeCode .no_dock
315-
316- @dock_type .setter
317- def dock_type (self , value : RoborockDockTypeCode | int ) -> None :
318- """Set dock type."""
319- self .status .dock_type = RoborockDockTypeCode (value )
320-
321241 @property
322242 def in_cleaning (self ) -> RoborockInCleaning :
323243 """Return global_clean_not_complete if cleaning, else complete."""
@@ -458,10 +378,10 @@ async def _handle_publish(self, message: RoborockMessage, channel: FakeChannel)
458378 def trigger_push_update (self ) -> None :
459379 """Trigger an unsolicited push state update to all subscribers."""
460380 dps_payload = {
461- str (int (RoborockDataProtocol .STATE )): self .state ,
462- str (int (RoborockDataProtocol .BATTERY )): self .battery ,
463- str (int (RoborockDataProtocol .FAN_POWER )): self .fan_power ,
464- str (int (RoborockDataProtocol .WATER_BOX_MODE )): self .water_box_mode ,
381+ str (int (RoborockDataProtocol .STATE )): self .status . state . value if self . status . state else 0 ,
382+ str (int (RoborockDataProtocol .BATTERY )): self .status . battery ,
383+ str (int (RoborockDataProtocol .FAN_POWER )): self .status . fan_power ,
384+ str (int (RoborockDataProtocol .WATER_BOX_MODE )): self .status . water_box_mode ,
465385 }
466386
467387 payload = {"dps" : dps_payload , "t" : int (time .time ())}
0 commit comments