diff options
Diffstat (limited to 'projects/hackc/__main__.py')
-rw-r--r-- | projects/hackc/__main__.py | 13 |
1 files changed, 10 insertions, 3 deletions
diff --git a/projects/hackc/__main__.py b/projects/hackc/__main__.py index a7bc06f..1cb5c3f 100644 --- a/projects/hackc/__main__.py +++ b/projects/hackc/__main__.py @@ -4,7 +4,7 @@ import os from .parser import Parser -def compile_jack(input_path: Path, verbose: bool): +def compile_jack(input_path: Path, extensions: list, verbose: bool): try: filenames = os.listdir(input_path) files = [Path(input_path / f) for f in filenames] @@ -16,7 +16,7 @@ def compile_jack(input_path: Path, verbose: bool): return for input_fn in jack_files: - parser = Parser(input_fn) + parser = Parser(input_fn, extensions=extensions) parser.tokenize() parser.print_tokens() @@ -24,6 +24,13 @@ def compile_jack(input_path: Path, verbose: bool): if __name__ == "__main__": parser = ArgumentParser("hackc") parser.add_argument("-v", "--verbose", action="store_true", help="verbose mode") + parser.add_argument( + "-E", + "--extension", + metavar="extension", + action="append", + help="enable extension of standard Jack", + ) parser.add_argument("input_path", help="Jack file or directory") args = parser.parse_args() - compile_jack(Path(args.input_path), args.verbose) + compile_jack(Path(args.input_path), args.extension or [], args.verbose) |