python – 有没有办法将上下文管理器中的源代码作为字符串？

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了python – 有没有办法将上下文管理器中的源代码作为字符串？，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含1079字，纯文字阅读大概需要2分钟。

内容图文

使用inspect.getsourcelines(func)函数接收函数can be的源代码.有没有办法为上下文管理器做同样的事情？

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,之后您可以看到上下文管理器的块.

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

内容总结

以上是互联网集市为您收集整理的python – 有没有办法将上下文管理器中的源代码作为字符串？全部内容，希望文章能够帮你解决python – 有没有办法将上下文管理器中的源代码作为字符串？所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端