import pyximport

pyximport.install()

import _line
import asyncio


async def f():
    pass


print(_line.lines(f))


async def function_1(bool):
    if bool:
        async for _ in function_2():
            pass


async def function_2():
    if False:
        yield


_line.start()
asyncio.run(function_1(True))
_line.stop()

cdef extern from "Python.h":
    ctypedef struct PyObject
    ctypedef struct PyCodeObject
    ctypedef struct PyFrameObject
    ctypedef int (*Py_tracefunc)(PyObject* obj, PyFrameObject* frame, int what, PyObject* arg)
    cdef int PyTrace_CALL
    cdef int PyTrace_LINE
    cdef int PyTrace_RETURN
    cdef int PyCode_Addr2Line(PyCodeObject *co, int byte_offset)
    cdef void PyEval_SetTrace(Py_tracefunc func, PyObject* obj)
    cdef int PyFrame_GetLineNumber(PyFrameObject* frame)


def lines(function):
    code = function.__code__
    return [PyCode_Addr2Line(<PyCodeObject*>code, byte_offset) for byte_offset in range(0, len(code.co_code), 2)]


def start():
    PyEval_SetTrace(trace, NULL)


def stop():
    PyEval_SetTrace(NULL, NULL)


cdef int trace(PyObject* obj, PyFrameObject* frame, int what, PyObject* arg) noexcept:
    print(f"{PyFrame_GetLineNumber(frame):3}", {PyTrace_CALL: "call", PyTrace_LINE: "line", PyTrace_RETURN: "return"}.get(what, "other"))