main: allow giving custom input to compiler

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

View file

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