# __enter__、__exit__ 用来对上下文管理
class content:def __init__(self, filepath, mode):self.filepath = filepathself.mode = modeself.encoding = 'utf-8'self.fileobject = Nonedef __enter__(self):# 可以用来:打开文件、链接、数据操作self.fileobject = open(self.filepath, self.mode, encoding=self.encoding) # 得到一个文件对象,然后进行返回return selfdef __exit__(self, exc_type, exc_val, exc_tb):# 关闭 or 结束self.fileobject.close() # 操作完成后,进行关闭文件def send(self):self.fileobject.write('发送\n') # 通过获取的文件对象进行写操作def read(self):self.fileobject.write('发送222\n')# 写法1
obj = content('test', 22)# 如果需要支持 with,需要在类方法中,增加__enter、__exit__方法,否则报错with obj as xxx: # with的是一个对象,只要with这个对象时,自动执行__enter 方法,print(xxx) # 如果enter中return 123,此时as xxx 中的:xxx 就等于123# 写法2
# 也支持下面的这种写法:
with content('test.txt', 'a') as obj:print(obj) # 当with里面的代码执行完后,自动执行 __exit__ 方法# 此时执行的 obj.send/obj.read 都 在 __enter__ 之后obj.send()obj.read()