讲师中心 微信公众号
AI工具推荐 视频效率加速

Python 多入口脚本的组织方式

阿雪姑娘_1436

阿雪姑娘_1436

发布时间:2026-02-25 20:07:03

|

254人浏览过

|

来源于php中文网

原创

多个 if name == "__main__" 不能共存,Python 只执行最后一个;正确做法是统一入口、封装函数并用 argparse 或环境变量分发,或通过 setuptools entry_points 管理多命令。

python 多入口脚本的组织方式

多个 if __name__ == "__main__" 怎么共存

不能共存,Python 解释器只认最后一个。常见错误是复制粘贴不同脚本时各自保留了 if __name__ == "__main__" 块,结果只有末尾那个生效,前面的逻辑彻底静默。

正确做法是统一入口:把所有主逻辑封装成函数,比如 main_cli()、main_worker()、main_dev(),然后在单个 if __name__ == "__main__" 里用 sys.argv 或 argparse 分发:

if __name__ == "__main__":
    import sys
    if len(sys.argv) > 1 and sys.argv[1] == "worker":
        main_worker()
    elif len(sys.argv) > 1 and sys.argv[1] == "dev":
        main_dev()
    else:
        main_cli()
  • 别用 os.system("python script2.py") 启动另一个入口——进程隔离、参数传递麻烦、异常难捕获
  • argparse 比手动查 sys.argv 更健壮,尤其要支持 --help 和子命令时
  • 如果入口差异大(比如一个跑 HTTP 服务,一个跑定时任务),建议拆成独立脚本文件,而非硬塞进一个文件

用 setuptools 配 entry_points 管多个命令

这是发布级组织方式,适合打包后通过命令行直接调用,比如 mytool cli、mytool sync。核心是 setup.py 或 pyproject.toml 里声明 entry_points。

在 pyproject.toml 中写:

立即学习“Python免费学习笔记(深入)”;

[project.entry-points."console_scripts"]
mytool-cli = "mytool.cli:main"
mytool-sync = "mytool.sync:main"
mytool-clean = "mytool.utils:clean_cache"
  • 每个值格式必须是 "模块路径:可调用对象名",中间冒号不能漏,也不能多空格
  • 模块路径从包根开始算,不是相对路径;确保该模块能被 Python 正常 import(即有 __init__.py 或符合 PEP 420)
  • pip install -e . 后命令才可用;不加 -e 就得先 pip install . 再测试
  • Windows 上偶尔因 shebang 或路径缓存导致新命令不生效,可删掉 %USERPROFILE%\AppData\Roaming\pip\pip.ini 或重启终端

同一个脚本里混用 click 和原生 argparse

别混。两者都试图接管 sys.argv 解析和帮助输出,叠加会导致参数错乱、--help 重复打印、子命令注册失败等现象,错误信息往往不明确,比如 unrecognized arguments: --help 却又没提示可用选项。

Secure Python Code Sandbox
Secure Python Code Sandbox

安全 Python 代码沙箱:在隔离的沙箱中执行任意 Python 代码,预装 requests、NumPy、pandas 库。返回 stdout、stderr。

下载

选一个到底:

  • 简单场景(1–2 个开关参数)用原生 argparse,零依赖,启动快
  • 需要子命令、选项分组、类型自动转换、颜色帮助页,就用 click;它对多入口更友好,@click.group() 天然支持 mytool db migrate 这种嵌套调用
  • 如果已有 argparse 脚本想升级为多命令,不要强行套 click,而是把各子命令逻辑抽成函数,再用 click.group() 包一层,而不是保留旧的 ArgumentParser 实例

调试时怎么快速切换不同入口而不改代码

靠环境变量或临时配置最稳。硬编码判断 sys.argv 在 IDE 里调试不方便,每次都要改运行参数;而环境变量可以全局设置一次,所有入口都读取。

例如在脚本开头加:

import os
ENTRY_POINT = os.getenv("MYTOOL_ENTRY", "cli")

然后在 if __name__ == "__main__" 里根据 ENTRY_POINT 分发。调试时只需在 IDE 的 Run Configuration 里设环境变量,或终端里:

MYTOOL_ENTRY=worker python mytool/__main__.py
  • 避免用文件存在与否判断入口(如 if os.path.exists("dev.flag")),容易误提交、权限出问题、CI 环境不一致
  • 别用 __file__ 名称匹配来区分入口,重命名脚本就失效,且无法支持 entry_points 安装后的调用
  • 如果入口间共享大量初始化逻辑(如日志配置、DB 连接),把这些提前抽到单独函数,所有入口都先调用它,而不是复制粘贴

多入口真正的复杂点不在写法,而在状态隔离——比如一个入口开了全局线程池,另一个入口又去关它,或者两个入口同时读写同一份缓存文件。这种问题不会报错,但会悄悄破坏行为一致性。

热门AI工具

更多
二狗PPT
二狗PPT Hot

一款AI演示文稿工具,主要用于专为中式职场打造的AI PPT生成工具,适合需要提升相关任务效率的用户。

讯飞绘文

讯飞绘文是一款由科大讯飞推出的一站式 AIGC 内容运营平台。

小橡皮AI

一款AI工具,主要用于小橡皮一键去AI味,将机器味重的内容改得更有活人感,防止因为 AI 内容被平台打上 AI 标签,以及限流,适合需要提升相关任务效率的用户。

Seko
Seko Hot

一款AI视频创作工具,主要用于商汤科技推出的创编一体的AI短视频创作Agent,适合需要提升相关任务效率的用户。

PixPix
PixPix Hot

PixPix是一款面向电商视觉生产的AI商品图生成工具。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

超级麦吉

一款通过数字员工协作完成调研、数据分析与文档制作的AI办公平台。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

相关专题

更多
pip安装使用方法
pip安装使用方法

安装步骤:1、确保Python已经正确安装在您的计算机上;2、下载“get-pip.py”脚本;3、按下Win + R键,然后输入cmd并按下Enter键来打开命令行窗口;4、在命令行窗口中,使用cd命令切换到“get-pip.py”所在的目录;5、执行安装命令;6、验证安装结果即可。大家可以访问本专题下的文章,了解pip安装使用方法的更多内容。

3250

2023.10.09

更新pip版本
更新pip版本

更新pip版本方法有使用pip自身更新、使用操作系统自带的包管理工具、使用python包管理工具、手动安装最新版本。想了解更多相关的内容,请阅读专题下面的文章。

7423

2024.12.20

pip设置清华源
pip设置清华源

设置方法:1、打开终端或命令提示符窗口;2、运行“touch ~/.pip/pip.conf”命令创建一个名为pip的配置文件;3、打开pip.conf文件,然后添加“[global];index-url = https://pypi.tuna.tsinghua.edu.cn/simple”内容,这将把pip的镜像源设置为清华大学的镜像源;4、保存并关闭文件即可。

9062

2024.12.23

python升级pip
python升级pip

本专题整合了python升级pip相关教程,阅读下面的文章了解更多详细内容。

4687

2025.07.23

线程和进程的区别
线程和进程的区别

线程和进程的区别:线程是进程的一部分,用于实现并发和并行操作,而线程共享进程的资源,通信更方便快捷,切换开销较小。本专题为大家提供线程和进程区别相关的各种文章、以及下载和课程。

4458

2023.08.10

windows查看端口占用情况
windows查看端口占用情况

Windows端口可以认为是计算机与外界通讯交流的出入口。逻辑意义上的端口一般是指TCP/IP协议中的端口,端口号的范围从0到65535,比如用于浏览网页服务的80端口,用于FTP服务的21端口等等。怎么查看windows端口占用情况呢?php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

3319

2023.07.26

查看端口占用情况windows
查看端口占用情况windows

端口占用是指与端口关联的软件占用端口而使得其他应用程序无法使用这些端口,端口占用问题是计算机系统编程领域的一个常见问题,端口占用的根本原因可能是操作系统的一些错误,服务器也可能会出现端口占用问题。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2898

2023.07.27

windows照片无法显示
windows照片无法显示

当我们尝试打开一张图片时,可能会出现一个错误提示,提示说"Windows照片查看器无法显示此图片,因为计算机上的可用内存不足",本专题为大家提供windows照片无法显示相关的文章,帮助大家解决该问题。

1508

2023.08.01

Vite 项目如何打包和部署
Vite 项目如何打包和部署

Vite项目打包部署专题汇集生产构建、产物预览、输出目录、资源路径、Nginx与GitHub Pages部署及常见白屏和路由404问题的已审核文章,帮助读者完成上线流程。

40

2026.10.28

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Uvicorn官方安装与入门手册
Uvicorn官方安装与入门手册

共0课时 | 0人学习

Python uv官方项目管理指南
Python uv官方项目管理指南

共0课时 | 0人学习

Python uv官方安装与入门文档
Python uv官方安装与入门文档

共0课时 | 0人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn