python - 如何从 setup.py 中提取依赖信息

Question

我有一个 python 项目，我们称它为 foobar，像所有 Python 项目一样，项目根目录中有一个 setup.py 脚本。例如

• foobar
  • setup.py

setup.py文件内容:

from ez_setup import use_setuptools
use_setuptools()

from setuptools import setup, find_packages
setup(
    name='foobar',
    version='0.0.0',
    packages=find_packages(),
    install_requires=[
        'spam==1.2.3',
        'eggs>=4.5.6',
    ],
)

我需要使用 Python 从该 setup.py 文件中获取依赖项信息。我想要的部分是

[
    'spam==1.2.3',
    'eggs>=4.5.6',
]

在上面的例子中。我不想安装这个包，我需要的只是依赖信息。当然，我可以使用正则表达式来解析它，但那样会很丑陋，我也可以使用 Python AST 来解析它，但我认为应该已经有一些工具可以做到这一点。最好的方法是什么？

Answer 1

在我看来，你可以使用 mock 来完成这项工作(假设你已经安装了它并且你拥有所有 setup.py 要求...... ).这里的想法是模拟 setuptools.setup 并检查调用它的参数。当然，您实际上并不需要 mock 来执行此操作——如果需要，您可以直接修改 setuptools ...

import mock  # or `from unittest import mock` for python3.3+.
import setuptools

with mock.patch.object(setuptools, 'setup') as mock_setup:
    import setup  # This is setup.py which calls setuptools.setup

# called arguments are in `mock_setup.call_args`
args, kwargs = mock_setup.call_args
print kwargs.get('install_requires', [])