parse.py

import argparse, importlib, dis, subprocess, os, ast
import src.opscodes as ops

parser = argparse.ArgumentParser(add_help=True)
parser.add_argument("-f", type=str, help="file to convert")
args = parser.parse_args()

print("#################### INPUT ####################\n")
subprocess.call("cat pyscripts/%s.py" % (args.f,), shell=True)
print("\n")

with open("pyscripts/" + args.f + ".py") as f:
    c = f.read()
    a = ast.parse(c, args.f, "exec")
    s = compile(a, args.f, "exec")
    # print([ord(b) for b in s.co_code])

# Write PlusCal
b = ops.OpsCode(s, a, args.f)
b.convert()

with open("tla/%s.tla" % (args.f,), "w") as of:
    of.write(b.output.assemble())

print("\n#################### OUTPUT ####################\n")
subprocess.call("cat tla/%s.tla" % (args.f,), shell=True)
print("\n")

# Open Vim to add tests
# cmd =  '%s tla/%s.tla' % (os.environ.get('EDITOR', 'vi'), args.f)
# subprocess.call(cmd, shell=True)

# Translate PlusCal to TLA
# subprocess.call('pcal tla/%s.tla' % (args.f,), shell=True)

# Run Models
# subprocess.call('tlc tla/%s.tla' % (args.f,), shell=True)