python在临时fi中导入带有源代码的模块

Question

import tempfile

tmp = tempfile.NamedTemporaryFile(delete=True)
try:
    # do stuff with temp
    tmp.write(b'def fun():\n\tprint("hello world!")\n')
    if __name__ == '__main__':
        func = __import__(tmp.name)
        func.fun()
finally:
    tmp.close()  # deletes the file

所以我想创建一个临时文件，添加一些源代码，然后导入模块并调用函数，但我总是遇到这样的错误：

^{pr2}$

它似乎找不到临时文件的模块。我怎么解决这个问题？在

Answer 1

您的代码存在一些问题：

1. 文件名不以.py结尾，但Python模块应该以.py结尾。您可以通过在NamedTemporaryFile()中设置suffix='.py'来解决这个问题。在
2. __import__()不是从完整路径加载模块的正确方法。请看这里：How to import a module given the full path?
3. 在写入文件之后和导入之前不会刷新，因此即使Python确实找到了该文件，它也可能是空的。在写入后添加tmp.flush()以修复此问题。在