游乐游手机版
首页/前端开发/文章详情

Generator函数类型检测方法

时间:2026-07-03 06:59
判断一个函数是否为生成器函数,关键在于函数定义本身是否包含 yield 关键字,而非调用后返回的对象类型。最权威、最可靠的检测方式是使用 inspect isgeneratorfunction() —— 零副作用、100% 精准,甚至无需执行函数就能完成判断。 要判断一个函数是不是生成器函数,本质上
判断一个函数是否为生成器函数,关键在于函数定义本身是否包含 yield 关键字,而非调用后返回的对象类型。最权威、最可靠的检测方式是使用 inspect.isgeneratorfunction() —— 零副作用、100% 精准,甚至无需执行函数就能完成判断。

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

要判断一个函数是不是生成器函数,本质上只需要看它定义时是否含有 yield 语句,而不是看它被调用后生成了什么结果。核心理念是:不能真的去执行这个函数。因此,最稳妥、最标准的方法就是使用 inspect.isgeneratorfunction()。下面来分析为什么其他常见做法行不通。

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

这种写法会实际调用函数,带来的后果往往不可控:

  • 可能触发副作用——修改全局变量、写入文件、发起网络请求,这些操作可能在检测时意外发生。
  • 可能耗尽资源——比如打开连接后未关闭,或生成临时文件句柄。
  • 可能直接抛出异常——参数校验失败、IO 错误,导致检测逻辑本身崩溃。

它检查的是“调用结果”而非“函数类型”,属于运行时判断,完全不适合静态类型识别的需求。

为什么不能靠 hasattr 或 __iter__ / __next__

这些属性只能说明对象是迭代器或可迭代对象,但无法区分下面几种情况:

  • 一个普通类自己实现了 __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

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

这样包装后的函数,原生行为(比如 for 循环)完全保留,同时又统一了日志或监控逻辑。最关键的是——检测阶段根本不会执行原始函数,安全又可靠。

来源:https://www.php.cn/faq/2665623.html
上一篇JavaScript中Class作为模块导出的最佳实践完全指南与示例 下一篇Less与CSS原生变量冲突的转义处理技巧
本站内容用于信息整理与展示,如有侵权或内容问题请及时联系处理。

相关推荐

补充同频道和同主题内容,方便继续浏览更多相关内容。

同类最新

继续查看同栏目最近更新的文章。

更多
Vue应用中异步更新性能问题的优化策略详解
前端开发 · 2026-07-03

Vue应用中异步更新性能问题的优化策略详解

先来看一个令许多开发者感到困惑的场景:明明修改了数据,DOM 却“毫无反应”,无法获取最新的高度,也无法计算正确的坐标。这并非 Vue 的缺陷,反而是它精心设计的性能优化策略。核心在于——你需要学会与它“异步更新”的特性协作,而非硬碰硬。 所谓的“异步更新性能问题”,本质上是一种认知偏差。Vue 的

如何避免原型对象挂载大体积动态数组内存污染
前端开发 · 2026-07-03

如何避免原型对象挂载大体积动态数组内存污染

原型链上的大数组:一个隐蔽的内存冲击波 先给个核心判断:直接在原型对象上挂载一个大体积动态数组,这既不是传统意义上的内存“污染”,也不是安全漏洞那种“污染”,而是一种相当隐蔽但后果严重的内存管理失当。它会导致所有实例共享同一份数据,而且正因为生命周期跟整个原型链绑定得太紧,垃圾回收器(GC)根本看不

利用堆栈信息精准定位显式绑定错误对象致未定义异常
前端开发 · 2026-07-03

利用堆栈信息精准定位显式绑定错误对象致未定义异常

深入追踪:显式绑定传错对象引发的未定义异常 说实话,这类问题在JavaScript开发中相当常见——显式绑定传错了对象,然后方法执行时静默失败、访问undefined、或者抛出TypeError。但真正的难点不在于“报了什么错”,而在于“到底是哪个对象被绑错了”。要解决它,需要跳出堆栈的表层报错信息

ES模块中默认导出和具名导出的执行上下文
前端开发 · 2026-07-03

ES模块中默认导出和具名导出的执行上下文

export default 与具名导出在 ES Module 中的行为机制截然不同,核心差异不在于“值如何传递”,而在于绑定如何建立以及导入时如何使用。先给出总结性结论,再逐一详细拆解。 export default 是一种语法糖,而非真正的变量声明 这种设计容易引起误解。实际上,export d

详解HTML中iframe标签loading=lazy属性实现嵌入内容懒加载方法
前端开发 · 2026-07-03

详解HTML中iframe标签loading=lazy属性实现嵌入内容懒加载方法

先聊聊 loading= "lazy " 这个属性——它本意是让 iframe 实现延迟加载,但实际落地时常常“失效”。这并非程序漏洞,而是浏览器内置的防御机制:只有所有条件同时触发,它才会真正推迟资源请求。比如 src 必须是跨域地址(类似 https: widget example com emb