python – 如何记录在代码块中执行的所有导入？

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了python – 如何记录在代码块中执行的所有导入？，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含1802字，纯文字阅读大概需要3分钟。

内容图文

我正在编写测试套件,我正在测试的代码过度使用延迟模块导入.因此,对于相同方法的5个不同输入,这可能最终导入5个额外模块.我希望能够做的是设置测试,以便我可以断言用一个输入运行方法会导致一次导入,而不会导致其他4.

我有一些关于如何开始这个的想法,但到目前为止都没有成功.我已经有了一个自定义导入器,我可以将日志代码放入导入器中.但这不起作用,因为import语句只运行一次.无论先前是否已导入模块,我都需要执行日志语句.刚运行del sys.modules [‘modname’]也不起作用,因为它在测试代码中运行,我无法在被测试的代码中重新加载模块.

我尝试的下一件事是继承dict来进行监视,并用这个子类替换sys.modules.此子类具有重新实现的__getitem__方法,但调用import模块似乎不会触发子类中的__getitem__调用.我也不能直接分配给sys.modules .__ getitem__,因为它是只读的.

我正在努力做甚么可能吗？

UPDATE

nneonneo的答案似乎只有在logImports()的实现与使用它的模块相同时才有效.如果我创建一个包含此功能的基础测试类,则会出现问题.首先是它找不到__import__,错误：

#     old_import = __import__
# UnboundLocalError: local variable '__import__' referenced before assignment

当我将其更改为__builtin __.__ import__时,我又出现了另一个错误：

myunittest.py：

import unittest
class TestCase(unittest.TestCase):
    def logImports(self):
        old_import = __builtins__.__import__
        def __import__(*args, **kwargs):
            print args, kwargs
            return old_import(*args, **kwargs)

        __builtins__.__import__ = __import__

test.py：

import myunittest
import unittest
class RealTest(myunittest.TestCase):
    def setUp(self):
        self.logImports()
    def testSomething(self):
        import unittest
        self.assertTrue(True)
unittest.main()

#     old_import = __builtins__.__import__
# AttributeError: 'dict' object has no attribute '__import__'

解决方法:

尝试

old_import = __import__
def __import__(*args, **kwargs):
    print args, kwargs
    return old_import(*args, **kwargs)

__builtins__.__import__ = __import__

这会完全覆盖__import__,允许您监视每次导入调用.

内容总结

以上是互联网集市为您收集整理的python – 如何记录在代码块中执行的所有导入？全部内容，希望文章能够帮你解决python – 如何记录在代码块中执行的所有导入？所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

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

内容手机端