使用Python setuptools安装后脚本

Question

是否可以将安装后的Python脚本文件指定为setuptools setup.py文件的一部分，以便用户可以运行该命令：python setup.py install在本地项目文件存档，或pip install <name>对于PyPI项目，脚本将在标准setuptools安装完成后运行？我希望执行可以在单个Python脚本文件中编码的安装后任务（例如，向用户提供自定义安装后消息，从其他远程源存储库中提取其他数据文件）。几年前我遇到了这个回答这个主题的答案，听起来好像当时的共识是你需要创建一个install子命令。如果仍然如此，那么有人可以提供一个如何执行此操作的示例，以便用户无需输入第二个命令来运行脚本吗？

Answer 1

当安装后脚本要求已经安装了软件包依赖项时，这是唯一对我有用的策略：

import atexit

from setuptools.command.install import install

def _post_install():

    print('POST INSTALL')

class new_install(install):

    def __init__(self, *args, **kwargs):

        super(new_install, self).__init__(*args, **kwargs)

        atexit.register(_post_install)

setuptools.setup(

    cmdclass={'install': new_install},