Expose include_dirs in python tools

Author: zfergus

docs/spec2html.py

@@ -27,17 +27,22 @@ def rules_from_pointer(self, pointer):
         "--output", "-o", help="Path to the output html file", required=True)
     parser.add_argument(
         "--title", "-t", help="Title", default="JSON Specification")
+    parser.add_argument(
+        "--include-dirs", nargs="+", help="Directories to search for includes",
+        default=None)
     args = parser.parse_args()
 
+    if args.include_dirs is None:
+        args.include_dirs = [pathlib.Path(args.spec_path).parent]
+
     # Template Lookup
     mylookup = TemplateLookup(directories=['.'])
 
     print("01 - Loading specification")
     with open(args.input) as input_file:
         input = json.load(input_file)
 
-    # TODO: Expose the include path as a parameter
-    input = inject_include(input, [pathlib.Path(args.input).parent])
+    input = inject_include(input, args.include_dirs)
 
     # print(input)
 

scripts/generate_defaults.py

@@ -51,13 +51,18 @@ def main():
     parser.add_argument(
         "-o,--output", dest="output", help="Path to the output file",
         default="defaults.json")
+    parser.add_argument(
+        "--include-dirs", nargs="+", help="Directories to search for includes",
+        default=None)
     args = parser.parse_args()
 
+    if args.include_dirs is None:
+        args.include_dirs = [pathlib.Path(args.spec_path).parent]
+
     with open(args.spec_path) as f:
         spec = json.load(f)
 
-    # TODO: Expose the include path as a parameter
-    spec = inject_include(spec, [pathlib.Path(args.spec_path).parent])
+    spec = inject_include(spec, args.include_dirs)
 
     defaults = generate_defaults(spec)