调用pip install my_module时能否执行指定Python代码？

当然可以实现！当你用pip install my_module安装wheel包时，确实有几种可靠的方式触发自定义代码的执行，下面是最常用的两种方案：

方案1：自定义安装命令（基于setup.py） #

你可以通过重写setuptools的安装命令，在标准安装流程完成后插入自己的代码逻辑。具体操作如下：

在你的包的setup.py文件中添加自定义安装类：

from setuptools import setup, find_packages
from setuptools.command.install import install
import subprocess
import sys

class PostInstallCommand(install):
    def run(self):
        # 先执行pip默认的安装流程
        install.run(self)
        # 这里添加你要执行的自定义代码
        print("✅ my_module安装完成，开始执行后续逻辑...")
        # 示例：运行你预先准备好的脚本
        subprocess.run([sys.executable, "my_module/scripts/post_install.py"])

setup(
    name="my_module",
    version="0.1.0",
    packages=find_packages(),
    cmdclass={
        'install': PostInstallCommand,
    }
)

构建wheel包时用python setup.py bdist_wheel，这样生成的wheel安装时就会自动执行你定义的后续代码。

方案2：用户主动触发的控制台命令（更安全推荐） #

如果是公开分发的包，自动执行代码可能会引发用户的安全顾虑。这种情况下，更推荐用entry_points定义一个控制台命令，让用户安装后可以主动触发：

在setup.py中添加入口配置：

setup(
    # 其他基础配置...
    entry_points={
        'console_scripts': [
            'my-module-setup=my_module.post_install:main',
        ]
    }
)

然后在你的包内创建my_module/post_install.py文件：

def main():
    print("开始执行my_module的初始化配置...")
    # 这里编写你的自定义逻辑，比如配置文件生成、环境检查等

if __name__ == "__main__":
    main()

安装完成后，用户只需要在终端输入my-module-setup就能执行你的代码，这种方式更透明，也符合Python包分发的安全最佳实践。

关键注意事项 #

• 权限与环境：如果用户用--user参数或虚拟环境安装，要确保你的代码路径、权限适配当前环境，避免出现权限错误。
• 静默安装场景：如果用户用-q/--quiet参数安装，你的打印信息会被隐藏，必要时可以把日志写入文件。
• 安全风险：自动执行代码只适合内部或信任场景，公开包尽量避免强制自动执行，尊重用户的选择权。

内容的提问来源于stack exchange，提问作者Quasar