From 5bf52edd4411cfda1cbbd5527678af374d5ae7f8 Mon Sep 17 00:00:00 2001 From: Antoine Viallon Date: Wed, 10 May 2023 01:37:42 +0200 Subject: [PATCH] main: allow giving custom input to compiler --- compiler/__main__.py | 15 ++++++++++++--- 1 file changed, 12 insertions(+), 3 deletions(-) diff --git a/compiler/__main__.py b/compiler/__main__.py index 556fd32..707b40d 100644 --- a/compiler/__main__.py +++ b/compiler/__main__.py @@ -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)