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

main: allow giving custom input to compiler

Browse source
This commit is contained in:
Antoine Viallon 2023-05-10 01:37:42 +02:00
parent 017aefa750
commit 5bf52edd44
Signed by: aviallon
GPG key ID: D126B13AB555E16F
1 changed files with 12 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

9
compiler/__main__.py
View file

@ -1,5 +1,6 @@
from __future__ import annotations from __future__ import annotations
import argparse
import sys import sys
from pprint import pprint from pprint import pprint
@ -11,10 +12,18 @@ from .tokenizer import Tokenizer, Tokens
def main(): def main():
parser = argparse.ArgumentParser()
parser.add_argument("--mock", action="store_true", help="Use default test input")
args = parser.parse_args()
data = """ data = """
2 + 8 - 1 * (byte = 3 + 5) 2 + 8 - 1 * (byte = 3 + 5)
/ (byte = 255) + byte / (byte = 255) + byte
""" """
if not args.mock:
data = sys.stdin.read().strip()
print("Source:\n", data) print("Source:\n", data)
tokenizer = Tokenizer() tokenizer = Tokenizer()
tokens = tokenizer.tokenize(data) tokens = tokenizer.tokenize(data)