Add more supported models (#1292)

* dreamevacuum: add dreame.vacuum.mc1808 as supported * airqualitymonitor_miot: add cgllc.airm.cgdn1 to supported models * Make flake8 happy
rytilahti · Jan 10, 2022 · 069bee8 · 069bee8
1 parent 0558c63
commit 069bee8
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 4 deletions.
diff --git a/miio/airqualitymonitor_miot.py b/miio/airqualitymonitor_miot.py
@@ -170,6 +170,7 @@ class AirQualityMonitorCGDN1(MiotDevice):
     """Qingping Air Monitor Lite."""
 
     mapping = _MAPPING_CGDN1
+    _supported_models = [MODEL_AIRQUALITYMONITOR_CGDN1]
 
     @command(
         default_output=format_output(

diff --git a/miio/integrations/vacuum/dreame/dreamevacuum_miot.py b/miio/integrations/vacuum/dreame/dreamevacuum_miot.py
@@ -195,6 +195,7 @@ class DreameVacuumMiot(MiotDevice):
     """Interface for Vacuum 1C STYTJ01ZHM (dreame.vacuum.mc1808)"""
 
     mapping = _MAPPING
+    _supported_models = ["dreame.vacuum.mc1808"]
 
     @command(
         default_output=format_output(

diff --git a/miio/protocol.py b/miio/protocol.py
@@ -131,10 +131,7 @@ def get_length(x) -> int:
     def is_hello(x) -> bool:
         """Return if packet is a hello packet."""
         # not very nice, but we know that hellos are 32b of length
-        if "length" in x:
-            val = x["length"]
-        else:
-            val = x.header.value["length"]
+        val = x.get("length", x.header.value["length"])
 
         return bool(val == 32)