Python如何在模块导入时动态定义变量？

要实现导入foo模块时自动通过fancy函数动态生成my_var_fancy（无需显式定义变量），可以利用Python模块的全局命名空间或结合hasattr()检查来实现，以下是两种可行方案：

方案1：通过globals()直接注入变量 #

让fancy函数负责将生成的结果直接添加到模块的命名空间中，无需手动写变量赋值语句。

foo.py 代码 #

def fancy(var_name, greeting="hello"):
    # 生成目标字符串
    result = f"{greeting}_world"
    # 将结果注入到模块的全局命名空间
    globals()[var_name] = result

# 模块加载时自动执行，生成目标变量
fancy("my_var_fancy")
fancy("my_var_fancy_es", "Hola")

方案2：用hasattr()检查后动态生成 #

如果需要确保变量仅在不存在时才生成，可以结合hasattr()判断模块中是否已有该变量，避免重复生成。

foo.py 代码 #

def fancy(greeting="hello"):
    return f"{greeting}_world"

# 检查模块中是否存在my_var_fancy，不存在则生成
if not hasattr(__name__, "my_var_fancy"):
    my_var_fancy = fancy()

# 生成西班牙语版本的变量
if not hasattr(__name__, "my_var_fancy_es"):
    my_var_fancy_es = fancy("Hola")

main.py 调用示例 #

无论用哪种方案，main.py的调用方式都是一致的：

import foo

print(foo.my_var_fancy)    # 输出: hello_world
print(foo.my_var_fancy_es) # 输出: Hola world

原理说明 #

• Python模块在导入时会自上而下执行所有代码，模块级别的函数调用会在导入完成前执行，完成变量的动态生成。
• globals()返回当前模块的全局命名空间字典，直接修改这个字典就能添加新的模块变量。
• hasattr(__name__, var_name)用于检查当前模块（__name__代表模块自身）是否已存在指定变量，避免重复赋值。

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