python - 在 Python 中调用另一个已安装的包

Question

我是 Python 新手。我想要做的是在我的应用程序中设置esptool包 ( pip install esptool) 用一堆参数调用它的 main 方法。就像是：

esptool.py -p /dev/ttyUSB0 write_flash -fm qio 0x0000

我遇到了一个问题。esptool不在 python 中要导入的包列表中（它已经与 pip 一起安装）。我将如何使用import并调用主要方法？

标签： pythonpip

Answer 1

解决导入问题

您不能简单地调用import esptool，因为esptool.py它是一个可执行脚本，因此不能像普通模块一样被导入。但是，有一些解决方法可以从可执行脚本中导入代码；这是我知道的两个：

延伸sys.path

您可以扩展sys.path以包含包含esptool.py脚本的 bindir。从命令行进行简单检查：

$ PYTHONPATH=$(which esptool.py)/.. python -c "import esptool; esptool.main()"

应该打印你的使用帮助文本。

sys.path在代码中扩展：

import os
import sys

try:
    from shutil import which
except ImportError:
    from distutils.spawn import find_executable as which


bindir = os.path.dirname(which('esptool.py'))
sys.path.append(bindir)  # after this line, esptool becomes importable

import esptool


if __name__ == '__main__':
    esptool.main()

使用进口机械

sys.path您可以通过使用从任意文件导入 Python 代码的机制来避免扩展。我喜欢这个解决方案，而不是摆弄 . sys.path，但不幸的是，它不能在 Python 2 和 3 之间移植。

Python 3.5+

import importlib.machinery
import importlib.util

from shutil import which


if __name__ == '__main__':
    loader = importlib.machinery.SourceFileLoader('esptool', which('esptool.py'))
    spec = importlib.util.spec_from_loader(loader.name, loader)
    esptool = importlib.util.module_from_spec(spec)
    loader.exec_module(esptool)  # after this line, esptool is imported

    esptool.main()

蟒蛇 2.7

import imp
from distutils.spawn import find_executable as which


if __name__ == '__main__':
    esptool = imp.load_source('esptool', which('esptool.py'))
    esptool.main()

传递命令行参数

命令行参数存储在sys.argv列表中，因此您必须临时覆盖它才能将参数传递给主函数：

# assuming esptool is imported
import sys

if __name__ == '__main__':
    # save the current arguments
    argv_original = sys.argv[:]
    # overwrite arguments to be passed to esptool argparser
    sys.argv[:] = ['', '-p', '/dev/ttyUSB0', 'write_flash', '-fm', 'qio', '0x0000']
    try:
        esptool.main()
    except Exception:
        # TODO deal with errors here
        pass
    finally:  # restore original arguments
        sys.argv[:] = argv_original