From 4f63fb9dfc8cfa156ec0e932ff2f5a98af517a9e Mon Sep 17 00:00:00 2001 From: Antoine Viallon Date: Fri, 12 Apr 2024 16:43:08 +0200 Subject: [PATCH] main: enable using files as input --- compiler/__main__.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/compiler/__main__.py b/compiler/__main__.py index 97ca040..fbee99e 100644 --- a/compiler/__main__.py +++ b/compiler/__main__.py @@ -33,6 +33,7 @@ def main(): valid_mock_inputs = list(filter(lambda x: x[0] != "_", dir(mock))) parser.add_argument("--mock-input", dest="mock_input", type=str, choices=valid_mock_inputs, default="legacy_mock", help="Specify mock input") + parser.add_argument(dest="sources", metavar="SOURCE_FILES", help="Input files", type=Path, nargs="*") args = parser.parse_args() @@ -43,6 +44,10 @@ def main(): print("Source:\n", data) _input_stream = io.StringIO(data) + if len(args.sources): + assert len(args.sources) == 1 + _input_stream = io.open(args.sources[0]) + input_stream = TextIOWithMemory(_input_stream) tokens = \