Bug Fix: Avoid schema rebuilding if provided as argument

strollby · Dec 10, 2023 · 24071dc · 24071dc
1 parent 13faef1
commit 24071dc
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/graphene_federation/main.py b/graphene_federation/main.py
@@ -21,12 +21,13 @@ def _get_query(schema: Schema, query_cls: Optional[ObjectType] = None) -> Object
 
 
 def build_schema(
-    query: Optional[ObjectType] = None,
-    mutation: Optional[ObjectType] = None,
-    enable_federation_2=False,
-    **kwargs
+        query: Optional[ObjectType] = None,
+        mutation: Optional[ObjectType] = None,
+        enable_federation_2=False,
+        schema: Schema | None = None,
+        **kwargs
 ) -> Schema:
-    schema = Schema(query=query, mutation=mutation, **kwargs)
+    schema = schema or Schema(query=query, mutation=mutation, **kwargs)
     schema.auto_camelcase = kwargs.get("auto_camelcase", True)
     schema.federation_version = 2 if enable_federation_2 else 1
     federation_query = _get_query(schema, query)