魔术方法2-上下文管理

目录

上下文管理对象

1. __enter__

2.__exit__

3. __enter__方法和__exit__方法的参数

上下文管理应用

---

上下文​​​​​​​管理对象

当一个对象同时实现了__enter__()和__exit__()方法，它就属于上下文管理的对象

1. __enter__

进入与此对象相关的上下文。如果存在该方法，with语法会把该方法的返回值作为绑定到as子句中指定的变量上

2.__exit__

退出与此对象相关的上下文

class Test:
    def __init__(self):
        print("init")

    def __enter__(self):
        print("enter")

    def __exit__(self, exc_type, exc_val, exc_tb):
        print("exit")

with Test() as f:
    print("test")

说明：

• 实例化对象的时候，并不会调用__enter__，进入with语句块调用__enter__方法，然后执行语句体。最后离开with语句块的时候，调用 __exit__ 方法
• __enter__ 方法返回值就是上下文中使用的对象，with语法会把它的返回值赋给as子句的变量。

class Test:
    def __init__(self):
        print("init")

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

    def __exit__(self, exc_type, exc_val, exc_tb):
        print("exit")
T = Test()
with T as f:
    print(T == f)
    print("test")

注：with可以开启一个上下文运行环境，在执行前做一些准备工作，执行后做一些收尾工作

3. __enter__方法和__exit__方法的参数

__enter__方法没有其他参数
__exit__方法有3个参数
__exit__  (self，exc_type， exc_value，traceback)这三个参数都与异常有关。
如果该上下文退出时没有异常，这3个参数都为None。如果有异常，参数意义如下:
exc_type，异常类型
exc_value，异常的值

traceback,异常的追踪信息

__exit__方法返回一个等效True的值，则压制异常;否则，继续抛出异常

class Test:
    def __init__(self):
        print("init")

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

    def __exit__(self, exc_type, exc_val, exc_tb):
        print(exc_type)
        print(exc_val)
        print(exc_tb)
        print("exit")
        return "exit test"
T = Test()
with T as f:
    raise Exception('test error')
    print("test")

print("end")

上下文管理应用

• 增强功能：在代码执行的前后增加代码，以增强其功能。
• 资源管理：类似装饰器的功能。打开了资源需要关闭，例如文件对象、网络连接、数据库连接等
• 权限验证：在执行代码之前，做权限的验证，在__enter__中处理