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

2023年 12月 8日 39.2k 0

前言

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

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

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

现在调用则很简单了:

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

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

这个限制能去掉吗?

相关文章

JavaScript2024新功能:Object.groupBy、正则表达式v标志
PHP trim 函数对多字节字符的使用和限制
新函数 json_validate() 、randomizer 类扩展…20 个PHP 8.3 新特性全面解析
使用HTMX为WordPress增效:如何在不使用复杂框架的情况下增强平台功能
为React 19做准备:WordPress 6.6用户指南
如何删除WordPress中的所有评论

发布评论