首页 > 其他 > 详细

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

时间:2019-04-09 10:22:24      阅读:139      评论:0      收藏:0      [点我收藏+]
 
# .开始之前先明确一下with机制
# 1.类包函数__enter__()和__exit__()函数,即是可以被上下文管理的类
# __enter__用来执行with时的方法,__exit__返回对象给with的as 对象
# 2.通过函数实现要用到contextmanager装饰器与yield关键字

# 一、通过类来实现支持whith上下文管理功能
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测试上下文管理器")
#用以上两种方法来实现相同的功能

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

原文:https://www.cnblogs.com/yhleng/p/10674986.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!