所謂上下文管理器即在一個類中重寫了__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時，為不顯示程序異常信息。