检测生成器函数(Generator Function)类型

轻静君_7703

轻静君_7703

2026-06-17

168人浏览

原创

应使用inspect.isgeneratorfunction()检测生成器函数,因为它通过函数code object的co_generator标志位静态判断,零副作用、100%精准;而isinstance(fn(), types.generatortype)会执行函数引发副作用或异常,hasattr检查迭代器属性无法区分真生成器与普通可迭代对象。

检测生成器函数(generator function)类型

要准确检测一个函数是否为生成器函数,核心是检查其定义本身是否含有 yield(而非看它调用后返回什么),且不能执行该函数。最可靠、安全、标准的方式是使用 inspect.isgeneratorfunction()。

为什么不用 isinstance(fn(), types.GeneratorType)

这种写法会真正调用函数,带来三类风险:

  • 可能触发副作用(如修改全局状态、写文件、发请求)
  • 可能耗尽资源(如打开的连接、临时文件句柄)
  • 可能直接抛出异常(比如参数校验失败、IO错误),导致检测逻辑中断

它检测的是“调用结果”,不是“函数类型”,本质是运行时判断,不符合静态识别需求。

为什么不能靠 hasattr 或 __iter__ / __next__

这些属性只说明对象是迭代器或可迭代对象,但无法区分:

Variant AI
Variant AI

一款AI图像与设计工具,主要用于YC孵化的UI/UX智能设计工具,一句话自动生成网页、APP设计方案,适合需要提升相关任务效率的用户。

下载
  • 一个普通类实现了 __iter__ 和 __next__,返回自定义迭代器
  • 一个返回 range(10) 的普通函数
  • 一个真正的生成器函数(含 yield)

三者都满足 hasattr(obj, '__iter__') and hasattr(obj, '__next__'),但只有第三种才是生成器函数。

inspect.isgeneratorfunction() 是唯一权威方案

它直接读取函数对象底层的 code object 中的 co_flags 标志位(CO_GENERATOR),这个标志在函数被 def 定义编译时就已确定,无需调用、零副作用、100% 精准:

  • 兼容 @staticmethod、@classmethod、实例方法
  • 对普通函数、lambda、内置函数、functools.partial 等返回 False,不会误判
  • 标准库支持,无需第三方依赖,Python 3.2+ 全版本可用

实际使用示例

以下代码可直接用于装饰器、插桩、测试框架等场景:

import inspect
<p>def safe_wrap(fn):
if inspect.isgeneratorfunction(fn):
def wrapped(*args, <strong>kwargs):
print(f"[GEN] Calling {fn.<strong>name</strong>}")
yield from fn(*args, *<em>kwargs)  # 保持生成器协议完整
return wrapped
else:
def wrapped(</em>args, </strong>kwargs):
print(f"[FUNC] Calling {fn.<strong>name</strong>}")
return fn(*args, **kwargs)
return wrapped</p>

这样包装后的函数,既保留了原生行为(for 循环仍可用),又统一了日志或监控逻辑。

相关专题

更多
function是什么
function是什么

function是函数的意思,是一段具有特定功能的可重复使用的代码块,是程序的基本组成单元之一,可以接受输入参数,执行特定的操作,并返回结果。本专题为大家提供function是什么的相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.04

2640

5

js函数function用法
js函数function用法

js函数function用法有:1、声明函数;2、调用函数;3、函数参数;4、函数返回值;5、匿名函数;6、函数作为参数;7、函数作用域;8、递归函数。本专题提供js函数function用法的相关文章内容,大家可以免费阅读。

2023.10.07

454

5

JavaScript中的typeof用法
JavaScript中的typeof用法

在JavaScript中,typeof是一个用来确定给定变量的数据类型的操作符。可以用来确定一个变量是字符串、数字、布尔值、函数、对象或undefined的数据类型。更多关于typeof用法相关文章,详情请看本专题下面的文章,php中文网欢迎大家前来学习。

2023.11.23

7087

13

Buffalo框架数据库开发全教程
Buffalo框架数据库开发全教程

本专题围绕Buffalo框架数据库开发,讲解database.yml多环境配置、soda与fizz迁移生成回滚、模型结构体标签、增删改查与条件查询、一对多与多对多关联、数据校验、回调钩子、事务处理及原生SQL执行能力。

2026.09.23

180

15

Buffalo框架路由与请求处理实操指南
Buffalo框架路由与请求处理实操指南

本专题讲解Buffalo框架路由与请求处理机制,涵盖路由注册与分组、资源路由、Handler编写规范、Context上下文方法、参数绑定、中间件编写挂载、Session与Cookie读写、Flash消息及错误页面定制方法。

2026.09.23

80

15

Buffalo框架零基础入门教程
Buffalo框架零基础入门教程

本专题整理Buffalo框架入门内容,涵盖Go环境准备、buffalo CLI安装、新项目生成、目录结构说明、dev热加载启动、数据库连接配置与常见报错排查,帮助新手按约定优于配置的思路跑通第一个Buffalo框架应用。

2026.09.23

80

15

Conan创建软件包配方指南
Conan创建软件包配方指南

本专题介绍通过conanfile.py创建软件包的方法,讲解包名、版本、依赖和构建设置等基础信息,以及source、build、package、package_info等常用方法的作用及编写思路。

2026.09.22

60

12

Conan二进制包配置指南
Conan二进制包配置指南

本专题介绍Conan根据操作系统、编译器、架构和构建类型生成二进制包的方法,讲解Profile、Settings、Options及Package ID的作用,帮助管理不同平台和编译环境下的包版本。

2026.09.22

60

13

Conan私有仓库搭建教程
Conan私有仓库搭建教程

本专题系统的讲解Conan私有仓库的搭建流程,涵盖仓库服务部署、存储目录配置、用户认证、权限划分和远程地址添加,并介绍内部C++依赖包的上传、下载及版本维护方法。

2026.09.22

60

19

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
热门推荐
/
最新课程
phpStudy极速入门视频教程
phpStudy极速入门视频教程

共6课时 | 54.6万人学习

独孤九贱(4)_PHP视频教程
独孤九贱(4)_PHP视频教程

共89课时 | 133.3万人学习