Add timezone

homeassistant/components/xiaomi_miio/vacuum.py

@@ -80,6 +80,7 @@
 ATTR_STATUS = "status"
 ATTR_ZONE_ARRAY = "zone"
 ATTR_ZONE_REPEATER = "repeats"
+ATTR_TIMEZONE = "timezone"
 ATTR_TIMERS = "timers"
 
 VACUUM_SERVICE_SCHEMA = vol.Schema({vol.Optional(ATTR_ENTITY_ID): cv.comp_entity_ids})
@@ -248,6 +249,7 @@ def __init__(self, name, vacuum):
         self._fan_speeds = None
         self._fan_speeds_reverse = None
 
+        self.timezone = None
         self._timers = None
 
     @property
@@ -347,6 +349,9 @@ def device_state_attributes(self):
             if self.vacuum_state.got_error:
                 attrs[ATTR_ERROR] = self.vacuum_state.error
 
+            if self.timezone:
+                attrs[ATTR_TIMEZONE] = self.timezone
+
             if self.timers:
                 attrs[ATTR_TIMERS] = self.timers
         return attrs
@@ -475,6 +480,7 @@ def update(self):
             self.last_clean = self._vacuum.last_clean_details()
             self.dnd_state = self._vacuum.dnd_status()
 
+            self.timezone = self._vacuum.timezone()
             self._timers = self._vacuum.timer()
 
             self._available = True

tests/components/xiaomi_miio/test_vacuum.py

@@ -33,6 +33,7 @@
     ATTR_MAIN_BRUSH_LEFT,
     ATTR_SIDE_BRUSH_LEFT,
     ATTR_TIMERS,
+    ATTR_TIMEZONE,
     CONF_HOST,
     CONF_NAME,
     CONF_TOKEN,
@@ -105,6 +106,8 @@ def mirobo_is_got_error_fixture():
 
     mock_vacuum.timer.return_value = [mock_timer_1, mock_timer_2]
 
+    mock_vacuum.timezone.return_value = "Europe/Berlin"
+
     with mock.patch(
         "homeassistant.components.xiaomi_miio.vacuum.Vacuum"
     ) as mock_vaccum_cls:
@@ -181,6 +184,8 @@ def mirobo_is_on_fixture():
 
     mock_vacuum.timer.return_value = [mock_timer_1, mock_timer_2]
 
+    mock_vacuum.timezone.return_value = "Europe/Berlin"
+
     with mock.patch(
         "homeassistant.components.xiaomi_miio.vacuum.Vacuum"
     ) as mock_vaccum_cls:
@@ -266,6 +271,7 @@ async def test_xiaomi_vacuum_services(hass, caplog, mock_mirobo_is_got_error):
         {"enabled": True, "time": time(hour=9, minute=0)},
         {"enabled": False, "time": time(hour=11, minute=0)},
     ]
+    assert state.attributes.get(ATTR_TIMEZONE) == "Europe/Berlin"
 
     # Call services
     await hass.services.async_call(
@@ -370,6 +376,7 @@ async def test_xiaomi_specific_services(hass, caplog, mock_mirobo_is_on):
         {"enabled": True, "time": time(hour=9, minute=0)},
         {"enabled": False, "time": time(hour=11, minute=0)},
     ]
+    assert state.attributes.get(ATTR_TIMEZONE) == "Europe/Berlin"
 
     # Xiaomi vacuum specific services:
     await hass.services.async_call(