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

我是数据外星人 2024-02-20 02:05:07
前言

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

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

为了让函数有智能提示,这里不能使用 *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 字典

现在调用则很简单了:

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

这个限制能去掉吗?

不要忘记一键三连。你的点赞、收藏、关注,是我创作的动力。

推荐文章:

0 阅读:0

我是数据外星人

简介:感谢大家的关注