python - 仅在特定文件运行时如何运行功能？

Question

我有多个 file1,file2,file3 我们可以将文件名作为环境变量获取为env_file：

如果 env 变量是 file2 则只运行 add 函数

env_file = "ref/file1"

env_file = "ref/file2"

env_file = "ref/file3"

主文件

def sub():
    do something
    
def add():
    run this function if file2 runs
    
def mul():
    do something 

预期输出为：

所有函数都应该运行，但是fun add()应该只在文件 2 分开运行时才应该运行所有其他函数。

标签： python

Answer 1

您需要从 os 模块获取环境变量。

import os
file = os.environ["env_file "]

您只需要在 add 函数中检查文件是否为“ref/file2”。