python实现可以被with上下文管理的类或函数

枫铃3年前 (2021-07-09)Python244

开始之前先明确一下with机制
1.类包函数__ enter__()和__exit__()函数,即是可以被上下文管理的类
__enter__用来执行with时的方法,__exit__返回对象给with的as 对象
2.通过函数实现要用到contextmanager装饰器与yield关键字

一、通过类来实现支持whith上下文管理功能

'''
遇到问题没人解答?小编创建了一个Python学习交流QQ群:857662006 
寻找有志同道合的小伙伴,互帮互助,群里还有不错的视频学习教程和PDF电子书!
'''
class fp(object):
  def __init__(self, fileobj, mode):
    self.fileobj = open(fileobj, mode)

  def __enter__(self):
    return self.fileobj

  def __exit__(self, type, value, Traceback):
    self.fileobj.close()
    return True

with fp("abc1.txt", "w+") as fi:
  fi.write("测试上下文管理器")

二、通过函数yield关键字实现

from contextlib import contextmanager

@contextmanager
def fp(fileobj, mode):
      f = open(fileobj, mode)
      yield f
      f.close()


with fp("abc2.txt", "w+") as fi:
  fi.write("context测试上下文管理器")

用以上两种方法来实现相同的功能

相关文章

Python中一些高效的数据操作

1.列表统计 char...

python字符串/列表/字典互相转换

字符串与列表 字符串转列表 1.整体转换 str1 = 'hello world' print(str1.split('...

Python老司机给上路新手的3点忠告

Python老司机给上路新手的3点忠告

在你学习编程的过程中,是否有过以下经历,或正在面临类似的局面: ● 网上找了很多资料,不知道从哪里看...

Django中六个常用的自定义装饰器

装饰器作用 decorator是当今最流行的设计模式之一,很多使用它的人并不知道它是一种设计模式。使用它可以很方便地修改对象行为,...

发表评论

访客

看不清,换一张

◎欢迎参与讨论,请在这里发表您的看法和观点。