前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >你的想象力限制了python能力,自动化识别函数调用关系,还能可视化

你的想象力限制了python能力,自动化识别函数调用关系,还能可视化

作者头像
咋咋
发布2023-09-01 10:09:43
2750
发布2023-09-01 10:09:43
举报
文章被收录于专栏:数据大宇宙

前言

我喜欢用 python 做一些临时性数据工作,简单情况下,直接一把梭写到底。比如简单的多文件合并数据:

定义函数?一辈子都不可能。

不过,稍微复杂一些的情况,比如下面是 tableau prep 数据任务挑战中一道简单题目——寻找可能具有欺诈性的交易。

代码画风突变成这样子:

不让我定义函数?想要我命了吧!

得益于 pandas 的管道功能,我们可以更容易管理复杂的数据任务代码。关于如何以正确的思路使用 pandas 管道(pipe) ,具体可以查看我的 pandas 专栏。

数据处理是一种"重流程"的编程。但是,你会发现,上面的代码不管如何划分,你也无法容易理清楚数据流程。这才是痛点。

那如果有一种工具,可以把函数调用关系,以可视化方式展示给你,并且你可以轻松查看每一步处理结果的数据,还能直接跳转到具体代码行?看看演示:

  • 自动生成函数调用图。流程图可以缩放,拖动平移
  • 点击每个节点,下方出现函数处理结果的表数据。还可以通过勾选,快速筛选数据

当然,如果不能快速定位到代码,那就没有意思。

工具使用 nicegui 制作

pandas 专栏马上开始最后关于工程化的阶段,本节介绍的可视化工具就是为了专栏而制作。工程化的章节内容,将会是大量 tableau prep 数据处理挑战任务实战。

要做到这样的可视化,必需找到一种方式,可以在 python 中,自动化识别函数调用关系。

今天,我们探讨一下,如何做到这一切。重点是分享里面涉及到的 python 知识。

目前我想到3种实现方式,本文讲解其中一种


验证想法

要设计一个新的功能,我们需要从最简单的问题开始,验证想法是否能行。假设两个简单的函数

  • 在函数 b 中,调用了 函数 a

现在我们需要的是,得到一个记录信息,能反映出,函数 b 中,使用了函数 a。

python 中可以做到吗?

这涉及 python 中一个概念——闭包。直观来说,闭包就是一个函数中,直接使用了外部定义的变量。就像上面例子中,函数 b 中并没有定义变量 a,那么代码中使用的变量 a ,就是外部定义的函数 a。

我们可以使用 inspect 模块的 getclosurevars 获取闭包变量。

  • 注意, 我们没有执行函数 b
  • 得到的是一个 ClosureVars 对象。其中有一个 globals 属性,可以获取函数中全局闭包变量映射表(字典)

注意字典的 value 是函数对象。有了函数对象,我们就可以获取它的一切信息。比如函数定义在哪个文件的哪一行,有什么参数等等。

现在,可以把功能封装起来,看起来像这样子:

  • 行37:我们只关注函数之间的调用,所以这里做了过滤

这样子调用:

准确控制

但是,现在是通过我们手工传入函数 b ,这样子太麻烦了。在实际使用中,我们希望直接调用一个函数,就能自动检测当前环境所有的全局变量,并找出调用关系。

有小伙伴可能会想到,可以用 globals 函数获取所有的全局变量字典。但是不适合我们的情况。因为我们的功能函数是单独定义在一个模块文件中。

如果在我们定义的函数中使用 globals,只会获取到当前模块的全局变量。

此时仍然可以使用 inspect 模块的 currentframe 获取当前调用帧栈,从而获取上一层帧栈:

这里的意思就是:"谁调用我,我就拿了谁的全局变量"

帧栈相关知识,可以查看我的相关文章

剩下就非常简单,遍历这个字典,筛选出函数对象,然后调用之前定义的 get_func_relationships

  • 行81:得到的是一个 列表中的列表
  • 行80:使用 itertools 模块的 chain 给展开成一层列表

这里还存在一些问题,我们希望它不要什么函数都获取,由使用者为需要检测关系的函数打上标记。比如:

只有打上 @check 装饰器的函数,才需要获取调用关系。

只需要创建一个类即可:

装饰器知识点以前就有讲解

我们需要把之前的功能函数中的目标类型判断修改为 TargetFn :

一切就绪

  • 行1:使用时,先导入
  • 行8:需要检测的函数,打上装饰器
  • 行40:只需要在最后调用 build_all_relationships 即可

有了关系信息,做功能界面就没有太大难度了。

关于使用 nicegui 制作功能界面,我就放在视频教学中再讲解。

推荐文章:

  1. Python进阶:你定义的变量到底保存在哪里
  2. 多了解Python一点点,为什么我们需要定义变量?
本文参与 腾讯云自媒体同步曝光计划,分享自微信公众号。
原始发表:2023-08-10,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 数据大宇宙 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 验证想法
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档