前言
分享一个最近发现的实用小技巧。有时候我会封装一下别人的库中的某个功能函数,比如下面是一个第三方库的函数:
函数的参数非常多。而我需要自定义一个与它参数一样的函数,里面做一些小处理后,调用它的原函数:
- 为了让函数有智能提示,这里不能使用 *args 与 **kws 收集参数
- 把传入的字符串处理一下
这里的问题是,难道我一定要把参数逐一传入原函数吗?
- 上面的第29行代码能不能简化?
- 代码 行 24-27 的处理能不能简化?
我们首先要知道一点,函数的参数,其实与我们普通定义的变量没有多大差别。区别只是函数参数只是从外部指定值。
既然函数的参数是普通的变量,那么很自然我们可以想到使用内置函数 locals 获取本地变量字典:
- 行24:利用 locals 函数获取所有参数的字典
- 行26:利用 ** 解包传参调用
非常简单地解决了问题。不过这里注意,最好在函数一开始的时候就做这个事情。这是因为如果在执行 locals 之前,你自己在函数中定义了变量,那么也会保存在 locals 里面。
- 行24:不能在调用 locals 之前定义变量
其实有了参数字典,批量处理字符串参数的问题就很容易解决。
虽然这里没有什么问题,但是通常来说,我们不应该在遍历时修改原字典。可以换成推导式或复制再修改
但是,现在我有好几个这种间接调用的函数。难道每一个我都要像上面那样写一大串代码吗?
我一开始也想到,这不是很简单事情吗,把上面的代码抽出来,到一个函数就可以:
需要时,调用即可:
真的可以吗?
显然不行。因为调用 locals 时,函数 strip_str_args 根本没有任何参数!
办法总比困难多,既然不能直接使用 locals ,那就让外面传进来吧:
调用的时候就需要使用 locals:
还能不能再简化?每次都要传入 locals ,不符合我的"懒人"风格。
以前我就有两篇文章讲解过调用栈。当 python 每次进入一个函数,都会启用一个"帧栈",用于保存当前函数执行过程中的信息。这些信息当然包括了定义的变量。
其实 locals 函数就是从当前 "帧栈" 获取的数据:
- 行16:导入 inspect 模块
- 行20:获取当前帧栈
- 行24: cur_frame.f_back 获取上一层帧栈,就是调用这个函数的人的环境信息。 f_locals 相当于获得了外部传进来的 locals 字典
现在调用则很简单了:
- 要确保在函数一开始调用
目前,还有一个限制,"确保在函数一开始调用" 。
这个限制能去掉吗?