fix: UnionType imported in python versions < 3.10

Author: mak626

graphene_pydantic/converters.py

@@ -7,7 +7,6 @@
 import sys
 import typing as T
 import uuid
-from types import UnionType
 from typing import Type, get_origin
 
 import graphene
@@ -33,6 +32,10 @@
 from .registry import Registry
 from .util import construct_union_class_name, evaluate_forward_ref
 
+PYTHON10 = sys.version_info >= (3, 10)
+if PYTHON10:
+    from types import UnionType
+
 GRAPHENE2 = graphene.VERSION[0] < 3
 
 try:
@@ -114,10 +117,15 @@ def convert_pydantic_field(
     """
     declared_type = getattr(field, "annotation", None)
 
-    # Convert Python 11 UnionType to T.Union
-    is_union_type = (
-        get_origin(declared_type) is T.Union or get_origin(declared_type) is UnionType
-    )
+    # Convert Python 10 UnionType to T.Union
+    if PYTHON10:
+        is_union_type = (
+            get_origin(declared_type) is T.Union
+            or get_origin(declared_type) is UnionType
+        )
+    else:
+        is_union_type = get_origin(declared_type) is T.Union
+
     if is_union_type:
         declared_type = T.Union[declared_type.__args__]
 
@@ -195,9 +203,10 @@ def find_graphene_type(
     throwing an error if we don't know what to map it to.
     """
 
-    # Convert Python 11 UnionType to T.Union
-    if isinstance(type_, UnionType):
-        type_ = T.Union[type_.__args__]
+    # Convert Python 10 UnionType to T.Union
+    if PYTHON10:
+        if isinstance(type_, UnionType):
+            type_ = T.Union[type_.__args__]
 
     if type_ == uuid.UUID:
         return UUID

graphene_pydantic/inputobjecttype.py

@@ -1,5 +1,4 @@
 import typing as T
-from types import UnionType
 
 import graphene
 import pydantic
@@ -46,7 +45,8 @@ def construct_fields(
     fields = {}
     for name, field in fields_to_convert:
         # Graphql does not accept union as input. Refer https://github.com/graphql/graphql-spec/issues/488
-        if isinstance(getattr(field, "annotation", None), UnionType):
+        annotation = getattr(field, "annotation", None)
+        if isinstance(annotation, str) or isinstance(annotation, int):
             union_types = field.annotation.__args__
             if type(None) not in union_types or len(union_types) > 2:
                 continue
@@ -80,8 +80,8 @@ def __init_subclass_with_meta__(
         _meta=None,
         **options,
     ):
-        assert model and issubclass(
-            model, pydantic.BaseModel
+        assert (
+            model and issubclass(model, pydantic.BaseModel)
         ), f'You need to pass a valid Pydantic model in {cls.__name__}.Meta, received "{model}"'
 
         assert isinstance(

graphene_pydantic/objecttype.py

@@ -4,7 +4,6 @@
 import pydantic
 from graphene.types.objecttype import ObjectTypeOptions
 from graphene.types.utils import yank_fields_from_attrs
-from graphql import GraphQLResolveInfo
 
 from .converters import convert_pydantic_field
 from .inputobjecttype import PydanticInputObjectType
@@ -72,8 +71,8 @@ def __init_subclass_with_meta__(
         _meta=None,
         **options,
     ):
-        assert model and issubclass(
-            model, pydantic.BaseModel
+        assert (
+            model and issubclass(model, pydantic.BaseModel)
         ), f'You need to pass a valid Pydantic model in {cls.__name__}.Meta, received "{model}"'
 
         assert isinstance(
@@ -159,7 +158,7 @@ def resolve_placeholders(cls):
         meta.fields.update(fields_to_update)
 
     @classmethod
-    def is_type_of(cls, root, _: GraphQLResolveInfo) -> bool:
+    def is_type_of(cls, root, info) -> bool:
         if isinstance(root, PydanticInputObjectType):
-            return type(root._meta.model) is type(cls._meta.model)
+            return type(root._meta.model) is type(cls._meta.model)  # noqa: E721
         return isinstance(root, cls._meta.model)