Python中如何避免os.getenv重复输入变量名？实现变量自动初始化

针对你提出的需求——避免重复输入环境变量名调用os.getenv，可以通过以下几种方式实现类似功能：

1. 单变量初始化函数 #

这个函数接收环境变量的名字符串，自动将变量注入到全局命名空间中：

import os

def get_and_init_env(var_name):
    value = os.getenv(var_name)
    globals()[var_name] = value
    return value

# 调用示例
get_and_init_env("BUCKET")
get_and_init_env("IN_CSV")
get_and_init_env("OUT_CSV")

# 直接使用变量
print(BUCKET)

注：这里必须传入变量名的字符串（而非未定义的变量），因为直接传入未定义的BUCKET会触发NameError。

2. 批量初始化函数 #

如果需要初始化多个环境变量，批量处理会更高效：

import os

def init_env_vars(var_names):
    for name in var_names:
        globals()[name] = os.getenv(name)

# 一次性初始化所有需要的变量
init_env_vars(["BUCKET", "IN_CSV", "OUT_CSV"])

3. 用数据类管理环境变量（更规范的方案） #

对于中大型项目，用数据类统一管理环境变量会让代码结构更清晰，也便于维护：

import os
from dataclasses import dataclass

@dataclass
class EnvConfig:
    BUCKET: str = os.getenv("BUCKET")
    IN_CSV: str = os.getenv("IN_CSV")
    OUT_CSV: str = os.getenv("OUT_CSV")

# 实例化后使用
env = EnvConfig()
print(env.BUCKET)

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