Skip to content

Commit 3ea2c91

Browse files
committed
Slight improvement of ensuring default directives.
1 parent 2027aad commit 3ea2c91

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

graphql/utils/build_ast_schema.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -247,14 +247,15 @@ def make_input_object_def(definition):
247247
directives = [get_directive(d) for d in directive_defs]
248248

249249
# If skip and include were not explicitly declared, add them.
250-
filtered_default_directive = ''.join(directive.name for directive in directives if directive.name == 'skip' or
251-
directive.name == 'include')
250+
filter_directives = (directive.name for directive in directives if directive.name == 'skip' or
251+
directive.name == 'include')
252+
possible_default_directives = ''.join((next(filter_directives, ''), next(filter_directives, '')))
252253
default_directives = {
253254
'skip': [GraphQLIncludeDirective],
254255
'include': [GraphQLSkipDirective],
255256
'': [GraphQLIncludeDirective, GraphQLSkipDirective]
256257
}
257-
directives.extend(default_directives.get(filtered_default_directive, []))
258+
directives.extend(default_directives.get(possible_default_directives, []))
258259

259260
schema_kwargs = {'query': get_object_type(ast_map[query_type_name])}
260261

0 commit comments

Comments
 (0)