Python 有很多魔法方法，本文记录一下可以自定义 with 语句的上下文管理器所使用到的两个魔法方法，也就是 __enter__ 和 __exit__ 方法的实用性。
自定义上下文管理类
最常见的 with 语句就是 open 函数了，这里不做解释，直接来看一个自定义类的例子。
代码语言：javascript复制class TestHandler():
def __init__(self):
pass
def __enter__(self):
return self
def __exit__(self, exc_type, exc_val, exc_tb):
print('exc_type:', exc_type)
print('exc_val:', exc_val)
print('exc_tb:', exc_tb)
def func(self):
print(1 1)
def bad_func(self):
print('a' 1)上面定义了一个类，这个类定义了两个打印值的方法，其中一个方法会报错，同时该类使用到了两个魔法方法，有了这两个方法，这个类就可以使用 with 语句来进行调用，来看看调用正常函数的结果：
代码语言：javascript复制with TestHandler() as t:
t.func()结果如下：
代码语言：javascript复制2
exc_type: None
exc_val: None
exc_tb: None再来看看调用报错函数的结果
代码语言：javascript复制with TestHandler() as t:
t.bad_func()exc_type: <class 'TypeError'>
exc_val: Can't convert 'int' object to str implicitly
exc_tb: <traceback object at 0x0000021CEB484B08>
Traceback (most recent call last):
File "D:/Mycode/TestCase/mark.py", line 23, in <module>
t.bad_func()
File "D:/Mycode/TestCase/mark.py", line 17, in bad_func
print('a' 1)
TypeError: Can't convert 'int' object to str implicitly从上面两次调用，可以看到，__exit__ 函数里面的三个参数（定义函数的时候默认会要求加入）分别代表了报错类型、报错原因、报错追溯，只有当 with 语句调用报错时候，这三个参数才有值，否则就是 None，看到这里，你是否能够想到什么？可以利用这三个参数进行异常判断和处理。
上下文管理实用性
已经知道如何定义 with 语句了，也知道遇到异常会出现什么，那么现在来看看自定义 with 语句的使用场景有哪些。
with 语句比较适合的场景是打开->操作->关闭，在我们常用的除了文件操作外，还有数据库操作、SSH 操作会涉及这个过程，所以，直接看看这两个操作的例子。
代码语言：javascript复制import sqlite3
class DBHandler():
def __init__(self, database):
self.database = database
self.conn = sqlite3.connect(self.database)
self.cursor = self.conn.cursor()
def __enter__(self):
return self.cursor
def __exit__(self, exc_type, exc_val, exc_tb):
if exc_type is None:
self.conn.commit()
self.conn.close()上面这个关于数据操作的例子就很典型，它包括了数据库连接、数据库操作（with 语句之后）、异常处理、数据库关闭连接等操作。
看一下 with 语句的使用：
代码语言：javascript复制with DBHandler(database) as db:
db.executescript(create_sql)是不是非常的方便，当然，如果再结合 try 语句来进行连接操作，就更安全可靠。
再来看看 SSH 的操作例子
代码语言：javascript复制import paramiko
class SSHHandler():
def __init__(self, host, username, password, port=22):
self.host = host
self.username = username
self.password = password
self.port = port
self.ssh = paramiko.SSHClient()
self.ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy)
self.ssh.connect(self.host, self.port, self.username, self.password)
def __enter__(self):
return self.ssh
def __exit__(self, exc_type, exc_val, exc_tb):
self.ssh.close()很明显，上面的自定义类 with 语句返回的是一个 SSHClient 对象，所以使用时直接按照这个对象的方法调用即可，调用结束会自动断开连接。
总结：自定义 with 语句简单理解就是非常适合一些“有始有终”的场景，通过自定义上下文管理器，可以把一些需要重复执行的固定操作简化，只需要关注特定的操作本身。
版权声明：如无特殊说明，文章均为本站原创，转载请注明出处 本文链接：https://cloud.tencent.com/developer/article/2123566 许可协议：署名-非商业性使用 4.0 国际许可协议
