@@ -56,7 +56,9 @@ def enter_variable_definition(self, *_args: Any) -> VisitorAction:
56
56
57
57
def enter_variable (self , node : VariableNode , * _args : Any ) -> VisitorAction :
58
58
type_info = self ._type_info
59
- usage = VariableUsage (node , type_info .get_input_type (), type_info .get_default_value ())
59
+ usage = VariableUsage (
60
+ node , type_info .get_input_type (), type_info .get_default_value ()
61
+ )
60
62
self ._append_usage (usage )
61
63
return None
62
64
@@ -73,9 +75,13 @@ class ASTValidationContext:
73
75
74
76
_fragments : Optional [Dict [str , FragmentDefinitionNode ]]
75
77
_fragment_spreads : Dict [SelectionSetNode , List [FragmentSpreadNode ]]
76
- _recursively_referenced_fragments : Dict [OperationDefinitionNode , List [FragmentDefinitionNode ]]
78
+ _recursively_referenced_fragments : Dict [
79
+ OperationDefinitionNode , List [FragmentDefinitionNode ]
80
+ ]
77
81
78
- def __init__ (self , ast : DocumentNode , on_error : Callable [[GraphQLError ], None ]) -> None :
82
+ def __init__ (
83
+ self , ast : DocumentNode , on_error : Callable [[GraphQLError ], None ]
84
+ ) -> None :
79
85
self .document = ast
80
86
self .on_error = on_error # type: ignore
81
87
self ._fragments = None
@@ -112,13 +118,17 @@ def get_fragment_spreads(self, node: SelectionSetNode) -> List[FragmentSpreadNod
112
118
if isinstance (selection , FragmentSpreadNode ):
113
119
append_spread (selection )
114
120
else :
115
- set_to_visit = cast (NodeWithSelectionSet , selection ).selection_set
121
+ set_to_visit = cast (
122
+ NodeWithSelectionSet , selection
123
+ ).selection_set
116
124
if set_to_visit :
117
125
append_set (set_to_visit )
118
126
self ._fragment_spreads [node ] = spreads
119
127
return spreads
120
128
121
- def get_recursively_referenced_fragments (self , operation : OperationDefinitionNode ) -> List [FragmentDefinitionNode ]:
129
+ def get_recursively_referenced_fragments (
130
+ self , operation : OperationDefinitionNode
131
+ ) -> List [FragmentDefinitionNode ]:
122
132
fragments = self ._recursively_referenced_fragments .get (operation )
123
133
if fragments is None :
124
134
fragments = []
@@ -200,7 +210,9 @@ def get_variable_usages(self, node: NodeWithSelectionSet) -> List[VariableUsage]
200
210
self ._variable_usages [node ] = usages
201
211
return usages
202
212
203
- def get_recursive_variable_usages (self , operation : OperationDefinitionNode ) -> List [VariableUsage ]:
213
+ def get_recursive_variable_usages (
214
+ self , operation : OperationDefinitionNode
215
+ ) -> List [VariableUsage ]:
204
216
usages = self ._recursive_variable_usages .get (operation )
205
217
if usages is None :
206
218
get_variable_usages = self .get_variable_usages
0 commit comments