wrap mode to avoid crashing on None, use __repr__ instead of __str__

rytilahti · Nov 18, 2017 · 9d3fdcc · 9d3fdcc
1 parent 243f14d
commit 9d3fdcc
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/miio/powerstrip.py b/miio/powerstrip.py
@@ -43,10 +43,12 @@ def load_power(self) -> Optional[float]:
         return None
 
     @property
-    def mode(self) -> PowerMode:
-        return PowerMode(self.data["mode"])
+    def mode(self) -> Optional[PowerMode]:
+        if self.data["mode"] is not None:
+            return PowerMode(self.data["mode"])
+        return None
 
-    def __str__(self) -> str:
+    def __repr__(self) -> str:
         s = "<PowerStripStatus power=%s, temperature=%s, " \
             "load_power=%s mode=%s>" % \
             (self.power,