simplify cli

Author: keski

graphql-api-generator/generator.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 import argparse
+import os
 from io import UnsupportedOperation
 
 import yaml
@@ -21,13 +22,24 @@ def cmd(args):
     else:
         config = {}
 
-    # concat input files
+    # get list of schema files
+    files = []
+    if os.path.isdir(args.input):
+        for filename in os.listdir(args.input):
+            if filename.endswith(".graphql"):
+                files.append(f'{args.input}/{filename}')
+    else:
+        files = args.input.split(',')
+
+    # build schema
     schema_string = ''
-    for file in args.input.split(','):
+    for file in files:
         with open(file, 'r') as f:
             schema_string += f.read() + '\n'
+    schema = build_schema(schema_string)
 
-    schema = run(build_schema(schema_string), config)
+    # run
+    schema = run(schema, config)
 
     # write to file or stdout
     if args.output:
@@ -186,11 +198,12 @@ def drop_comments(schema):
 if __name__ == '__main__':
     parser = argparse.ArgumentParser()
     parser.add_argument('--input', type=str, required=True,
-                        help='Input schema files (separated by commas)')
+                        help='GraphQL DB schema files (separated by commas), or a path to a schema directory')
     parser.add_argument('--output', type=str,
                         help='Output schema file (default stdout)')
     parser.add_argument('--config', type=str,
                         help='Path to configuration file')
+
     cmd(parser.parse_args())
 
 

graphql-api-generator/main.py

@@ -1,21 +1,5 @@
 #!/usr/bin/env python3
-import yaml
-from graphql import build_schema
-
-import generator
-
-config_file = './resources/config.yml'
-input_files = './resources/schema.graphql'
-
-config = ''
-with open(config_file) as f:
-    config = yaml.safe_load(f)
+import os
 
 # input files
-schema_string = ''
-for file in input_files.split(','):
-    with open(file, 'r') as f:
-        schema_string += f.read() + '\n'
-
-schema = generator.run(build_schema(schema_string), config)
-print(generator.print_schema(schema))
+os.system('python3 generator.py --input ./resources --config ./resources/config.yml')