Fix union of inline models

Author: forest-benchling

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

@@ -17,10 +17,13 @@
 )
 from .model_with_additional_properties_refed import ModelWithAdditionalPropertiesRefed
 from .model_with_any_json_properties import ModelWithAnyJsonProperties
-from .model_with_any_json_properties_additional_property import ModelWithAnyJsonPropertiesAdditionalProperty
+from .model_with_any_json_properties_additional_property_item0 import ModelWithAnyJsonPropertiesAdditionalPropertyItem0
 from .model_with_primitive_additional_properties import ModelWithPrimitiveAdditionalProperties
 from .model_with_primitive_additional_properties_a_date_holder import ModelWithPrimitiveAdditionalPropertiesADateHolder
 from .model_with_union_property import ModelWithUnionProperty
+from .model_with_union_property_inlined import ModelWithUnionPropertyInlined
+from .model_with_union_property_inlined_fruit_item0 import ModelWithUnionPropertyInlinedFruitItem0
+from .model_with_union_property_inlined_fruit_item1 import ModelWithUnionPropertyInlinedFruitItem1
 from .test_inline_objects_json_body import TestInlineObjectsJsonBody
 from .test_inline_objects_response_200 import TestInlineObjectsResponse_200
 from .validation_error import ValidationError

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

@@ -113,12 +113,12 @@ def _parse_a_camel_date_time(data: Any) -> Union[datetime.datetime, datetime.dat
             data = None if isinstance(data, Unset) else data
             a_camel_date_time: Union[datetime.datetime, datetime.date]
             try:
-                a_camel_date_time = isoparse(data)
+                isoparse(data)
 
                 return a_camel_date_time
             except:  # noqa: E722
                 pass
-            a_camel_date_time = isoparse(data).date()
+            isoparse(data).date()
 
             return a_camel_date_time
 

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

@@ -2,7 +2,9 @@
 
 import attr
 
-from ..models.model_with_any_json_properties_additional_property import ModelWithAnyJsonPropertiesAdditionalProperty
+from ..models.model_with_any_json_properties_additional_property_item0 import (
+    ModelWithAnyJsonPropertiesAdditionalPropertyItem0,
+)
 from ..types import Unset
 
 T = TypeVar("T", bound="ModelWithAnyJsonProperties")
@@ -13,14 +15,14 @@ class ModelWithAnyJsonProperties:
     """  """
 
     additional_properties: Dict[
-        str, Union[ModelWithAnyJsonPropertiesAdditionalProperty, List[str], str, float, int, bool]
+        str, Union[ModelWithAnyJsonPropertiesAdditionalPropertyItem0, List[str], str, float, int, bool]
     ] = attr.ib(init=False, factory=dict)
 
     def to_dict(self) -> Dict[str, Any]:
 
         field_dict: Dict[str, Any] = {}
         for prop_name, prop in self.additional_properties.items():
-            if isinstance(prop, ModelWithAnyJsonPropertiesAdditionalProperty):
+            if isinstance(prop, ModelWithAnyJsonPropertiesAdditionalPropertyItem0):
                 field_dict[prop_name] = prop.to_dict()
 
             elif isinstance(prop, list):
@@ -43,24 +45,26 @@ def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
 
             def _parse_additional_property(
                 data: Any,
-            ) -> Union[ModelWithAnyJsonPropertiesAdditionalProperty, List[str], str, float, int, bool]:
+            ) -> Union[ModelWithAnyJsonPropertiesAdditionalPropertyItem0, List[str], str, float, int, bool]:
                 data = None if isinstance(data, Unset) else data
                 additional_property: Union[
-                    ModelWithAnyJsonPropertiesAdditionalProperty, List[str], str, float, int, bool
+                    ModelWithAnyJsonPropertiesAdditionalPropertyItem0, List[str], str, float, int, bool
                 ]
                 try:
-                    additional_property = ModelWithAnyJsonPropertiesAdditionalProperty.from_dict(data)
+                    ModelWithAnyJsonPropertiesAdditionalPropertyItem0.from_dict(data)
 
                     return additional_property
                 except:  # noqa: E722
                     pass
                 try:
-                    additional_property = cast(List[str], data)
+                    cast(List[str], data)
 
                     return additional_property
                 except:  # noqa: E722
                     pass
-                return cast(Union[ModelWithAnyJsonPropertiesAdditionalProperty, List[str], str, float, int, bool], data)
+                return cast(
+                    Union[ModelWithAnyJsonPropertiesAdditionalPropertyItem0, List[str], str, float, int, bool], data
+                )
 
             additional_property = _parse_additional_property(prop_dict)
 
@@ -75,11 +79,13 @@ def additional_keys(self) -> List[str]:
 
     def __getitem__(
         self, key: str
-    ) -> Union[ModelWithAnyJsonPropertiesAdditionalProperty, List[str], str, float, int, bool]:
+    ) -> Union[ModelWithAnyJsonPropertiesAdditionalPropertyItem0, List[str], str, float, int, bool]:
         return self.additional_properties[key]
 
     def __setitem__(
-        self, key: str, value: Union[ModelWithAnyJsonPropertiesAdditionalProperty, List[str], str, float, int, bool]
+        self,
+        key: str,
+        value: Union[ModelWithAnyJsonPropertiesAdditionalPropertyItem0, List[str], str, float, int, bool],
     ) -> None:
         self.additional_properties[key] = value
 

end_to_end_tests/golden-record-custom/custom_e2e/models/model_with_any_json_properties_additional_property.py renamed to end_to_end_tests/golden-record-custom/custom_e2e/models/model_with_any_json_properties_additional_property_item0.py

@@ -2,11 +2,11 @@
 
 import attr
 
-T = TypeVar("T", bound="ModelWithAnyJsonPropertiesAdditionalProperty")
+T = TypeVar("T", bound="ModelWithAnyJsonPropertiesAdditionalPropertyItem0")
 
 
 @attr.s(auto_attribs=True)
-class ModelWithAnyJsonPropertiesAdditionalProperty:
+class ModelWithAnyJsonPropertiesAdditionalPropertyItem0:
     """  """
 
     additional_properties: Dict[str, str] = attr.ib(init=False, factory=dict)
@@ -22,10 +22,10 @@ def to_dict(self) -> Dict[str, Any]:
     @classmethod
     def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
         d = src_dict.copy()
-        model_with_any_json_properties_additional_property = cls()
+        model_with_any_json_properties_additional_property_item0 = cls()
 
-        model_with_any_json_properties_additional_property.additional_properties = d
-        return model_with_any_json_properties_additional_property
+        model_with_any_json_properties_additional_property_item0.additional_properties = d
+        return model_with_any_json_properties_additional_property_item0
 
     @property
     def additional_keys(self) -> List[str]:

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

@@ -44,18 +44,16 @@ 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
-                _a_property = data
-                if _a_property is not None and _a_property is not UNSET:
-                    a_property = AnEnum(_a_property)
+                _a_property_item0 = data
+                if _a_property_item0 is not None and _a_property_item0 is not UNSET:
+                    AnEnum(_a_property_item0)
 
                 return a_property
             except:  # noqa: E722
                 pass
-            a_property = UNSET
-            _a_property = data
-            if _a_property is not None and _a_property is not UNSET:
-                a_property = AnIntEnum(_a_property)
+            _a_property_item1 = data
+            if _a_property_item1 is not None and _a_property_item1 is not UNSET:
+                AnIntEnum(_a_property_item1)
 
             return a_property
 

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

@@ -0,0 +1,68 @@
+from typing import Any, Dict, Type, TypeVar, Union, cast
+
+import attr
+
+from ..models.model_with_union_property_inlined_fruit_item0 import ModelWithUnionPropertyInlinedFruitItem0
+from ..models.model_with_union_property_inlined_fruit_item1 import ModelWithUnionPropertyInlinedFruitItem1
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="ModelWithUnionPropertyInlined")
+
+
+@attr.s(auto_attribs=True)
+class ModelWithUnionPropertyInlined:
+    """  """
+
+    fruit: Union[Unset, ModelWithUnionPropertyInlinedFruitItem0, ModelWithUnionPropertyInlinedFruitItem1] = UNSET
+
+    def to_dict(self) -> Dict[str, Any]:
+        fruit: Union[Unset, ModelWithUnionPropertyInlinedFruitItem0, ModelWithUnionPropertyInlinedFruitItem1]
+        if isinstance(self.fruit, Unset):
+            fruit = UNSET
+        elif isinstance(self.fruit, ModelWithUnionPropertyInlinedFruitItem0):
+            fruit = UNSET
+            if not isinstance(self.fruit, Unset):
+                fruit = self.fruit.to_dict()
+
+        else:
+            fruit = UNSET
+            if not isinstance(self.fruit, Unset):
+                fruit = self.fruit.to_dict()
+
+        field_dict: Dict[str, Any] = {}
+        field_dict.update({})
+        if fruit is not UNSET:
+            field_dict["fruit"] = fruit
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
+        d = src_dict.copy()
+
+        def _parse_fruit(
+            data: Any,
+        ) -> Union[Unset, ModelWithUnionPropertyInlinedFruitItem0, ModelWithUnionPropertyInlinedFruitItem1]:
+            data = None if isinstance(data, Unset) else data
+            fruit: Union[Unset, ModelWithUnionPropertyInlinedFruitItem0, ModelWithUnionPropertyInlinedFruitItem1]
+            try:
+                _fruit_item0 = data
+                if not isinstance(_fruit_item0, Unset):
+                    ModelWithUnionPropertyInlinedFruitItem0.from_dict(cast(Dict[str, Any], _fruit_item0))
+
+                return fruit
+            except:  # noqa: E722
+                pass
+            _fruit_item1 = data
+            if not isinstance(_fruit_item1, Unset):
+                ModelWithUnionPropertyInlinedFruitItem1.from_dict(cast(Dict[str, Any], _fruit_item1))
+
+            return fruit
+
+        fruit = _parse_fruit(d.pop("fruit", UNSET))
+
+        model_with_union_property_inlined = cls(
+            fruit=fruit,
+        )
+
+        return model_with_union_property_inlined

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

@@ -0,0 +1,54 @@
+from typing import Any, Dict, List, Type, TypeVar, Union
+
+import attr
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="ModelWithUnionPropertyInlinedFruitItem0")
+
+
+@attr.s(auto_attribs=True)
+class ModelWithUnionPropertyInlinedFruitItem0:
+    """  """
+
+    apples: Union[Unset, str] = UNSET
+    additional_properties: Dict[str, Any] = attr.ib(init=False, factory=dict)
+
+    def to_dict(self) -> Dict[str, Any]:
+        apples = self.apples
+
+        field_dict: Dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if apples is not UNSET:
+            field_dict["apples"] = apples
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
+        d = src_dict.copy()
+        apples = d.pop("apples", UNSET)
+
+        model_with_union_property_inlined_fruit_item0 = cls(
+            apples=apples,
+        )
+
+        model_with_union_property_inlined_fruit_item0.additional_properties = d
+        return model_with_union_property_inlined_fruit_item0
+
+    @property
+    def additional_keys(self) -> List[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

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

@@ -0,0 +1,54 @@
+from typing import Any, Dict, List, Type, TypeVar, Union
+
+import attr
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="ModelWithUnionPropertyInlinedFruitItem1")
+
+
+@attr.s(auto_attribs=True)
+class ModelWithUnionPropertyInlinedFruitItem1:
+    """  """
+
+    bananas: Union[Unset, str] = UNSET
+    additional_properties: Dict[str, Any] = attr.ib(init=False, factory=dict)
+
+    def to_dict(self) -> Dict[str, Any]:
+        bananas = self.bananas
+
+        field_dict: Dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if bananas is not UNSET:
+            field_dict["bananas"] = bananas
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: Dict[str, Any]) -> T:
+        d = src_dict.copy()
+        bananas = d.pop("bananas", UNSET)
+
+        model_with_union_property_inlined_fruit_item1 = cls(
+            bananas=bananas,
+        )
+
+        model_with_union_property_inlined_fruit_item1.additional_properties = d
+        return model_with_union_property_inlined_fruit_item1
+
+    @property
+    def additional_keys(self) -> List[str]:
+        return list(self.additional_properties.keys())
+
+    def __getitem__(self, key: str) -> Any:
+        return self.additional_properties[key]
+
+    def __setitem__(self, key: str, value: Any) -> None:
+        self.additional_properties[key] = value
+
+    def __delitem__(self, key: str) -> None:
+        del self.additional_properties[key]
+
+    def __contains__(self, key: str) -> bool:
+        return key in self.additional_properties

end_to_end_tests/golden-record/my_test_api_client/models/__init__.py

@@ -17,10 +17,13 @@
 )
 from .model_with_additional_properties_refed import ModelWithAdditionalPropertiesRefed
 from .model_with_any_json_properties import ModelWithAnyJsonProperties
-from .model_with_any_json_properties_additional_property import ModelWithAnyJsonPropertiesAdditionalProperty
+from .model_with_any_json_properties_additional_property_item0 import ModelWithAnyJsonPropertiesAdditionalPropertyItem0
 from .model_with_primitive_additional_properties import ModelWithPrimitiveAdditionalProperties
 from .model_with_primitive_additional_properties_a_date_holder import ModelWithPrimitiveAdditionalPropertiesADateHolder
 from .model_with_union_property import ModelWithUnionProperty
+from .model_with_union_property_inlined import ModelWithUnionPropertyInlined
+from .model_with_union_property_inlined_fruit_item0 import ModelWithUnionPropertyInlinedFruitItem0
+from .model_with_union_property_inlined_fruit_item1 import ModelWithUnionPropertyInlinedFruitItem1
 from .test_inline_objects_json_body import TestInlineObjectsJsonBody
 from .test_inline_objects_response_200 import TestInlineObjectsResponse_200
 from .validation_error import ValidationError

end_to_end_tests/golden-record/my_test_api_client/models/a_model.py

@@ -113,12 +113,12 @@ def _parse_a_camel_date_time(data: Any) -> Union[datetime.datetime, datetime.dat
             data = None if isinstance(data, Unset) else data
             a_camel_date_time: Union[datetime.datetime, datetime.date]
             try:
-                a_camel_date_time = isoparse(data)
+                isoparse(data)
 
                 return a_camel_date_time
             except:  # noqa: E722
                 pass
-            a_camel_date_time = isoparse(data).date()
+            isoparse(data).date()
 
             return a_camel_date_time