Allow alternative timezone format seen in Xioawa E25 (rytilahti#760)

* Allow alternative timezone format seen in Xioawa E25 * unwrap the response
xvlady · May 9, 2021 · 7971ffc · 7971ffc
1 parent 12f80bf
commit 7971ffc
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/miio/vacuum.py b/miio/vacuum.py
@@ -540,7 +540,17 @@ def locale(self):
     @command()
     def timezone(self):
         """Get the timezone."""
-        return self.send("get_timezone")[0]
+        res = self.send("get_timezone")[0]
+        if isinstance(res, dict):
+            # Xiaowa E25 example
+            # {'olson': 'Europe/Berlin', 'posix': 'CET-1CEST,M3.5.0,M10.5.0/3'}
+            if "olson" not in res:
+                raise VacuumException("Unsupported timezone format: %s" % res)
+
+            return res["olson"]
+
+        # Gen1 vacuum: ['Europe/Berlin']
+        return res
 
     def set_timezone(self, new_zone):
         """Set the timezone."""