aviallon/compiler
1
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

main: enable using files as input

Browse source
This commit is contained in:
Antoine Viallon 2024-04-12 16:43:08 +02:00
parent 368629caa5
commit 4f63fb9dfc
Signed by: aviallon
GPG key ID: 186FC35EDEB25716
1 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

5
compiler/__main__.py
View file

@ -33,6 +33,7 @@ def main():
valid_mock_inputs = list(filter(lambda x: x[0] != "_", dir(mock))) 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", parser.add_argument("--mock-input", dest="mock_input", type=str, choices=valid_mock_inputs, default="legacy_mock",
help="Specify mock input") help="Specify mock input")
parser.add_argument(dest="sources", metavar="SOURCE_FILES", help="Input files", type=Path, nargs="*")
args = parser.parse_args() args = parser.parse_args()
@ -43,6 +44,10 @@ def main():
print("Source:\n", data) print("Source:\n", data)
_input_stream = io.StringIO(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) input_stream = TextIOWithMemory(_input_stream)
tokens = \ tokens = \