fix: @sharable is applied multiple times on PageInfo in multithreaded…

… environments
strollby · Apr 22, 2024 · 3e4013b · 3e4013b
1 parent bee1ded
commit 3e4013b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/graphene_federation/schema.py b/graphene_federation/schema.py
@@ -48,12 +48,13 @@ def _add_sharable_to_page_info_type(
     """
     Add @sharable directive to PageInfo type
     """
-    if PageInfo.__name__ in schema.graphql_schema.type_map:
+
+    if page_info := schema.graphql_schema.type_map.get(PageInfo.__name__):
         try:
             # PageInfo needs @sharable directive
             sharable = get_directive_from_name("shareable", federation_version)
             types.append(
-                directive_decorator(target_directive=sharable)(field=None)(PageInfo)
+                directive_decorator(target_directive=sharable)(field=None)(page_info)
             )
         except ValueError:
             # Federation Version does not support @sharable