Return enum values

detriment · Feb 24, 2023 · 03a6bb3 · 03a6bb3
1 parent a78f531
commit 03a6bb3
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/regrws/arin_xml_encoder.py b/regrws/arin_xml_encoder.py
@@ -1,3 +1,4 @@
+from enum import Enum
 from typing import Any
 
 from pydantic_xml.serializers import XmlEncoder
@@ -7,4 +8,6 @@ class ARINXmlEncoder(XmlEncoder):
     def encode(self, obj: Any) -> str:
         if isinstance(obj, bool):
             return "true" if obj else "false"
+        if isinstance(obj, Enum):
+            return obj.value
         return super().encode(obj)