上下文管理器之__enter__和__exit__
前言
有个学生在第四轮面试中被CTO问到:如何自定义实现with open的功能。然后就一脸懵逼的回来找我了……
看到with open
大家都应该知道,这是在问关于上下文管理协议,这个所谓的上下文管理协议一般是指我们在文件操作中常用的那个with ... as ..
。
Python允许with语句使用上下文管理器,而上下文管理器则是支持__enter__
与__exit__
方法的对象。__enter__
方法没有参数,当程序执行到with
语句时被调用,返回值绑定在文件对象f
上,而__exit__
方法则有三个参数,包括异常类型,异常对象,异常回溯。
with as是如何工作的
我们通过示例来了解with ... as ...
语句是如何工作的,先来个简单的:
class MyOpen1(object):
def __enter__(self):
print('当with语句开始执行时,我被执行啦…………')
def __exit__(self, exc_type, exc_val, exc_tb):
print('当with语句执行完毕,我被执行啦…………')
with MyOpen1() as f1:
print('假装再做文件操作…………')
'''打印结果
当with语句开始执行时,我被执行啦…………
假装再做文件操作…………
当with语句执行完毕,我被执行啦…………
'''
上例的打印结果印证了我们之前所说。并且,我们知道当MyOpen1
加括号实例化的时候,还触发了实例化方法__init__
的执行:
class MyOpen2(object):
def __init__(self):
print('我是__init__方法,当 {} 加括号时,我被执行啦…………'.format(self.__class__.__name__))
def __enter__(self):
print('当with语句开始执行时,我被执行啦…………')
def __exit__(self, exc_type, exc_val, exc_tb):
print('当with语句执行完毕,我被执行啦…………')
with MyOpen2() as f2:
print('假装再做文件操作…………')
'''打印结果
我是__init__方法,当 MyOpen2 加括号时,我被执行啦…………
当with语句开始执行时,我被执行啦…………
假装再做文件操作…………
当with语句执行完毕,我被执行啦…………
'''
自定制open方法
由上面的两个示例我们隐约可以明白with ... as ..
语句内部是怎么玩的了。我们可以试着更深一步的完善代码:
class MyOpen3(object):
def __init__(self, file_obj, mode='r', encoding='utf-8'):
self.file_obj = file_obj
self.mode = mode
self.encoding = encoding
def __enter__(self):
self.f = open(file=self.file_obj, mode=self.mode, encoding=self.encoding)
return self.f
def __exit__(self, exc_type, exc_val, exc_tb):
self.f.close()
with MyOpen3('a.txt', mode='w', encoding='utf-8') as f3:
f3.write('custom with ... as ...')
结果已经不用多说,完美!
更多的示例
更多的示例和其他的扩展:
class MyOpen(object):
""" 手动实现with open """
def __init__(self, file_obj, mode='r', encoding=None):
"""
当实例化的时候,需要传递三个参数
:param file_obj: 文件名称(路径)
:param mode: 操作模式
:param encoding: 编码
"""
# 调用内置函数open,获取文件对象f
self.f = open(file_obj, mode=mode, encoding=encoding)
def __enter__(self):
"""
当with语句被调用时,返回文件对象f
:return: f
"""
print('with语句触发__enter__执行')
return self
def __exit__(self, exc_type, exc_value, traceback):
"""
退出关联到此对象的运行时上下文。 各个参数描述了导致上下文退出的异常。 如果上下文是无异常地退出的,三个参数都将为 None
:param exc_type:
:param exc_value:
:param traceback:
:return:
"""
self.f.close()
print('文件对象 f 已被关闭')
def read(self):
"""
自定义读方法
"""
print('自定义的read方法被执行')
return self.f.read()
def write(self, content):
self.f.write(content)
def __getattr__(self, item):
"""
当自定义的文件对象调用MyOpen类中没实现个方法时,就通过getattr,从内置函数的open对象中返回
:param item: 比如seek方法
:return: 将seek方法返回
"""
return getattr(self.f, item)
f = MyOpen('a.txt', 'a', "utf8")
f.write('我爱北京天安门')
f.close()
f = MyOpen('a.txt', 'r', "utf8")
print(f.read())
f.close()
with MyOpen('a.txt', 'r+', "utf8") as f1:
# print(f1.__doc__) # 通过打印结果可以看到,with语句使用__enter__方法将MyOpen实例化的对象赋值给f1
f1.seek(3) # 注意,由于utf8编码对普通中文用3个字节表示,所以seek的时候,需要注意
f1.write('我是你爸爸')
print(f1.read())
再来一个搭配pickle
的:
ps:原谅洒家无耻的拷贝!
import pickle
class MyPickledump:
def __init__(self,path):
self.path = path
def __enter__(self):
self.f = open(self.path, mode='ab')
return self
def dump(self,content):
pickle.dump(content,self.f)
def __exit__(self, exc_type, exc_val, exc_tb):
self.f.close()
class Mypickleload:
def __init__(self,path):
self.path = path
def __enter__(self):
self.f = open(self.path, mode='rb')
return self
def __exit__(self, exc_type, exc_val, exc_tb):
self.f.close()
def load(self):
return pickle.load(self.f)
def loaditer(self):
while True:
try:
yield self.load()
except EOFError:
break
# with MyPickledump('file') as f:
# f.dump({1,2,3,4})
with Mypickleload('file') as f:
for item in f.loaditer():
print(item)
with和pickle
更多的:
import pickle
class MyPickledump:
def __init__(self,path):
self.path = path
def __enter__(self):
self.f = open(self.path, mode='ab')
return self
def dump(self,content):
pickle.dump(content,self.f)
def __exit__(self, exc_type, exc_val, exc_tb):
self.f.close()
class Mypickleload:
def __init__(self,path):
self.path = path
def __enter__(self):
self.f = open(self.path, mode='rb')
return self
def __exit__(self, exc_type, exc_val, exc_tb):
self.f.close()
def __iter__(self):
while True:
try:
yield pickle.load(self.f)
except EOFError:
break
# with MyPickledump('file') as f:
# f.dump({1,2,3,4})
with Mypickleload('file') as f:
for item in f:
print(item)
with和pickle和iter
see also: Python官网关于with的描述 | PEP 343 with语句 | with 语句上下文管理器 | Context Manager Types | Python概念-上下文管理协议中的__enter__和__exit__ | python之路——面向对象进阶
欢迎斧正,that's all
上下文管理器之__enter__和__exit__的更多相关文章
- Python上下文管理协议:__enter__和__exit__
上下文管理器(context manager)是Python2.5开始支持的一种语法,用于规定某个对象的使用范围.一旦进入或者离开该使用范围,会有特殊操作被调用 (比如为对象分配或者释放内存).它的语 ...
- python基础----实现上下文管理协议__enter__和__exit__
我们知道在操作文件对象的时候可以这么写 with open('a.txt') as f: '代码块' 上述叫做上下文管理协议,即with语句,为了让一个对象兼容with语句,必须在这个对象的类中声明_ ...
- __enter__,__exit__上下文管理协议
上下文管理协议__enter__,__exit__ 用途或者说好处: 1.使用with语句的目的就是把代码块放入with中执行,with结束后,自动完成清理工作,无须手动干预 2.在需要管理一些资源比 ...
- (转)Python中的上下文管理器和Tornado对其的巧妙应用
原文:https://www.binss.me/blog/the-context-manager-of-python-and-the-applications-in-tornado/ 上下文是什么? ...
- (转)contextlib — 上下文管理器工具
原文:https://pythoncaff.com/docs/pymotw/contextlib-context-manager-tool/95 这是一篇社区协同翻译的文章,你可以点击右边区块信息里的 ...
- Pthon魔术方法(Magic Methods)-上下文管理
Pthon魔术方法(Magic Methods)-上下文管理 作者:尹正杰 版权声明:原创作品,谢绝转载!否则将追究法律责任. 一.上下文管理方法 __enter__: 进入与此对象相关的上下文.如果 ...
- Python高级笔记(八)with、上下文管理器
1. 上下文管理器 __enter__()方法返回资源对象,__exit__()方法处理一些清除资源 如:系统资源:文件.数据库链接.Socket等这些资源执行完业务逻辑之后,必须要关闭资源 #!/u ...
- Python - Context Manager 上下文管理器
什么是上下文管理器 官方解释... 上下文管理器是一个对象 它定义了在执行 with 语句时要建立的运行时上下文 上下文管理器处理进入和退出所需的运行时上下文以执行代码块 上下文管理器通常使用 wit ...
- Python进阶(上下文管理器与with语句)
/*上下文管理器必须有__enter__和__exit__方法*/ class MyResource: def __enter__(self): print('链接资源') return self / ...
随机推荐
- NOI2008 志愿者招募 (费用流)
题面 申奥成功后,布布经过不懈努力,终于成为奥组委下属公司人力资源部门的主管.布布刚上任就遇到了一个难题:为即将启动的奥运新项目招募一批短期志愿者.经过估算,这个项目需要N 天才能完成,其中第i 天至 ...
- Druid连接池 报错:abandon connection原因分析
问题现象:使用Druid的数据库连接池,在进行一个查询SQL的时候,抛出了异常: [2017-10-20 01:40:59.269 ERROR com.alibaba.druid.pool.Druid ...
- Redis学习之一--基础知识
一.定义 REmote DIctionary Server(Redis) 是一个以字典结构存储数据的key-value存储系统:使用ANSI C语言编写.遵守BSD协议.支持网络.可基于内存亦可持久化 ...
- Cogs 2221. [SDOI2016 Round1] 数字配对(二分图)
[SDOI2016 Round1] 数字配对 ★★★ 输入文件:menci_pair.in 输出文件:menci_pair.out 简单对比 时间限制:1 s 内存限制:128 MB [题目描述] 有 ...
- 解决vue多个路由共用一个页面的问题
在日常的vue开发中我们可能会遇见多个路由需要共用一个页面的需求,特别是当路由是通过动态添加的,不同的路由展示的东西只是数据不同其他没有变化.例如: ? 1 2 3 4 5 6 7 8 9 10 11 ...
- IDEA中获取资源路径问题
更正 以src开始,就能用相对路径了... shift+ctrl+alt+s 调出项目结构, 在Modules里,就是设置 Sources Resources Test的界面, 右面的路径就是相对路径 ...
- [信息收集]Nmap命令详解
0x00[介绍] Nmap,也就是Network Mapper,中文为"网络映射器". Nmap是一款开源的网络探测和安全审核的工具,它的设计目标是快速地扫描大型网络. 它是网络管 ...
- Python实现协程
什么是进程和线程 有一定基础的小伙伴们肯定都知道进程和线程. 进程是什么呢? 直白地讲,进程就是应用程序的启动实例.比如我们运行一个游戏,打开一个软件,就是开启了一个进程. 进程拥有代码和打开的文件资 ...
- Vue源码分析(二) : Vue实例挂载
Vue源码分析(二) : Vue实例挂载 author: @TiffanysBear 实例挂载主要是 $mount 方法的实现,在 src/platforms/web/entry-runtime-wi ...
- DELPHI解析JSON格式化的日期
DELPHI解析JSON格式化的日期 json返回的日期是 /Date(1560355200000)/ 这样的格式. 这个1560355200000,是指1970年以后的秒数. DELPHI如何解析这 ...