如何区分tasks.py的导入来源：main.py还是Celery命令？

当然可行，但得先理清一个关键点——单纯靠__name__变量没法直接区分「被main.py导入」和「被Celery加载」这两种场景，因为这两种情况都是模块被导入，此时tasks.py的__name__值都是'tasks'，只有当你直接运行python tasks.py时，__name__才会是'__main__'。

不过我们有几种靠谱的办法实现你的需求，按推荐程度排序：

方案一：重构代码，显式调用（最推荐） #

这是Python项目中最清晰、最健壮的做法——把文件读取逻辑封装成函数，只在main.py导入后主动调用，而不是让它在模块顶层自动执行。

比如修改tasks.py：

# tasks.py
def load_file_data():
    """封装文件读取逻辑"""
    with open('your_file.txt', 'r') as f:
        data = f.read()
        # 这里可以加后续处理逻辑
        return data

# 模块顶层不再执行文件读取

然后在main.py里显式调用：

# main.py
import tasks

# 只有当main.py需要时，才主动触发文件读取
file_data = tasks.load_file_data()

这样不管是Celery通过celery -A tasks -l info加载模块，还是其他地方导入tasks.py，都不会自动执行文件读取，完全由main.py控制时机，逻辑清晰且无耦合。

方案二：用环境变量标记导入来源（次推荐） #

如果不想重构代码，可以通过环境变量来标记导入来源。在main.py导入tasks.py前设置一个环境变量，tasks.py里检查这个变量是否存在，来决定是否执行文件读取。

修改main.py：

# main.py
import os

# 设置环境变量标记是main.py在导入
os.environ['FROM_MAIN_IMPORT'] = '1'

import tasks

然后修改tasks.py：

# tasks.py
import os

# 只有检测到main.py设置的环境变量时，才执行文件读取
if os.environ.get('FROM_MAIN_IMPORT') == '1':
    with open('your_file.txt', 'r') as f:
        data = f.read()
        # 后续处理

这种方式比检查调用栈更可靠，不过需要main.py配合设置变量，存在轻微耦合。

方案三：检查调用栈（不推荐） #

可以用inspect模块查看调用栈，判断导入tasks.py的是不是main.py，但这种方式依赖文件路径和文件名，不够健壮（比如main.py改名或移动位置就会失效），代码也更复杂。

示例代码（仅作演示，不推荐在生产环境使用）：

# tasks.py
import inspect

def is_imported_by_main():
    # 遍历调用栈，检查是否来自main.py
    for frame in inspect.stack()[1:]:
        if frame.filename.endswith('main.py'):
            return True
    return False

if is_imported_by_main():
    with open('your_file.txt', 'r') as f:
        data = f.read()
        # 后续处理

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