测试开发技术网站

[TOC]
# 1、with上下文管理器的用法
* with用的最多的可能就是打开文件，读写文件的场景，如下代码：

```python
with open("demo.txt","w+",encoding="utf8") as f:
    f.write("hello world")
```
其等同于如下代码：

```python
f=open("demo.txt","w+",encoding="utf8")
f.write("hello world")
f.close()
```
使用with语句的时候不需要显式的去关闭文件资源，因为它自动会关闭，那么这个自动关闭是怎么实现的呢，这其实就是__enter__和__exit__魔法函数在起作用

# 2、\_\_enter__和__exit__魔法函数的工作原理
* 以上面with   as 的形式，那么首先是进入类函数中__enter__魔法函数，__enter__魔法函数返回结果赋值给as后面的变量，当with as 语句结束时，会自动调用__exit__魔法函数

下面重写写一个类似open的类，来实现with  as 打开文件读写文件的实例，如下：

```python

class OpenFile(object):
    def __init__(self,file_name="",mode="r",encoding="utf8",**kwargs):
        print("-----------------in init------------------")
        self.__file=open(file_name,mode,encoding=encoding)

def __enter__(self):
        print("-----------------in enter ----------------")
        return self

def write(self,ctx):
        print("-----------------begin write-------------")
        self.__file.write(ctx)
        print("-----------------finish write-------------")

def __exit__(self,exec_type,exec_value,exec_tb):
        print("-----------------in exit------------------")
        self.__file.close()

with OpenFile("demo.txt","w+",encoding="utf8") as f:
    f.write("hello world")
```

执行结果如下：

```python
-----------------in init------------------
-----------------in enter ----------------
-----------------begin write-------------
-----------------finish write-------------
-----------------in exit------------------
```

因此，如果想实现一个自定义的上下文管理器，只需要在自定义类中实现这两个魔法函数即可，这里有一个关键点需要理解的就是with   as 语句中as 后面的变量就是__enter__魔法函数返回值，理解了这一点，就很容易实现自己的上下文管理器了

测试开发技术网站

博客