当前位置:首页 > 文章列表 > 文章 > python教程 > Python 中如何实现异步 contextmanager

Python 中如何实现异步 contextmanager

2026-02-25 13:53:37 0浏览 收藏
本文深入解析了在Python中正确实现异步上下文管理器的两种核心方法:推荐使用Python 3.7+内置的`@asynccontextmanager`装饰器(简洁安全,支持`async with`、自动异常传播和异步清理),或手动实现`__aenter__`与`__aexit__`异步方法(适用于需精细控制状态或复用的场景);同时重点警示了常见陷阱——如绝不可将同步`@contextmanager`用于`async with`、避免混用同步/异步资源、注意Python版本兼容性,并强调必须依托真正异步的底层库(如aiofiles)才能发挥实效,为构建健壮、可维护的异步代码提供了清晰、实用且经生产验证的最佳实践指南。

Python 如何让 contextmanager 支持异步上下文

Python 的 @contextmanager 装饰器本身不支持异步上下文(即 async with),因为它设计用于同步生成器。要实现异步上下文管理,需改用 @asynccontextmanager —— 这是 contextlib 模块从 Python 3.7 开始提供的专用工具。

使用 @asynccontextmanager 替代 @contextmanager

这是最直接、标准的方案。它接受一个 async def 函数,该函数必须包含恰好一个 yield,且 yield 后的值将作为 async with 语句中 as 绑定的对象。

  • 确保函数是 async def,且内部可使用 await(如异步初始化、清理)
  • yield 必须存在,且只能出现一次;yield 后的表达式(如有)即为上下文对象
  • 异常会正常传播,yield 后的代码相当于 __aexit__,可用于异步清理

示例:

from contextlib import asynccontextmanager

@asynccontextmanager async def db_connection(): conn = await acquire_db_connection() # 假设这是个协程 try: yield conn finally: await conn.close() # 异步清理

使用方式

async def main(): async with db_connection() as conn: await conn.execute("SELECT 1")

手动实现 __aenter____aexit__

若需更精细控制(如状态保持、多次重用、非生成器逻辑),可定义普通类并实现两个异步方法。

  • __aenter__ 应返回 awaitable,通常用 return await ... 或直接返回协程
  • __aexit__ 必须是 async def,接收 exc_type, exc_value, traceback,返回 True 可抑制异常
  • 类实例无需可调用,但需确保生命周期清晰(避免意外重复进入)

示例:

class AsyncFile:
    def __init__(self, path):
        self.path = path
        self.file = None
async def __aenter__(self):
    self.file = await aiofiles.open(self.path, 'r')
    return self.file

async def __aexit__(self, exc_type, exc_val, exc_tb):
    if self.file:
        await self.file.close()

使用

async with AsyncFile('data.txt') as f: content = await f.read()

注意兼容性与常见误区

混用同步和异步上下文管理器会导致运行时错误(如 TypeError: object X is not async context manager)。

  • 不要对 @contextmanager 包裹的函数加 async def —— 它无法被 async with 识别
  • Python < 3.7 不含 @asynccontextmanager,需自行实现或升级解释器
  • 若底层资源本身不支持异步(如普通文件 I/O),强行套异步装饰器无意义,应优先使用真正异步的库(如 aiofilesaiomysql

组合多个异步上下文(类似 ExitStack

标准 ExitStack 不支持异步。如需动态管理多个异步上下文,可用 async_exit_stack 第三方包,或手动维护协程列表:

  • 收集所有 __aenter__ 协程,用 await asyncio.gather(...) 并发进入
  • 退出时按反序 await 每个 __aexit__,并正确处理异常传播
  • 生产环境建议使用成熟方案如 async_exit_stack(PyPI 可安装),避免手动出错

文中关于的知识介绍,希望对你的学习有所帮助!若是受益匪浅,那就动动鼠标收藏这篇《Python 中如何实现异步 contextmanager》文章吧,也可关注golang学习网公众号了解相关技术文章。

百度搜索app怎么改配色?简单设置教程百度搜索app怎么改配色?简单设置教程
上一篇
百度搜索app怎么改配色?简单设置教程
Win11多屏壁纸设置教程
下一篇
Win11多屏壁纸设置教程
查看更多
最新文章
查看更多
课程推荐
  • 前端进阶之JavaScript设计模式
    前端进阶之JavaScript设计模式
    设计模式是开发人员在软件开发过程中面临一般问题时的解决方案,代表了最佳的实践。本课程的主打内容包括JS常见设计模式以及具体应用场景,打造一站式知识长龙服务,适合有JS基础的同学学习。
    543次学习
  • GO语言核心编程课程
    GO语言核心编程课程
    本课程采用真实案例,全面具体可落地,从理论到实践,一步一步将GO核心编程技术、编程思想、底层实现融会贯通,使学习者贴近时代脉搏,做IT互联网时代的弄潮儿。
    516次学习
  • 简单聊聊mysql8与网络通信
    简单聊聊mysql8与网络通信
    如有问题加微信:Le-studyg;在课程中,我们将首先介绍MySQL8的新特性,包括性能优化、安全增强、新数据类型等,帮助学生快速熟悉MySQL8的最新功能。接着,我们将深入解析MySQL的网络通信机制,包括协议、连接管理、数据传输等,让
    500次学习
  • JavaScript正则表达式基础与实战
    JavaScript正则表达式基础与实战
    在任何一门编程语言中,正则表达式,都是一项重要的知识,它提供了高效的字符串匹配与捕获机制,可以极大的简化程序设计。
    487次学习
  • 从零制作响应式网站—Grid布局
    从零制作响应式网站—Grid布局
    本系列教程将展示从零制作一个假想的网络科技公司官网,分为导航,轮播,关于我们,成功案例,服务流程,团队介绍,数据部分,公司动态,底部信息等内容区块。网站整体采用CSSGrid布局,支持响应式,有流畅过渡和展现动画。
    485次学习
查看更多
AI推荐
  • ljg-skills -
    ljg-skills
    ljg-skills 是李继刚开源的 AI 技能与提示词集合,面向大模型使用者整理了一批可复用的 prompt、角色设定和任务技能模板,适合用于学习提示词设计、搭建个人 AI 工作流和沉淀团队常用智能体能力。
    2123次使用
  • MELO音乐 - AI 音乐生成平台,支持多模态创作能力
    MELO音乐
    MELO音乐是一站式AI视频与音乐制作助手,对标suno, udio的高品质体验。提供伴奏生成、原创写词、无损导出、哼唱识曲、混音变声等全套音频与短视频编辑工具。无论是流行Kpop、电音说唱、民谣古风、摇滚儿歌还是商用轻音乐,MELO为你免费谱曲,轻松做同款!
    1964次使用
  • UniScribe - AI 免费在线音视频转文字平台
    UniScribe
    UniScribe 是一款 AI 音视频转文字与内容整理工具,支持上传音频、视频文件或粘贴 YouTube 链接,自动生成转写文本、摘要、思维导图和关键问题,并支持多格式导出,适合会议记录、课程学习、访谈整理和内容创作复盘。
    1909次使用
  • 剧云 - 免费 AI 智能中文剧本创作平台
    剧云
    剧云是专业中文剧本创作平台,安全稳定运行十余年,集成AI编剧、剧本医生审核、人物小传、剧情关系图、大纲编写、多人协作、Word导入导出、版权管控功能,数据安全防护,轻松高效创作剧本。
    2113次使用
  • 万象有声 - AI 一站式有声内容创作平台
    万象有声
    万象有声,一个专为有声创作者打造的新一代智能有声内容创作平台。平台提供专业的智能拆章、智能画本编辑、AI配音、AI生成音效、后期制作、智能对轨、智能审听等有声创作全流程工具,可以帮助创作者高效、低成本创作出引人入胜的有声作品。立即体验,让有声书制作更简单!
    2097次使用
微信登录更方便
  • 密码登录
  • 注册账号
登录即同意 用户协议隐私政策
返回登录
  • 重置密码