Print native string value instead Enum when using ExternalDevice.Devi…

…ceType value (#478)
homewizard · Jan 7, 2025 · d1b3937 · d1b3937
1 parent d478b7e
commit d1b3937
Show file tree

Hide file tree

Showing 5 changed files with 11 additions and 11 deletions.
diff --git a/homewizard_energy/models.py b/homewizard_energy/models.py
@@ -4,7 +4,7 @@
 
 from dataclasses import dataclass, field
 from datetime import datetime
-from enum import Enum
+from enum import StrEnum
 from typing import Any
 
 from mashumaro.config import BaseConfig
@@ -448,7 +448,7 @@ def __post_deserialize__(cls, obj: Measurement) -> Measurement:
 class ExternalDevice(BaseModel):
     """Represents externally connected device."""
 
-    class DeviceType(Enum):
+    class DeviceType(StrEnum):
         """Device type allocations."""
 
         GAS_METER = "gas_meter"