81 lines
2.4 KiB
Python
Executable File
81 lines
2.4 KiB
Python
Executable File
#!/bin/python3
|
|
import argparse
|
|
from os import system,popen
|
|
from glob import glob
|
|
|
|
def run_input_cases(program, verbose, output):
|
|
input_files = glob("./*.in")
|
|
if len(input_files) < 1:
|
|
|
|
return system(program)
|
|
for input_file in input_files:
|
|
print(f"Running {input_file}")
|
|
end_command = f"{program} < {input_file}"
|
|
if verbose:
|
|
print("Input:")
|
|
system(f"cat {input_file}")
|
|
print("")
|
|
if (out := glob(input_file.replace(".in", ".out"))[0]):
|
|
if verbose:
|
|
print("Expected output:")
|
|
system(f"cat {out}")
|
|
print("")
|
|
if output:
|
|
end_command += f" | diff {out} -"
|
|
print(popen(end_command).read())
|
|
|
|
print("-----------")
|
|
|
|
return
|
|
|
|
def python_compile(file_name, verbose=False):
|
|
run_input_cases("python3 " + file_name, verbose)
|
|
return
|
|
|
|
def java_compile(file_name, maven=False, verbose=False):
|
|
if maven:
|
|
system("mvn clean verify")
|
|
system("java -jar target/*.jar")
|
|
return
|
|
|
|
system("javac " + file_name)
|
|
run_input_cases("java " + file_name[:file_name.rfind(".")], verbose)
|
|
return
|
|
|
|
def c_compile(file_name, verbose=False, output=False):
|
|
system("gcc -Wall -pedantic --std=c99 -g -o program -lm -Wno-unused-result " + file_name)
|
|
run_input_cases("./program", verbose, output)
|
|
return
|
|
|
|
def main():
|
|
SUPPORTED_LANGS = [".c", ".py"]
|
|
parser = argparse.ArgumentParser(
|
|
prog='UCompiler',
|
|
description='Compiles based on language',
|
|
epilog='LOL')
|
|
parser.add_argument("filename", metavar="N", type=str)
|
|
parser.add_argument("-v", "--verbose", action="store_true", default=False)
|
|
parser.add_argument("-a", "--alternative", action="store_true", default=False)
|
|
parser.add_argument("-o", "--output", action="store_true", default=False)
|
|
|
|
args = parser.parse_args()
|
|
file_name = args.filename
|
|
file_ext = file_name[file_name.rfind("."):]
|
|
if file_ext not in SUPPORTED_LANGS:
|
|
print("Unsupported language")
|
|
return
|
|
|
|
if file_ext == ".c":
|
|
c_compile(file_name, args.verbose, args.output)
|
|
elif file_ext == ".py":
|
|
python_compile(file_name, args.verbose)
|
|
elif file_ext == ".java":
|
|
java_compile(file_name, args.verbose, maven=args.alternative)
|
|
return
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|
|
|
|
|