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__方法,关闭服务器连接
总结
以上就是这篇文章的全部内容了,希望本文的内容对大家的学习或者工作具有一定的参考学习价值,谢谢大家对我们的支持。