真python技巧实战应用，函数参数自动收集并批量处理

前言

分享一个最近发现的实用小技巧。有时候我会封装一下别人的库中的某个功能函数，比如下面是一个第三方库的函数：

函数的参数非常多。而我需要自定义一个与它参数一样的函数，里面做一些小处理后，调用它的原函数：

• 为了让函数有智能提示，这里不能使用 *args 与 **kws 收集参数
• 把传入的字符串处理一下

这里的问题是，难道我一定要把参数逐一传入原函数吗？

1. 上面的第29行代码能不能简化？
2. 代码 行 24-27 的处理能不能简化？

我们首先要知道一点，函数的参数，其实与我们普通定义的变量没有多大差别。区别只是函数参数只是从外部指定值。

既然函数的参数是普通的变量，那么很自然我们可以想到使用内置函数 locals 获取本地变量字典

• 行24：利用 locals 函数获取所有参数的字典
• 行26：利用 ** 解包传参调用

非常简单地解决了问题。不过这里注意，最好在函数一开始的时候就做这个事情。这是因为如果在执行 locals 之前，你自己在函数中定义了变量，那么也会保存在 locals 里面。

• 行24：不能在调用 locals 之前定义变量

其实有了参数字典，批量处理字符串参数的问题就很容易解决。

虽然这里没有什么问题，但是通常来说，我们不应该在遍历时修改原字典。可以换成推导式或复制再修改

但是，现在我有好几个这种间接调用的函数。难道每一个我都要像上面那样写一大串代码吗？

需要时，调用即可:

真的可以吗？

显然不行。因为调用 locals 时，函数 strip_str_args 根本没有任何参数！

办法总比困难多，既然不能直接使用 locals ，那就让外面传进来吧：

调用的时候就需要使用 locals

还能不能再简化？每次都要传入 locals ，不符合我的"懒人"风格。

其实 locals 函数就是从当前 "帧栈" 获取的数据

• 行16：导入 inspect 模块
• 行20：获取当前帧栈
• 行24： cur_frame.f_back 获取上一层帧栈，就是调用这个函数的人的环境信息。 f_locals 相当于获得了外部传进来的 locals 字典

现在调用则很简单了：

• 要确保在函数一开始调用

目前，还有一个限制，"确保在函数一开始调用" 。

这个限制能去掉吗？

• Python进阶：你定义的变量到底保存在哪里
• 你的想象力限制了python能力，自动化识别函数调用关系，还能可视化