def process_answer(answer, extraction = True): code = extract_code(answer) code = re.sub("\$.*\$", "", code) try: data = pyth.general_parse(code)[1] except Exception as e: print(code) print(e) data = {i: [0]*3 for i in "bdGHkNTYZ"} data = {i: np.array(v + [v[0]+v[2]]) for i, v in data.items()} data["Code"] = code return pd.Series(data)
#!/usr/bin/env python3 import os import sys sys.path.append(os.path.join(os.path.dirname(os.path.realpath(__file__)), 'pyth')) import pyth import readline import traceback while True: try: exec(pyth.general_parse(input('>>> ')), pyth.environment) except EOFError: break except: traceback.print_exc()