pedroqueiroga/minimal-ast

extracts minimal working AST for a line of python code that belongs to some source code

The idea is to go from a traceback to a nice context of the traceback. For example, traceback in buggy_script.py is

  File "buggy_script.py", line 26, in calculate
    return res
NameError: name 'res' is not defined

Feed a python script and a method to minimal extractor, and it should return the minimal source code that is required to run the method in the script.

For example, take example01.py:

import sys
import fire


def add_numbers(a, b):
    return a + b


def multiply_numbers(a, b):
    return a * b


def divide_numbers(a, b):
    return a / b


def calculate(operation, num1, num2):
    if operation == "add":
        result = add_numbers(num1, num2)
    else:
        print("Invalid operation")

    return res


if __name__ == "__main__":
    fire.Fire(calculate)

returns when extracted with 'calculate'

def add_numbers(a, b):
    return a + b

def calculate(operation, num1, num2):
    if operation == 'add':
        result = add_numbers(num1, num2)
    else:
        print('Invalid operation')
    return res