使用命令名/选项直接调用distutils'或setuptools'setup()函数,而无需解析命令行?

问题描述 投票:8回答:2

我想以一种稍微不同寻常的方式调用Python的distutils'或setuptools'setup()函数,但我不确定distutils是否适用于这种用法。

举个例子,假设我目前有一个'setup.py'文件,看起来像这样(从distutils文档逐字解码 - setuptools用法几乎相同):

from distutils.core import setup

setup(name='Distutils',
      version='1.0',
      description='Python Distribution Utilities',
      author='Greg Ward',
      author_email='[email protected]',
      url='http://www.python.org/sigs/distutils-sig/',
      packages=['distutils', 'distutils.command'],
     )

通常,为了构建此模块的RPM的.spec文件,我可以运行python setup.py bdist_rpm --spec-only,它解析命令行并调用'bdist_rpm'代码来处理RPM特定的东西。 .spec文件以'./dist'结尾。

如何更改我的setup()调用,以便它使用'--spec-only'选项运行'bdist_rpm'命令,无需解析命令行参数?我可以将命令名称和选项作为参数传递给setup()吗?或者,我可以手动构建命令行,并将其作为参数传递吗?

注意:我已经知道我可以在一个单独的进程中使用实际的命令行,使用os.system()或子进程模块或类似的东西来调用脚本。我试图避免使用任何类型的外部命令调用。我正在寻找一个在当前解释器中运行setup()的解决方案。

对于后台,我将一些发布管理shell脚本转换为单个Python程序。其中一项任务是运行'setup.py'来生成.spec文件,以进行进一步的预发布测试。运行'setup.py'作为外部命令,使用自己的命令行选项,似乎是一个尴尬的方法,它使程序的其余部分复杂化。我觉得可能有更多的Pythonic方式。

python setuptools distutils
2个回答
12
投票

从来没有试过这个,但我确实碰巧看了distutils / core.py,我注意到这在setup()的开头附近:

if 'script_name' not in attrs:
    attrs['script_name'] = os.path.basename(sys.argv[0])
if 'script_args' not in attrs:
    attrs['script_args'] = sys.argv[1:]

因此,看起来好像你可以通过添加“假冒”setup():

setup(
    ...
    script_name = 'setup.py',
    script_args = ['bdist_rpm', '--spec-only']
)

2
投票

只需“伪造”命令行参数 - 例如,启动脚本

import sys

sys.argv[1:] = ['bdist_rpm', '--spec-only']

from distutils.core import setup

setup(name='Distutils',

毕竟,这就是distutils获取命令行参数的方式:它看起来像sys.argv!因此,只需将sys.argv设置为您想要的,并且任何被误导的用户键入的命令行都将被完全忽略。

实际上,您可能想要检查用户是否确实输入了您要忽略的任何参数 - 在修改len(sys.argv) > 1之前输入sys.argv - 并发出警告,或者避免更改sys.argv,或者“合并”用户的内容打字等...但这与你实际要求的完全不同,所以我会把它留在那里;-)。

© www.soinside.com 2019 - 2024. All rights reserved.