Allow passing choices as dict

rytilahti · Mar 18, 2024 · 328fc72 · 328fc72
1 parent b1f6cd6
commit 328fc72
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/miio/descriptorcollection.py b/miio/descriptorcollection.py
@@ -1,5 +1,6 @@
 import logging
 from collections import UserDict
+from enum import Enum
 from inspect import getmembers
 from typing import TYPE_CHECKING, Generic, TypeVar, cast
 
@@ -121,7 +122,11 @@ def _handle_constraints(self, prop: PropertyDescriptor) -> None:
                 retrieve_choices_function = getattr(
                     self._device, prop.choices_attribute
                 )
-                prop.choices = retrieve_choices_function()
+                choices = retrieve_choices_function()
+                if isinstance(choices, dict):
+                    prop.choices = Enum(f"GENERATED_ENUM_{prop.name}", choices)
+                else:
+                    prop.choices = choices
 
             if prop.choices is None:
                 raise ValueError(