您现在的位置是：首页 > Python

当前栏目

Python上下文管理器

Python 文件对象

2023-03-02 11:13:19 时间

### 1. 上下文管理器 ###

一个类只要实现了 `__enter__()` 和 `__exit__()` 这个两个方法，通过该类创建的对象我们就称之为上下文管理器。

上下文管理器可以使用 with 语句，**with语句之所以这么强大，背后是由上下文管理器做支撑的**，也就是说刚才使用 open 函数创建的文件对象就是就是一个上下文管理器对象。

**自定义上下文管理器类,模拟文件操作:**

定义一个File类，实现 `__enter__()` 和 `__exit__()` 方法，然后使用 with 语句来完成操作文件， 示例代码：

# 自定义上下文管理器类
    class File(object):
        def __init__(self, file_name, file_mode):
            self.file_name = file_name
            self.file_mode = file_mode
    
        def __enter__(self):
            # 上文方法，负责返回操作对象资源，比如：文件对象，数据库连接对象
            self.file = open(self.file_name, self.file_mode)
            return self.file
    
        def __exit__(self, exc_type, exc_val, exc_tb):
            # 下文方法，负责释放对象资源，比如：关闭文件，关闭数据库连接对象
            self.file.close()
            print('over')
    
    
    # with语句 结合上下文管理器对象使用
    with File('1.txt', 'r') as f:
        # content = f.read()
        # print(content)
        f.write('qqq')  # 报错，但是仍然执行了关闭连接操作

**代码说明:**

*  `__enter__` 表示上文方法，需要返回一个操作文件对象
 *  `__exit__` 表示下文方法，with语句执行完成会自动执行，即使出现异常也会执行该方法

### 2. 上下文管理器装饰器方式实现 ###

假如想要让一个函数成为上下文管理器，Python 还提供了一个 @contextmanager 的装饰器，更进一步简化了上下文管理器的实现方式。通过 yield 将函数分割成两部分，yield 上面的语句在 `__enter__` 方法中执行，yield 下面的语句在 `__exit__` 方法中执行，紧跟在 yield 后面的参数是函数的返回值。

from contextlib import contextmanager
    
    
    # 加上装饰器，那么下面函数创建的对象就是一个上下文管理器
    @contextmanager
    def my_open(file_name, file_mode):
        global file
        try:
            file = open(file_name, file_mode)
            # yield关键字之前的代码可以认为是上文方法，负责返回操作对象资源
            yield file
        except Exception as e:
            print(e)
        finally:
            # yield关键字后面的代码可以认为是下文方法，负责释放操作对象的资源
            file.close()
            print('over')
    
    
    # 普通函数不能结合with语句使用
    with my_open('1.txt', 'r') as file:
        # content = file.read()
        # print(content)
        file.write('1')

猜你喜欢

测试如何提高自己的话语权
从一个小角度观察敏捷实践
专项测试怎样才“好玩”
测试人员如何甩锅
你还记得测试策略么
测试用例，写不写？
测试的最终产物是什么
体系认证、白皮书与行业大会
【智慧电力巡检】基于EasyCVR视频技术构建远程监控综合管理系统
数据什么会走丢了呢？
团队交付质量如何评估
测试中的数学问题
TPS、并发数与线程数，傻傻分不清楚？
网站源代码安全审计之wordpress漏洞
质量与效能的再思考
从实测出发，掌握 NebulaGraph Exchange 性能最大化的秘密
我眼中的性能测试
构建性能测试知识体系
敏捷测试系列文章合集
测试基础10问-上

zl程序教程

当前栏目

Python上下文管理器

相关文章