python - 内置类型 之 上下文管理器类型

目录

1.上下文管理器例子

2.什么是上下文管理器

3.如何写上下文管理器

4.contextlib模块

---

参考:https://www.cnblogs.com/wongbingming/p/10519553.html

1.上下文管理器例子

with open('test.txt') as f:
    print f.readlines()

 

2.什么是上下文管理器

• 上下文管理器就是实现了上下文管理协议的对象。主要用于保存和恢复各种全局状态，关闭文件等，上下文管理器本身就是一种装饰器。

基本语法

with EXPR as VAR:
    BLOCK

 

3.如何写上下文管理器

简单点说，就是在一个类里，实现了__enter__和__exit__的方法，这个类的实例就是一个上下文管理器。

class Resource():
    def __enter__(self):
        print('===connect to resource===')
        return self
    def __exit__(self, exc_type, exc_val, exc_tb):
        print('===close resource connection===')
        
    def operate(self):
        print('===in operation===')
        
with Resource() as res:
    res.operate()

 

4.contextlib模块

•  python提供内建的模块contextlib来实现上下文管理器机制

import contextlib

@contextlib.contextmanager
def open_func(file_name):
    # __enter__方法
    print('open file:', file_name, 'in __enter__')
    file_handler = open(file_name, 'r')
    
    # 【重点】：yield
    yield file_handler

    # __exit__方法
    print('close file:', file_name, 'in __exit__')
    file_handler.close()
    return

with open_func('/Users/MING/mytest.txt') as file_in:
    for line in file_in:
        print(line)