python中with语句结合上下文管理器操作详解
前言
所谓上下文管理器即在一个类中重写了__enter__方法和__exit__方法的类就可以成为上下文管理器类。
我们可以通过with语句结合上下文管理器简化一些操作。
使用with语句结合自定义上下文管理器完成数据库相应的操作,代码实现如下:
#1.导入模块
importpymysql
#创建自定义上下文管理器对象
classMyDatabase(object):
#接收参数并创建数据库连接对象
def__init__(self,host,port,user,passwd,database):
self.__db=pymysql.Connection(host,port,user,passwd,database,charset='utf8')
#返回数据库连接对象
def__enter__(self):
returnself.__db
#关闭数据库连接
def__exit__(self,exc_type,exc_val,exc_tb):
self.__db.close()
defmain():
#使用with关键字接收enter返回的对象给db
withMyDatabase('localhost',3306,'root','mysql','JDDB')asdb:
#利用db创建游标
cur=db.cursor()
sql='''select*from%s'''
cur.execute(sql,(goods,))
result=cur.fetchall()
foriinresult:
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__方法关闭数据库连接。
注意:withMyDatabase()asdb--->db=MyDatabase().__enter__()
利用with结合自定义上下文类实现HTTP服务端:
#1.导入socket模块
importsocket
classMySocket(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)
#返回套接字对象
returnself.__sk
#5.编写exit方法关闭套接字对象
def__exit__(self,exc_type,exc_val,exc_tb):
self.__sk.close()
defmain():
#使用with关键字并接受返回的套接字对象给sk
withMySocket(8000)assk:
#等待客户端连接
clicent,ip_port=sk.accept()
recv_data=clicent.recv(1024)
print(recv_data.decode('utf-8'))
#编写HTTP响应报文
http_line='HTTP/1.1GET200OK\r\n'
http_header='ServerPWS/1.0\r\n'
http_body='Welcometoindex!\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__方法,关闭服务器连接
总结
以上就是这篇文章的全部内容了,希望本文的内容对大家的学习或者工作具有一定的参考学习价值,谢谢大家对毛票票的支持。
声明:本文内容来源于网络,版权归原作者所有,内容由互联网用户自发贡献自行上传,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任。如果您发现有涉嫌版权的内容,欢迎发送邮件至:czq8825#qq.com(发邮件时,请将#更换为@)进行举报,并提供相关证据,一经查实,本站将立刻删除涉嫌侵权内容。