python中利用上下文管理器来实现mysql数据库的封装

from pymysql import connect

class DB(object):
def __init__(self, password, database):
# 1.连接数据库
self.conn = connect(host='localhost', port=3306, user='root', password=str(password), database=str(database),
charset='utf8')
# 2.创建游标
self.cs = self.conn.cursor()

def __enter__(self):
return self.cs

def __exit__(self, exc_type, exc_val, exc_tb):
# 3.关闭
self.cs.close()
self.conn.close()

# 利用上下文管理器来封装的数据库的操作，只需要传入密码和需要连接的数据库就可以了
with DB("mysql", "student_selection") as db:
# 需要执行的数据库操作（sql语句）
db.execute("select * from student; ")
# 循环取出查找的数据
data = db.fetchall()
for content in data:
print(content)