From 24071dcd83b569959762931a9398c28a759ae186 Mon Sep 17 00:00:00 2001 From: Arun Suresh Kumar Date: Sun, 10 Dec 2023 18:20:43 +0530 Subject: [PATCH] Bug Fix: Avoid schema rebuilding if provided as argument --- graphene_federation/main.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/graphene_federation/main.py b/graphene_federation/main.py index 17cecf1..ec65ce0 100644 --- 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)