fix: return entity options in get_available_entities (#14)

Filter out null properties in response message.
unfoldedcircle · Feb 27, 2024 · bae863d · bae863d
1 parent b82730f
commit bae863d
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/ucapi/entities.py b/ucapi/entities.py
@@ -102,9 +102,13 @@ def get_all(self) -> list[dict[str, Any]]:
                 "device_id": entity.device_id,
                 "features": entity.features,
                 "name": entity.name,
-                "area": entity.area,
-                "device_class": entity.device_class,
             }
+            if entity.device_class:
+                res["device_class"] = entity.device_class
+            if entity.options:
+                res["options"] = entity.options
+            if entity.area:
+                res["area"] = entity.area
 
             entities.append(res)