Python跨文件变量传递及模块内函数重定义失效问题求助

问题背景 #

主文件代码：

import os
os.chdir('<<some folder>>')
import codeFile as CF

目标目录下的codeFile.py包含一行顶层执行代码：

print(a_Variable)

执行主文件时提示'a_Variable'未定义，即便在主文件提前定义变量：

import os
os.chdir('<<some folder>>')
a_Variable = 6
import codeFile as CF

错误依旧，需要让外部定义的变量对导入模块生效。

解决办法 #

Python模块拥有独立的命名空间，主文件中的变量不会自动传入模块，可尝试以下几种方案：

• 模块显式导入主文件变量
  修改codeFile.py，直接从主文件导入目标变量（主文件名需替换为实际名称，比如主文件是main.py）：

from main import a_Variable
print(a_Variable)

• 导入模块后手动赋值并封装执行逻辑
  由于模块顶层代码在导入时就会执行，更合理的方式是将模块代码封装为函数，主文件导入后先赋值再调用：
  codeFile.py修改为：

def run():
    print(a_Variable)

主文件代码：

import os
os.chdir('<<some folder>>')
import codeFile as CF
CF.a_Variable = 6
CF.run()

• 通过环境变量传递
  主文件中设置环境变量：

import os
os.environ['a_Variable'] = '6'
os.chdir('<<some folder>>')
import codeFile as CF

codeFile.py中读取环境变量：

import os
a_Variable = int(os.environ.get('a_Variable'))
print(a_Variable)

问题背景 #

codeFile.py代码如下：

def My_Fun(args):
    return args + 10
def My_Fun1(args):
    return args * 10

def init(a_Variable):
    if a_Variable == 10 :
        print('RUNNING..')
        def My_Fun(args):
            return args + 10 + 2
        def My_Fun1(args):
            return args * (10 + 2)

主文件执行代码：

import codeFile as CD
CD.init(10)
CD.My_Fun(100)

预期返回值为112，但实际得到110，模块原函数未被替换。

解决办法 #

init函数内定义的函数是局部变量，仅在init内部生效，不会自动覆盖模块顶层的函数。要替换模块级函数，需显式修改模块的命名空间：

方法1：使用global声明 #

修改codeFile.py的init函数，声明要修改模块级的全局函数：

def My_Fun(args):
    return args + 10
def My_Fun1(args):
    return args * 10

def init(a_Variable):
    global My_Fun, My_Fun1
    if a_Variable == 10 :
        print('RUNNING..')
        def My_Fun(args):
            return args + 10 + 2
        def My_Fun1(args):
            return args * (10 + 2)

方法2：直接修改模块对象属性 #

这种方式逻辑更清晰，适合复杂场景：

def My_Fun(args):
    return args + 10
def My_Fun1(args):
    return args * 10

def init(a_Variable):
    import sys
    # 获取当前模块的对象
    current_module = sys.modules[__name__]
    if a_Variable == 10 :
        print('RUNNING..')
        def My_Fun(args):
            return args + 10 + 2
        def My_Fun1(args):
            return args * (10 + 2)
        # 替换模块顶层的函数
        current_module.My_Fun = My_Fun
        current_module.My_Fun1 = My_Fun1

修改后执行CD.init(10)，再调用CD.My_Fun(100)就会返回预期的112。

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