Merge pull request #832 from Kristinus/master

Include Enum to NinjaJSONEncoder

ninja/responses.py

@@ -1,3 +1,4 @@
+from enum import Enum
 from ipaddress import IPv4Address, IPv6Address
 from typing import Any, FrozenSet
 
@@ -22,6 +23,8 @@ def default(self, o: Any) -> Any:
             return o.model_dump()
         if isinstance(o, (IPv4Address, IPv6Address)):
             return str(o)
+        if isinstance(o, Enum):
+            return str(o)
         return super().default(o)
 
 

tests/test_response.py

@@ -1,4 +1,5 @@
 import json
+from enum import Enum
 from ipaddress import IPv4Address, IPv6Address
 from typing import List, Union
 
@@ -30,6 +31,11 @@ def __init__(self, id, user_name, password):
         self.password = password
 
 
+class MyEnum(Enum):
+    first = "first"
+    second = "second"
+
+
 def to_camel(string: str) -> str:
     return "".join(word.capitalize() for word in string.split("_"))
 
@@ -157,3 +163,10 @@ def test_ipv6address_encoding():
     response = Response(data)
     response_data = json.loads(response.content)
     assert response_data["ipv6"] == str(data["ipv6"])
+
+
+def test_enum_encoding():
+    data = {"enum": MyEnum.first}
+    response = Response(data)
+    response_data = json.loads(response.content)
+    assert response_data["enum"] == str(data["enum"])