fix: 🐛 Ensure UnionType is handled correctly on Python >= 3.10

Author: ddanier

pydantic_partial/partial.py

@@ -95,6 +95,8 @@ def _partial_annotation_arg(field_name_: str, field_annotation: type) -> type:
         if recursive or sub_fields_requested:
             field_annotation_origin = get_origin(field_annotation)
             if field_annotation_origin in (Union, UnionType, tuple, list, set, dict):
+                if field_annotation_origin is UnionType:
+                    field_annotation_origin = Union
                 field_annotation = field_annotation_origin[  # type: ignore
                     tuple(  # type: ignore
                         _partial_annotation_arg(field_name, field_annotation_arg)

tests/test_partial_without_mixin.py

@@ -1,4 +1,4 @@
-import json
+import sys
 from typing import Union
 
 import pydantic
@@ -24,6 +24,13 @@ class SomethingWithMixin(PartialModelMixin, pydantic.BaseModel):
     name: str
 
 
+class SomethingWithUnionTypes(PartialModelMixin, pydantic.BaseModel):
+    name_as_union: Union[str, int]
+
+    if sys.version_info >= (3, 10):
+        name_as_uniontype: str | int
+
+
 def test_setup_is_sane():
     assert _field_is_required(Something, "name") is True
     assert _field_is_required(Something, "age") is True
@@ -67,3 +74,7 @@ def test_partial_class_name_can_be_overridden():
     partial_cls_name = "SomethingWithOptionalName"
     SomethingWithOptionalName = create_partial_model(Something, "name", partial_cls_name=partial_cls_name)
     assert SomethingWithOptionalName.__name__ == partial_cls_name
+
+
+def test_recursive_on_unions_work():  # see https://github.com/team23/pydantic-partial/issues/52
+    create_partial_model(SomethingWithUnionTypes, recursive=True)

tox.ini

@@ -10,4 +10,8 @@ envlist =
 [testenv]
 deps =
     pytest
+    fastapi
+    anyio
+    httpx
+    trio
 commands = pytest