溫馨提示×
您好，登錄后才能下訂單哦！
點擊 登錄注冊 即表示同意《億速云用戶服務條款》
您好，登錄后才能下訂單哦！
前言
所謂上下文管理器即在一個類中重寫了__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__方法，關閉服務器連接
總結
以上就是這篇文章的全部內容了，希望本文的內容對大家的學習或者工作具有一定的參考學習價值，謝謝大家對億速云的支持。
免責聲明：本站發布的內容（圖片、視頻和文字）以原創、轉載和分享為主，文章觀點不代表本網站立場，如果涉及侵權請聯系站長郵箱：is@yisu.com進行舉報，并提供相關證據，一經查實，將立刻刪除涉嫌侵權內容。
