Roborock: Add some more enum sensor metadata

rytilahti · Mar 20, 2024 · c03410a · c03410a
1 parent 356c6c0
commit c03410a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/miio/integrations/roborock/vacuum/vacuumcontainers.py b/miio/integrations/roborock/vacuum/vacuumcontainers.py
@@ -228,6 +228,8 @@ def error_code(self) -> int:
         "Error string",
         id=VacuumId.ErrorMessage,
         icon="mdi:alert",
+        device_class="enum",
+        options=list(ERROR_CODES.values()),
         entity_category="diagnostic",
         enabled_default=False,
     )
@@ -255,6 +257,8 @@ def dock_error_code(self) -> Optional[int]:
     @sensor(
         "Dock error string",
         icon="mdi:alert",
+        device_class="enum",
+        options=list(dock_error_codes.values()),
         entity_category="diagnostic",
         enabled_default=False,
     )