函数内通过YAML加载全局变量失效问题求助

问题背景 #

需要通过YAML文件管理程序的多个全局参数，通过函数实现加载逻辑，但函数内的globals().update()方法无法生效，且因变量数量和名称不确定，无法使用global关键字逐个声明。

YAML配置文件（原内容存在语法错误） #

test:
  - 12
  - 13
  - 14
  - stuff:
      john

test2: yo

原Python代码 #

import os
import yaml

def load_config():
    with open(os.path.join(os.getcwd(), "config.yaml"), 'r') as reader:
        vals = yaml.full_load(reader)
        globals().update(vals)

期望输出 #

load_config()

test
---------------
[12,13,14,{'stuff':'john'}]


test2
---------------
yo

实际结果 #

load_config()

test
---------------
NameError: name 'test' is not defined


test2
---------------
NameError: name 'test2' is not defined

解决思路及修正方案 #

1. 修复YAML文件语法错误 #

原YAML中stuff: john的john未加引号，YAML会将其视为未定义的锚点导致解析失败，需将字符串值用引号包裹：

test:
  - 12
  - 13
  - 14
  - stuff: "john"

test2: yo

2. 确保globals()指向目标命名空间 #

函数内的globals()默认返回定义该函数的模块的全局命名空间，分两种情况处理：

• 如果load_config和调用代码在同一个模块，直接使用globals().update()即可生效，但需确保文件路径正确、YAML语法无错。
• 如果load_config是从其他模块导入的，globals()会指向导入模块的命名空间，而非调用者的模块。此时需要获取调用者的全局命名空间，可通过inspect模块实现：

import os
import yaml
import inspect

def load_config():
    # 获取调用者的全局命名空间
    caller_frame = inspect.currentframe().f_back
    try:
        caller_globals = caller_frame.f_globals
        with open(os.path.join(os.getcwd(), "config.yaml"), 'r') as reader:
            vals = yaml.full_load(reader)
            if vals:
                caller_globals.update(vals)
    finally:
        # 手动删除帧引用，避免内存泄漏
        del caller_frame

3. 验证加载结果 #

调用load_config()后，即可直接访问全局变量：

load_config()
print(test)
# 输出：[12, 13, 14, {'stuff': 'john'}]
print(test2)
# 输出：yo

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