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, self.__db

    def __exit__(self, exc_type, exc_val, exc_tb):
        if exc_type:
            self.__db.rollback()
        else:
            self.__db.commit()
        self.__cur.close()
        self.__db.close()
        return True


def main():
    with Mysql('heima', 'root', 'mysql') as cursor:
        cur, conn = cursor
        sql = '''SQL語句'''
        cur.execute(sql)

if __name__ == '__main__':
    main()

 

注意：__exit(self, exc_type, exc_val, exc_tb)__函數中的三個參數exc_type, exc_val, exc_tb只有在程序異常退出時纔會有值，

只要判斷這三個參數中的其中一個，若是有值那就說明程序存在異常。

當__exit()__函數返回值爲Flase（默認）時，爲顯示程序異常信息（紅色字），

當函數返回值爲True時，爲不顯示程序異常信息。