profiks-3.py

import sys
from typing import Callable, TypeVar, Generic, Literal

A = TypeVar("A")
B = TypeVar("B")

def profile(f: Callable[A, B], args: A):
    # a log of f's execution
    trajectory = []

    sys.settrace(makeTrace(trajectory))
    result = f(args)
    sys.settrace(emptyTrace)
    return [result, trajectory]

def emptyTrace(_, __, ___):
    return None

def makeTrace(trajectory):
    def go(frame, event: Literal["call", "line", "return", "exception", "opcode"], arg):
        stack = [frame.f_code.co_name]
        while frame.f_back != None:
            frame = frame.f_back
            stack.append(frame.f_code.co_name)

        stack.reverse()
        trajectory.append(stack)
        return go

    return go

def fib(x: int) -> int:
    if x < 2:
        return 1
    return fib(x - 1) + fib(x - 2)

if __name__ == "__main__":
    for i in range(0, 7):
        print(profile(fib, i))