Python中类的魔术方法之上下文管理

2024-08-24 13:48

本文主要是介绍Python中类的魔术方法之上下文管理,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

文章目录

  • Python中类的魔术方法之上下文管理
    • 上下文管理
    • 上下文管理对象与管理的安全
    • 上下文应用场景
    • contextlib.countextmanager上下文管理装饰器

Python中类的魔术方法之上下文管理

上下文管理

文件IO操作可以对文件对象使用上下文管理,使用with…as语法。
示例:

with open("text") as f:pass

仿照上面例子写一个自己的类,实现上下文管理

class Point:passwith Point() as p: AttributeError:__exit__pass

提示属性错误,没有__exit__,看了需要这个属性
某些版本会显示没有__enter__

上下文管理对象与管理的安全

当一个对象同时实现了__enter__()__exit__()方法,它就属于上下文管理的对象

方法等价语法意义
__enter__(self)with obj() as c:pass进入语句块之前执行进入与此对象相关的上下文,如果存在该方法,with语法会把该方法的返回值绑定到as子句中指定的变量上
__exit__(self, exc_type, exc_value, traceback)with obj() as c:pass退出时执行退出与此对象相关的上下文。时执行

简单示例:

class Point:def __init__(self):print("init------------")def __enter__(self):print("enter~~~~~~~~~~~~")return "abc" #返回值决定了as 后面变量的值def __exit__(self,exc_type,exc_val,exc_tb):print("exit ================")with Point() as p:print("with in p = {}".format(p))

class5_001

  • 调用步骤:
    1. 先实例化with语句后面的对象obj。
    2. 调用对象obj中的__enter__(self)方法。如果有 as 语句将__enter__方法的返回值赋值该as后面的变量。
    3. 执行with下面面的语句体
    4. 离开with语句块的时候,调用__exit__(self,exc_type,exc_val,exc_tb)方法。如果with语句体内执行时发生异常,会被捕获并传入__exit__(self,exc_type,exc_val,exc_tb)的参数中。如果__exit__的返回值为等效于False表示不压制异常。反之压制异常。
  • 注意:with并不会开启一个新的作用域。
  • 方法参数
    1. __enter__(self)#方法没有其他参数。
    2. __exit__(self,exc_type,exc_value,exc_tb) #有3个参数
      • 如果没有异常,3个参数的值都为None,如果有异常,参数意义如下:
      • exc_type #异常类型
      • exc_value #异常的值
      • exc_tb #异常的追踪信息
      • 返回值:等效True的值为压制异常;否则,继续抛出异常

示例:演示异常

class Point:def __init__(self):print("init ----------")def __enter__(self):print("enter @@@@@@@@@@")return self #增加返回值def __exit__(self, exc_type, exc_val, exc_tb):print(1,exc_type)print(2,exc_val)print(3,exc_tb)print("exit ==============")# return ""return "a"with Point() as p:print("in with ------------")raise Exception("error")print("with over")print("执行完成。。。。")

class5_002

示例:用类来做装饰器,实现运行时间统计

import time,functools,datetimeclass Xdd_Timeit:"""时间统计类,也可以作为装饰器单独统计单个函数运行时长"""def __init__(self,fn):"""初始化方法:param fn: 需要统计的函数"""functools.update_wrapper(self,fn)self.fn = fnself.starttime = Noneself.stoptime = None#with语句开始时默认调用的方法def __enter__(self):self.starttime = datetime.datetime.now()return self#with语句结束后默认调用的方法def __exit__(self, exc_type, exc_val, exc_tb):self.stoptime = datetime.datetime.now()print("{} 总运行时长:{}".format(self.fn.__name__,(self.stoptime-self.starttime).total_seconds()))#使得类对象是个可调用函数def __call__(self, *args, **kwargs):start = datetime.datetime.now()sum = self.fn(*args, **kwargs)sound = datetime.datetime.now() - startprint("{} 运行时长:{}".format(self.fn.__name__,sound.total_seconds()))return sum@Xdd_Timeit
def add(x,y):time.sleep(1)return x+ywith Xdd_Timeit(add) as a:print(a(4,5),a(5,6),a(7,8))print(add(4,5),add(5,6),add(7,8))

class5_003

上下文应用场景

  1. 增强功能
    • 在代码执行的前后增加代码,以增强其功能。类似装饰器的功能。
  2. 资源管理
    • 打开了资源需要关闭,例如文件对象、网络连接、数据库连接等
  3. 权限验证
    • 在执行代码之前,做权限的验证,在 enter 中处理

contextlib.countextmanager上下文管理装饰器

  1. contextlib.contextmanager
    • 它是一个装饰器实现上下文管理,装饰一个函数,而不是像类一样实现__enter____exit__方法。
    • 对装饰的函数有要求:必须有yield,也就是这个函数必须返回一个生成器,且只有yield一个值。
from contextlib import contextmanager@contextmanager
def foo():print("enter") #相当于__enter__()yield 5  #yield 的值只能有一个,作为__enter__的返回值print("exit") #相当于__exit__()with foo() as f:print("with start-------")print(f)print("with stop----------")

class5_004

  • f接受yield的返回值。
  • 如果with中有异常,那么上面例子中不能保证exit的执行。需要在foo函数中增加try finally
from contextlib import contextmanager@contextmanager
def foo():print("enter")try:yield 5except :print("异常被捕获")finally:print("exit")with foo() as f:print(f)raise Exception()

class5_005

  1. 当yield发生处为生成器函数增加了上下文管理。这是为函数增加上下文机制的方式。
    • 把yield之前的当做__enter__方法执行
    • 把yield之后的当做__exit__方法执行
    • 把yield的值作为__enter__的返回值
  • 应用举例:实现时间统计
from contextlib import contextmanager
import datetime,time@contextmanager
def foo(fn):start = datetime.datetime.now()try:yield fnexcept:raisefinally:stop = (datetime.datetime.now()- start).total_seconds()print("运行时长:{}".format(stop))def add(x,y):time.sleep(1)return x+ywith foo(add) as a:print(a(2,3),a(1,2))print(add(4,5),add(5,6))

class5_006

  • 总结:
    如果业务逻辑简单可以使用函数加contextlib.contextmanager装饰器方式,如果业务复杂,用类的方式加 __enter____exit__ 方法方便

这篇关于Python中类的魔术方法之上下文管理的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!


原文地址:
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.chinasem.cn/article/1102654

相关文章

SpringBoot中使用Flux实现流式返回的方法小结

《SpringBoot中使用Flux实现流式返回的方法小结》文章介绍流式返回(StreamingResponse)在SpringBoot中通过Flux实现,优势包括提升用户体验、降低内存消耗、支持长连... 目录背景流式返回的核心概念与优势1. 提升用户体验2. 降低内存消耗3. 支持长连接与实时通信在Sp

Conda虚拟环境的复制和迁移的四种方法实现

《Conda虚拟环境的复制和迁移的四种方法实现》本文主要介绍了Conda虚拟环境的复制和迁移的四种方法实现,包括requirements.txt,environment.yml,conda-pack,... 目录在本机复制Conda虚拟环境相同操作系统之间复制环境方法一:requirements.txt方法

Nginx 重写与重定向配置方法

《Nginx重写与重定向配置方法》Nginx重写与重定向区别:重写修改路径(客户端无感知),重定向跳转新URL(客户端感知),try_files检查文件/目录存在性,return301直接返回永久重... 目录一.try_files指令二.return指令三.rewrite指令区分重写与重定向重写: 请求

MySQL 打开binlog日志的方法及注意事项

《MySQL打开binlog日志的方法及注意事项》本文给大家介绍MySQL打开binlog日志的方法及注意事项,本文通过实例代码给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要... 目录一、默认状态二、如何检查 binlog 状态三、如何开启 binlog3.1 临时开启(重启后失效)

基于Python开发Windows屏幕控制工具

《基于Python开发Windows屏幕控制工具》在数字化办公时代,屏幕管理已成为提升工作效率和保护眼睛健康的重要环节,本文将分享一个基于Python和PySide6开发的Windows屏幕控制工具,... 目录概述功能亮点界面展示实现步骤详解1. 环境准备2. 亮度控制模块3. 息屏功能实现4. 息屏时间

Python如何去除图片干扰代码示例

《Python如何去除图片干扰代码示例》图片降噪是一个广泛应用于图像处理的技术,可以提高图像质量和相关应用的效果,:本文主要介绍Python如何去除图片干扰的相关资料,文中通过代码介绍的非常详细,... 目录一、噪声去除1. 高斯噪声(像素值正态分布扰动)2. 椒盐噪声(随机黑白像素点)3. 复杂噪声(如伪

Python中图片与PDF识别文本(OCR)的全面指南

《Python中图片与PDF识别文本(OCR)的全面指南》在数据爆炸时代,80%的企业数据以非结构化形式存在,其中PDF和图像是最主要的载体,本文将深入探索Python中OCR技术如何将这些数字纸张转... 目录一、OCR技术核心原理二、python图像识别四大工具库1. Pytesseract - 经典O

基于Linux的ffmpeg python的关键帧抽取

《基于Linux的ffmpegpython的关键帧抽取》本文主要介绍了基于Linux的ffmpegpython的关键帧抽取,实现以按帧或时间间隔抽取关键帧,文中通过示例代码介绍的非常详细,对大家的学... 目录1.FFmpeg的环境配置1) 创建一个虚拟环境envjavascript2) ffmpeg-py

python使用库爬取m3u8文件的示例

《python使用库爬取m3u8文件的示例》本文主要介绍了python使用库爬取m3u8文件的示例,可以使用requests、m3u8、ffmpeg等库,实现获取、解析、下载视频片段并合并等步骤,具有... 目录一、准备工作二、获取m3u8文件内容三、解析m3u8文件四、下载视频片段五、合并视频片段六、错误

Python中提取文件名扩展名的多种方法实现

《Python中提取文件名扩展名的多种方法实现》在Python编程中,经常会遇到需要从文件名中提取扩展名的场景,Python提供了多种方法来实现这一功能,不同方法适用于不同的场景和需求,包括os.pa... 目录技术背景实现步骤方法一:使用os.path.splitext方法二:使用pathlib模块方法三