Skip to content

Commit 27e6bd2

Browse files
committed
Improve example script
1 parent 302577e commit 27e6bd2

File tree

1 file changed

+10
-3
lines changed

1 file changed

+10
-3
lines changed

examples/visitor_example.py

+10-3
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ def visit_document(self, node):
1414
return 1
1515

1616
def visit_operation_definition(self, node):
17-
print(node.__class__, node.get_operation(), node.get_name().get_value())
17+
print(node.__class__, node.get_operation())
1818
return 1
1919

2020
def visit_variable_definition(self, node):
@@ -65,7 +65,14 @@ def visit_int_value(self, node):
6565
print(node.__class__, node.get_value())
6666
return 1
6767

68+
def visit_string_value(self, node):
69+
print(node.__class__, node.get_value())
70+
return 1
71+
6872
if __name__ == '__main__':
69-
query = sys.argv[1]
70-
node = GraphQLParser.graphql_parse_string(query)
73+
import argparse
74+
parser = argparse.ArgumentParser(description='GraphQL parser example')
75+
parser.add_argument('query')
76+
args = parser.parse_args()
77+
node = GraphQLParser.graphql_parse_string(args.query)
7178
Visitor().visit_node(node)

0 commit comments

Comments
 (0)