🚶 improve for walking-pad (#261)

al-one · Nov 29, 2021 · 26c2cfd · 26c2cfd
1 parent 47a39fc
commit 26c2cfd
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/custom_components/xiaomi_miot/__init__.py b/custom_components/xiaomi_miot/__init__.py
@@ -1118,7 +1118,10 @@ async def async_update(self):
         self._state = True if self._state_attrs.get('power') else False
 
         if self._miot_service:
-            for d in ['sensor', 'binary_sensor', 'switch', 'number', 'select', 'fan', 'cover']:
+            for d in [
+                'sensor', 'binary_sensor', 'switch', 'number',
+                'select', 'fan', 'cover', 'number_select',
+            ]:
                 pls = self.custom_config_list(f'{d}_properties') or []
                 if pls:
                     self._update_sub_entities(pls, '*', domain=d)

diff --git a/custom_components/xiaomi_miot/core/device_customizes.py b/custom_components/xiaomi_miot/core/device_customizes.py
@@ -436,7 +436,8 @@
         'sensor_properties': 'current_distance,current_step_count,current_calorie_consumption,'
                              'left_distance,left_time,working_time',
         'number_properties': 'target_distance,target_time',
-        'select_properties': 'mode,speed_level',
+        'select_properties': 'mode',
+        'number_select_properties': 'speed_level',
     },
 
 }