reorganize-function-placement

mirumee · Apr 10, 2024 · 4f96aab · 4f96aab
1 parent bee71e8
commit 4f96aab
Showing 1 changed file with 20 additions and 20 deletions.
diff --git a/ariadne_graphql_proxy/query_filter.py b/ariadne_graphql_proxy/query_filter.py
@@ -149,26 +149,6 @@ def filter_operation_node(
             ),
         )
 
-    def extract_variables(
-        self,
-        value: VariableNode | ListValueNode | ObjectValueNode,
-        context: QueryFilterContext,
-    ):
-        if isinstance(value, VariableNode):
-            context.variables.add(value.name.value)
-        elif isinstance(value, ObjectValueNode):
-            for field in value.fields:
-                self.extract_variables(field.value, context)  # type: ignore
-        elif isinstance(value, ListValueNode):
-            for item in value.values:
-                self.extract_variables(item, context)  # type: ignore
-
-    def update_context_variables(
-        self, field_node: FieldNode, context: QueryFilterContext
-    ):
-        for argument in field_node.arguments:
-            self.extract_variables(argument.value, context)  # type: ignore
-
     def filter_field_node(
         self,
         field_node: FieldNode,
@@ -408,3 +388,23 @@ def get_type_fields_dependencies(
             return self.dependencies[schema_id][type_name]
 
         return None
+
+    def update_context_variables(
+        self, field_node: FieldNode, context: QueryFilterContext
+    ):
+        for argument in field_node.arguments:
+            self.extract_variables(argument.value, context)  # type: ignore
+
+    def extract_variables(
+        self,
+        value: VariableNode | ListValueNode | ObjectValueNode,
+        context: QueryFilterContext,
+    ):
+        if isinstance(value, VariableNode):
+            context.variables.add(value.name.value)
+        elif isinstance(value, ObjectValueNode):
+            for field in value.fields:
+                self.extract_variables(field.value, context)  # type: ignore
+        elif isinstance(value, ListValueNode):
+            for item in value.values:
+                self.extract_variables(item, context)  # type: ignore