pymips_main.py

import os
import select
import sys
MIN_PYTHON = (3, 6)
if sys.version_info < MIN_PYTHON:
    print("Require python" +
          str(MIN_PYTHON[0]) + "." + str(MIN_PYTHON[1]) +
          " got " + str(sys.version_info.major) + "." + str(sys.version_info.minor))
    sys.exit(1)

import argparse

from pathlib import Path
from compiler.compiler import compile_file, compile_src

if __name__ == "__main__":

    parser = argparse.ArgumentParser(description='')
    parser.add_argument('-i', '--input', dest='input',
                        help='folder or file for scripts', required=False)
    parser.add_argument('--debug', dest='debug', default=False,
                        help='Output additional debug information for the code', action='store_true')
    args = parser.parse_args()
    if not sys.stdin.isatty() and not os.name == 'nt':
        # TODO fix stdin for windows
        src = sys.stdin.read()
        print(compile_src(src, args.debug))
    elif args.input:
        a = Path(args.input)

        if Path.is_dir(a):
            for child in a.iterdir():
                if child.suffix == '.py':
                    compile_file(child, args.debug)
        elif a.suffix == '.py':
            compile_file(a, args.debug)
    else:
        print("Nothing on stdin and no --input given")
        sys.exit(1)