Default model added

rytilahti · Mar 24, 2018 · 702ef37 · 702ef37
1 parent 2731632
commit 702ef37
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/miio/chuangmi_plug.py b/miio/chuangmi_plug.py
@@ -9,11 +9,13 @@
 MODEL_CHUANGMI_PLUG_V3 = 'chuangmi.plug.v3'
 MODEL_CHUANGMI_PLUG_V1 = 'chuangmi.plug.v1'
 MODEL_CHUANGMI_PLUG_M1 = 'chuangmi.plug.m1'
+MODEL_CHUANGMI_PLUG_V2 = 'chuangmi.plug.v2'
 
 AVAILABLE_PROPERTIES = {
     MODEL_CHUANGMI_PLUG_V1: ['on', 'usb_on', 'temperature'],
     MODEL_CHUANGMI_PLUG_V3: ['on', 'usb_on', 'temperature', 'wifi_led'],
-    MODEL_CHUANGMI_PLUG_M1: ['power', 'temperature']
+    MODEL_CHUANGMI_PLUG_M1: ['power', 'temperature'],
+    MODEL_CHUANGMI_PLUG_V2: ['power', 'temperature'],
 }
 
 
@@ -91,7 +93,10 @@ def __init__(self, ip: str = None, token: str = None, start_id: int = 0,
                  model: str = MODEL_CHUANGMI_PLUG_M1) -> None:
         super().__init__(ip, token, start_id, debug, lazy_discover)
 
-        self.model = model
+        if model in AVAILABLE_PROPERTIES:
+            self.model = model
+        else:
+            self.model = MODEL_CHUANGMI_PLUG_M1
 
     def status(self) -> ChuangmiPlugStatus:
         """Retrieve properties."""