实现文件上下文管理(__enter__和__exit__)

目录

一、上下文管理协议

二、模拟open

三、优点


python从小白到总裁完整教程目录:https://blog.csdn.net/weixin_67859959/article/details/129328397?spm=1001.2014.3001.5502

  • 我们知道在操作文件对象的时候可以这么写
python 复制代码
with open('a.txt') as f:
    '代码块'
  • 上述叫做上下文管理协议,即with语句,为了让一个对象兼容with语句,必须在这个对象的类中声明__enter__和__exit__方法

一、上下文管理协议

python 复制代码
class Open:
    def __init__(self, name):
        self.name = name

    def __enter__(self):
        print('出现with语句,对象的__enter__被触发,有返回值则赋值给as声明的变量')
        # return self
    def __exit__(self, exc_type, exc_val, exc_tb):
        print('with中代码块执行完毕时执行我啊')


with Open('a.txt') as f:
    print('=====>执行代码块')
    # print(f,f.name)

出现with语句,对象的__enter__被触发,有返回值则赋值给as声明的变量

=====>执行代码块

with中代码块执行完毕时执行我啊

  • exit()中的三个参数分别代表异常类型,异常值和追溯信息,with语句中代码块出现异常,则with后的代码都无法执行
python 复制代码
class Open:
    def __init__(self, name):
        self.name = name

    def __enter__(self):
        print('出现with语句,对象的__enter__被触发,有返回值则赋值给as声明的变量')

    def __exit__(self, exc_type, exc_val, exc_tb):
        print('with中代码块执行完毕时执行我啊')
        print(exc_type)
        print(exc_val)
        print(exc_tb)


try:
    with Open('a.txt') as f:
        print('=====>执行代码块')
        raise AttributeError('***着火啦,救火啊***')
except Exception as e:
    print(e)
PYTHON 复制 全屏

出现with语句,对象的__enter__被触发,有返回值则赋值给as声明的变量

=====>执行代码块

with中代码块执行完毕时执行我啊

<class 'AttributeError'>

***着火啦,救火啊***

<traceback object at 0x1065f1f88>

***着火啦,救火啊***

  • 如果__exit()返回值为True,那么异常会被清空,就好像啥都没发生一样,with后的语句正常执行
python 复制代码
class Open:
    def __init__(self, name):
        self.name = name

    def __enter__(self):
        print('出现with语句,对象的__enter__被触发,有返回值则赋值给as声明的变量')

    def __exit__(self, exc_type, exc_val, exc_tb):
        print('with中代码块执行完毕时执行我啊')
        print(exc_type)
        print(exc_val)
        print(exc_tb)
        return True


with Open('a.txt') as f:
    print('=====>执行代码块')
    raise AttributeError('***着火啦,救火啊***')
print('0' * 100)  #------------------------------->会执行

出现with语句,对象的__enter__被触发,有返回值则赋值给as声明的变量

=====>执行代码块

with中代码块执行完毕时执行我啊

<class 'AttributeError'>

***着火啦,救火啊***

<traceback object at 0x1062ab048>

0000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000

二、模拟open

python 复制代码
class Open:
    def __init__(self, filepath, mode='r', encoding='utf-8'):
        self.filepath = filepath
        self.mode = mode
        self.encoding = encoding

    def __enter__(self):
        # print('enter')
        self.f = open(self.filepath, mode=self.mode, encoding=self.encoding)
        return self.f

    def __exit__(self, exc_type, exc_val, exc_tb):
        # print('exit')
        self.f.close()
        return True

    def __getattr__(self, item):
        return getattr(self.f, item)


with Open('a.txt', 'w') as f:
    print(f)
    f.write('aaaaaa')
    f.wasdf  #抛出异常,交给__exit__处理

<_io.TextIOWrapper name='a.txt' mode='w' encoding='utf-8'>

三、优点

  1. 使用with语句的目的就是把代码块放入with中执行,with结束后,自动完成清理工作,无须手动干预
  2. 在需要管理一些资源比如文件,网络连接和锁的编程环境中,可以在__exit__中定制自动释放资源的机制,你无须再去关系这个问题,这将大有用处
相关推荐
聚客AI9 分钟前
系统掌握PyTorch:图解张量、Autograd、DataLoader、nn.Module与实战模型
人工智能·pytorch·python·rnn·神经网络·机器学习·自然语言处理
狮子也疯狂16 分钟前
基于Python的气象数据分析及可视化研究
python·信息可视化·数据分析
甜甜的资料库24 分钟前
基于微信小程序的睡眠宝系统源码数据库文档
数据库·微信小程序·小程序
代码老y36 分钟前
Redis:现代应用开发的高效内存数据存储利器
数据库·redis·缓存
蓝婷儿1 小时前
6个月Python学习计划 Day 18 - 项目实战 · 学生成绩管理系统(OOP版)
开发语言·python·学习
HINOTOR_1 小时前
DAY 25 异常处理
开发语言·python
yorushika_1 小时前
python打卡训练营打卡记录day49
开发语言·python·tensorboard·cbam
还是鼠鼠1 小时前
HTTP 请求协议简单介绍
java·开发语言·网络·网络协议·http
远方16091 小时前
15-Oracle 23ai Vector Search Similarity Search-向量相似性和混合搜索-实操
数据库·ai·oracle
qq_189370491 小时前
Jupyter notebook的文章结构目录查看方式和汉化方法
python·jupyter