Pydantic compatibility features

graphene_pydantic/__init__.py

@@ -1,4 +1,4 @@
 from .inputobjecttype import PydanticInputObjectType
 from .objecttype import PydanticObjectType
 
-__all__ = ["PydanticObjectType", "PydanticInputObjectType"]
+__all__ = ["PydanticObjectType", "PydanticInputObjectType", "Schema"]

graphene_pydantic/schemas.py

@@ -0,0 +1,56 @@
+from graphene import Schema as GrapheneSchema
+from graphene.types.schema import normalize_execute_kwargs
+from graphql import (
+    GraphQLList,
+    GraphQLNamedType,
+    GraphQLNonNull,
+    GraphQLType,
+    OperationDefinitionNode,
+    graphql_sync,
+    is_input_object_type,
+    type_from_ast,
+)
+from graphql.language import parse
+
+
+class Schema(GrapheneSchema):
+    def execute(self, *args, **kwargs):
+        kwargs = normalize_execute_kwargs(kwargs)
+        var_types = self.override_input_fields(args, kwargs)
+        response = graphql_sync(self.graphql_schema, *args, **kwargs)
+        response.data = self.replace_empty_string_to_none(response.data)
+        for var_type, fields in var_types.items():
+            var_type.fields = fields
+        return response
+
+    def override_input_fields(self, args, kwargs) -> dict:
+        source = args[0]
+        document = parse(source)
+        operation_name = kwargs.get('operation_name')
+        operation = None
+        for definition in document.definitions:
+            if isinstance(definition, OperationDefinitionNode):
+                if (
+                    operation_name is None
+                    and not operation
+                    or definition.name
+                    and definition.name.value == operation_name
+                ):
+                    operation = definition
+        var_def_nodes = operation.variable_definitions
+        if not var_def_nodes:
+            return {}
+
+        var_types_fields = {}
+        for var_def_node in var_def_nodes:
+            var_type = type_from_ast(self.graphql_schema, var_def_node.type)
+            if is_input_object_type(var_type):
+                var_types_fields[var_type] = var_type.fields.copy()
+                fields = {}
+                for field_name, value in var_type.fields.items():
+                    if field_name in list(kwargs.get('variable_values', {}).values())[0]:
+                        fields.update({f'{field_name}': value})
+                var_type.fields = fields
+            else:
+                continue
+        return var_types_fields