[Python语言]函数外部改变函数内部变量

[Python语言]函数外改变函数内部变量

需求：

因需要继承同事写的一个类，类里面有一个800行的方法，继承这个类以后需要重写这个方法，这个方法只用修改里面的一个变量就能直接使用，重写的话需要把800行代码复制到继承的子类里面，代码会显得很臃肿，为了简化代码，编写了下面这段代码实现在函数外部改变内部的变量。

测试函数

编写一个demo.py测试脚本，内容是

def demo():
    a=1
    b=2
    print(a,b)

很显然调用这个函数会打印1，2

改变函数内变量程序

程序思路：

• 用inspect.getsource方法获得函数的代码
• 使用replace的方式把函数内变量改变
• 动态创建函数

代码实现

import inspect
from demo import demo
#传入函数，改变函数内部变量a，从a=1改变成a=2
def cfunc(func):
    #当前位置的全部局部变量
    lc=locals()
    #获得函数的内容
    func_code=inspect.getsource(func)
    #使用replace改变函数内部的变量
    func_code=func_code.replace('a=1','a=2')
    #动态创建函数
    exec(func_code)
    #获得函数并返回
    res_func=lc[func.__name__]
    return res_func

demo=cfunc(demo)
demo()

调用这个函数后打印的结果是2，2满足了在函数外部改变函数内部的变量