python之with语句结合上下文管理器
所谓上下文管理器即在一个类中重写了__enter__方法和__exit__方法的类就可以成为上下文管理器类。
我们可以通过with语句结合上下文管理器简化一些操作。
使用with语句结合自定义上下文管理器完成数据库相应的操作，代码实现如下：
# 1. 导入模块
import pymysql
# 创建自定义上下文管理器对象
class MyDatabase(object):
# 接收参数并创建数据库连接对象
def __init__(self, host, port, user, passwd, database):
self.__db = pymysql.Connection(host, port, user, passwd, database, charset='utf8')
# 返回数据库连接对象
def __enter__(self):
return self.__db
# 关闭数据库连接
def __exit__(self, exc_type, exc_val, exc_tb):
self.__db.close()
def main():
# 使用with关键字接收enter返回的对象给db
with MyDatabase('localhost', 3306, 'root', 'mysql', 'JDDB') as db:
# 利用db创建游标
cur = db.cursor()
sql = '''select * from %s'''
cur.execute(sql, (goods,))
result = cur.fetchall()
for i in result:
print(i)
# 关闭游标
cur.close()
# 程序入口
if __name__ == '__main__':
main()
上下文管理器类的代码流程：
1.编写__init__方法用来接收参数，并创建数据库连接对象；
2.重写__enter__方法，返回数据库连接对象；
3.重写__exit__方法，用来关闭数据库连接；
with语句代码流程：
1.当将创建对象的语句放到with语句里时不会创建对象，而是接受__enter__方法返回的对象并给对象起个别名；
2.使用接受到的对象即数据库连接对象，创建游标；
3.编写SQL语句，并通过游标执行SQL语句；
4.获取SQL语句的查询结果，并显示出来；
5.关闭游标；
6.当with语句内的代码执行完毕后自动执行__exit__方法关闭数据库连接。
注意：with MyDatabase() as db ---> db = MyDatabase().__enter__()
利用with结合自定义上下文类实现HTTP服务端：
# 1.导入socket模块
import socket
class MySocket(object):
# 2.编写init方法接收port参数
def __init__(self, port):
self.__port = port
# 3.创建socket对象
self.__sk = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 4.编写enter方法返回套接字对象
def __enter__(self):
# 设置端口复用
self.__sk.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, True)
# 绑定端口
self.__sk.bind(self.__port)
# 设置端口监听
self.__sk.listen(128)
# 返回套接字对象
return self.__sk
# 5.编写exit方法关闭套接字对象
def __exit__(self, exc_type, exc_val, exc_tb):
self.__sk.close()
def main():
# 使用with关键字 并接受返回的套接字对象给sk
with MySocket(8000) as sk:
# 等待客户端连接
clicent, ip_port = sk.accept()
recv_data = clicent.recv(1024)
print(recv_data.decode('utf-8'))
# 编写HTTP响应报文
http_line = 'HTTP/1.1 GET 200 OK\r\n'
http_header = 'Server PWS/1.0\r\n'
http_body = 'Welcome to index!\r\n'
send_data = (http_line + http_header + '\r\n' + http_body).encode('utf-8')
clicent.send(send_data)
# 关闭客户端连接
clicent.close()
# 编写程序入口
if __name__ == '__main__':
main()
自定义上下文管理器类的代码解读：
1.编写__init__方法，用来接收参数并创建套接字对象；
2.编写__enter__方法，并使用套接字对象设置端口复用、绑定端口、并设置监听，然后返回套接字对象；
3.编写__exit__方法，关闭套接字对象。
with语句代码解读：
1.接收enter返回的套接字对象，并起个别名，
2.通过返回套接字对象等待客户端连接，
3.接收客户端连接成功后会返回一个新的套接字和IP端口号，
4.使用客户端套接字发送HTTP响应报文
5.关闭客户端连接
6.当with语句中的代码执行完毕后自动执行__exit__方法，关闭服务器连接
with管理器例题：
# 要求1：可以使用with Mysql('heima', 'root', 'mysql') as cursor: # # # 语句获取游标对象cursor,操作cursor即可完成数据库操作； # # # 要求2：Mysql类中三个参数分别对应要连接的数据库名、用户名、密码； # # # 要求3：如果数据库操作正常，上下文中自动提交（自动执行commit()方法）, # # 关闭游标并且关闭数据库连接； # # # 要求4：如果数据库操作出现异常，上下文中自动回滚（自动执行rollback()方法）， # 关闭游标并且关闭数据库连接，捕获异常并打印报错信息。 import pymysql global bool class Mysql(object): def __init__(self, database, user, password): self.__db = pymysql.Connection(host='localhost', port=3306, database=database, user=user, password=password, charset='utf8') def __enter__(self): self.__cur = self.__db.cursor() return self.__cur def __exit__(self, exc_type, exc_val, exc_tb): if exc_type: self.__db.rollback()
print(exc_type)
print(exc_val)
print(exc_tb) else: self.__db.commit() self.__cur.close() self.__db.close() return True def main(): with Mysql('heima', 'root', 'mysql') as cursor: sql = '''SQL语句''' cursor.execute(sql) if __name__ == '__main__': main()
注意：__exit(self, exc_type, exc_val, exc_tb)__函数中的三个参数exc_type, exc_val, exc_tb只有在程序异常退出时才会有值，
只要判断这三个参数中的其中一个，如果有值那就说明程序存在异常。
当__exit()__函数返回值为Flase（默认）时，为显示程序异常信息（红色字），
当函数返回值为True时，为不显示程序异常信息。
