有没有办法以字符串形式获取上下文管理器中的源代码？

函数源码can be https://stackoverflow.com/a/427533/1113207收到与inspect.getsourcelines(func)功能。有没有办法为上下文管理器做同样的事情？

with test():
    print('123')

# How to get "print('123')" as line here?

您对此解决方案有何看法？

import traceback


class ContextManagerContent(object):

    def __enter__(self):
        return

    def __exit__(self, _type, value, _traceback):
        stack = traceback.extract_stack()
        f, last_line = self._get_origin_info(stack)

        with open(f) as fin:
            lines = list(fin)

        search = 'with {cls_name}'.format(cls_name=self.__class__.__name__)
        for i, x in enumerate(lines[:last_line + 1][::-1]):
            if search in x:
                first_line = len(lines) - i
                break

        selected_lines = lines[first_line:last_line + 1]
        print ''.join(selected_lines)

    def _get_origin_info(self, stack):
        origin = None
        for i, x in enumerate(stack[::-1]):
            if x[2] == '__exit__':
                origin = stack[::-1][i + 1]
                break

        return origin[0], origin[1] - 1


with ContextManagerContent():
    print '123'
    print '456'
    print '789'

如果您将其保存在.py文件并运行它，你可以看到打印了数字 123、456 和 789，之后你可以看到上下文管理器的块。

请注意，我没有处理可能的异常或输出格式，并且某些部分可以改进，但我认为这是一个很好的起点。