Regen w/ custom template

Author: packyg

end_to_end_tests/golden-record-custom/custom_e2e/api/tests/defaults_tests_defaults_post.py

@@ -77,7 +77,7 @@ def httpx_request(
 
     json_enum_prop: Union[Unset, AnEnum] = UNSET
     if not isinstance(enum_prop, Unset):
-        json_enum_prop = enum_prop.value
+        json_enum_prop = enum_prop
 
     params: Dict[str, Any] = {}
     if string_prop is not UNSET:

end_to_end_tests/golden-record-custom/custom_e2e/models/__init__.py

@@ -6,6 +6,7 @@
 from .body_upload_file_tests_upload_post import BodyUploadFileTestsUploadPost
 from .different_enum import DifferentEnum
 from .http_validation_error import HTTPValidationError
+from .model_with_union_property import ModelWithUnionProperty
 from .test_inline_objectsjson_body import TestInlineObjectsjsonBody
 from .test_inline_objectsresponse_200 import TestInlineObjectsresponse_200
 from .validation_error import ValidationError

end_to_end_tests/golden-record-custom/custom_e2e/models/a_model.py

@@ -74,15 +74,16 @@ def to_dict(self) -> Dict[str, Any]:
     def from_dict(d: Dict[str, Any]) -> "AModel":
         an_enum_value = AnEnum(d["an_enum_value"])
 
-        def _parse_a_camel_date_time(data: Dict[str, Any]) -> Union[datetime.datetime, datetime.date]:
+        def _parse_a_camel_date_time(data: Any) -> Union[datetime.datetime, datetime.date]:
+            data = None if isinstance(data, Unset) else data
             a_camel_date_time: Union[datetime.datetime, datetime.date]
             try:
-                a_camel_date_time = isoparse(d["aCamelDateTime"])
+                a_camel_date_time = isoparse(data)
 
                 return a_camel_date_time
             except:  # noqa: E722
                 pass
-            a_camel_date_time = isoparse(d["aCamelDateTime"]).date()
+            a_camel_date_time = isoparse(data).date()
 
             return a_camel_date_time
 

end_to_end_tests/golden-record-custom/custom_e2e/models/model_with_union_property.py

@@ -0,0 +1,59 @@
+from typing import Any, Dict, Union
+
+import attr
+
+from ..models.an_enum import AnEnum
+from ..models.an_int_enum import AnIntEnum
+from ..types import UNSET, Unset
+
+
+@attr.s(auto_attribs=True)
+class ModelWithUnionProperty:
+    """  """
+
+    a_property: Union[Unset, AnEnum, AnIntEnum] = UNSET
+
+    def to_dict(self) -> Dict[str, Any]:
+        a_property: Union[Unset, AnEnum, AnIntEnum]
+        if isinstance(self.a_property, Unset):
+            a_property = UNSET
+        elif isinstance(self.a_property, AnEnum):
+            a_property = UNSET
+            if not isinstance(self.a_property, Unset):
+                a_property = self.a_property
+
+        else:
+            a_property = UNSET
+            if not isinstance(self.a_property, Unset):
+                a_property = self.a_property
+
+        field_dict = {}
+        if a_property is not UNSET:
+            field_dict["a_property"] = a_property
+
+        return field_dict
+
+    @staticmethod
+    def from_dict(d: Dict[str, Any]) -> "ModelWithUnionProperty":
+        def _parse_a_property(data: Any) -> Union[Unset, AnEnum, AnIntEnum]:
+            data = None if isinstance(data, Unset) else data
+            a_property: Union[Unset, AnEnum, AnIntEnum]
+            try:
+                a_property = UNSET
+                if data is not None:
+                    a_property = AnEnum(data)
+
+                return a_property
+            except:  # noqa: E722
+                pass
+            a_property = UNSET
+            if data is not None:
+                a_property = AnIntEnum(data)
+
+            return a_property
+
+        a_property = _parse_a_property(d.get("a_property", UNSET))
+
+        return ModelWithUnionProperty(
+            a_property=a_property,
+        )