上下文管理器with的应用场景

上下文管理器的应用场景：
资源管理功能，即文件、网络连接、数据库连接等操作时需要关闭资源。

也可以在代码执行前后增加功能，类似于装饰器，比如代码之前做权限验证等。

一个数据库连接的使用：

import pymysql
class DBConnection(object):
def init(self,ip,user,passwd,db):
self.ip = ip
self.user = user
self.passwd = passwd
self.db = db

def __enter__(self):
    self.conn = pymysql.connect(self.ip, user=self.user, passwd=self.passwd, db=self.db)
    self.cur = conn.cursor()
    return self.cur

def __exit__(self, exc_type, exc_val, exc_tb):
    self.cur.close()
    self.conn.close()

with DBConnection(‘192.168.121.xxx’, user=“xxx”, passwd=“123456”, db=“xxx”) as cur:
cur.execute(“select * from studnet;”)
result = cur.fetchall()
print(result)
完成DBConnection这个类，每次连接数据库时，只要简单的调用with语句即可，不需要关心数据库的关闭、异常等。