Hacs Support and New Features

custom_components/ vacuum-styj02ym-ha/manifest.json

@@ -0,0 +1,12 @@
+{
+  "domain": "miio2",
+  "name": "Xiaomi miio vacuum STYJ02YM",
+  "documentation": "https://github.com/weslleycsil/vacuum-styj02ym-ha",
+  "version": "1.0.0",
+  "requirements": [
+      "construct==2.10.56",
+      "python-miio==0.5.12"
+  ],
+  "dependencies": [],
+  "codeowners": ["@weslleycsil"]
+}

vacuum.py → custom_components/ vacuum-styj02ym-ha/vacuum.py

@@ -3,7 +3,7 @@
 from functools import partial
 import logging
 
-from miio import DeviceException, Vacuum  # pylint: disable=import-error
+from miio import DeviceException, RoborockVacuum  # pylint: disable=import-error
 import voluptuous as vol
 
 from homeassistant.components.vacuum import (
@@ -25,7 +25,7 @@
     SUPPORT_START,
     SUPPORT_STATE,
     SUPPORT_STOP,
-    StateVacuumDevice,
+    StateVacuumEntity,
 )
 from homeassistant.const import (
     ATTR_ENTITY_ID,
@@ -120,7 +120,16 @@
 }
 
 ALL_PROPS = ["run_state", "mode", "err_state", "battary_life", "box_type", "mop_type", "s_time",
-             "s_area", "suction_grade", "water_grade", "remember_map", "has_map", "is_mop", "has_newmap"]
+             "s_area", "suction_grade", "water_grade", "remember_map", "has_map", "is_mop", "has_newmap",
+             "side_brush_life", "side_brush_hours", "main_brush_life", "main_brush_hours", "hypa_life", "hypa_hours", "mop_life", "mop_hours", "water_percent","hw_info", "sw_info", "start_time", "order_time", "v_state", "zone_data", "repeat_state", "light_state", "is_charge", "is_work", "cur_mapid", "mop_route", "map_num"]
+
+VACUUM_CARD_PROPS_REFERENCES = {
+    'main_brush_left': 'main_brush_hours',
+    'side_brush_left': 'side_brush_hours',
+    'filter_left': 'hypa_hours',
+    'sensor_dirty_left': 'mop_hours',
+    'cleaned_area': 's_area',
+    'cleaning_time': 's_time'}
 
 
 async def async_setup_platform(hass, config, async_add_entities, discovery_info=None):
@@ -134,7 +143,7 @@ async def async_setup_platform(hass, config, async_add_entities, discovery_info=
 
   # Create handler
   _LOGGER.info("Initializing with host %s (token %s...)", host, token[:5])
-  vacuum = Vacuum(host, token)
+  vacuum = RoborockVacuum(host, token)
 
   mirobo = MiroboVacuum2(name, vacuum)
   hass.data[DATA_KEY][host] = mirobo
@@ -176,7 +185,8 @@ async def async_service_handler(service):
     )
 
 
-class MiroboVacuum2(StateVacuumDevice):
+#class MiroboVacuum2(StateVacuumDevice):
+class MiroboVacuum2(StateVacuumEntity):
   """Representation of a Xiaomi Vacuum cleaner robot."""
 
   def __init__(self, name, vacuum):
@@ -369,6 +379,9 @@ def update(self):
 
       self.vacuum_state = dict(zip(ALL_PROPS, state))
 
+      for prop in VACUUM_CARD_PROPS_REFERENCES.keys():
+                self.vacuum_state[prop] = self.vacuum_state[VACUUM_CARD_PROPS_REFERENCES[prop]]
+
       self._available = True
 
       # Automatically set mop based on mop_type
@@ -410,4 +423,4 @@ async def async_clean_point(self, point):
     x, y = point
     self._last_clean_point = point
     await self._try_command("Unable to clean point: %s", self._vacuum.raw_command, 'set_uploadmap', [0]) \
-        and await self._try_command("Unable to clean point: %s", self._vacuum.raw_command, 'set_pointclean', [1, x, y])
+        and await self._try_command("Unable to clean point: %s", self._vacuum.raw_command, 'set_pointclean', [1, x, y])