引言

使用上下文管理器，可以讓代碼更加優(yōu)雅簡(jiǎn)潔。當(dāng)然，上下文的管理器的作用不止于此，它內(nèi)部的實(shí)現(xiàn)機(jī)制，能很好的處理代碼異常，提升代碼的復(fù)用性

1、先看看最簡(jiǎn)單的例子，with語(yǔ)句

# 創(chuàng)建一個(gè)文件寫入字符串“Python”
f = open('123.txt', 'w')
f.write("python")
f.close()
# 使用with語(yǔ)句調(diào)用上下文實(shí)現(xiàn)文件寫入操作
with open('123.txt', 'w') as f:
    f.write('python')

2、什么是上下文管理器？

上下文是 context 直譯的叫法，在程序中用來表示代碼執(zhí)行過程中所處的前后環(huán)境，比如在文件操作時(shí)，文件需要打開關(guān)閉，而文件讀寫操作就處于文件操作的上下文環(huán)境中；

上下文管理器，上下文管理器是指在一段代碼執(zhí)行之前，執(zhí)行一些預(yù)處理的工作，代碼執(zhí)行之后再執(zhí)行一些清理工作。

上下文管理器中有enter()和 exit() 兩個(gè)方法，enter()方法在執(zhí)行 with 后面的語(yǔ)句時(shí)執(zhí)行，一般用來處理操作前的內(nèi)容,比如一些創(chuàng)建對(duì)象，初始化等； exit() 方法在 with 內(nèi)的代碼執(zhí)行完畢后執(zhí)行，一般用來處理一些善后收尾工作，比如文件的關(guān)閉，數(shù)據(jù)庫(kù)的關(guān)閉等。

3、上下文管理器的原理過程如下：

• 調(diào)用enter()方法，進(jìn)行預(yù)處理操作
• 執(zhí)行用戶操作
• 調(diào)用 exit() 方法，完成清理操作

4、上下文管理器的應(yīng)用場(chǎng)景：

資源管理功能，即文件處理、網(wǎng)絡(luò)連接、數(shù)據(jù)庫(kù)連接等操作時(shí)需要關(guān)閉資源。

也可以在代碼執(zhí)行前后增加功能，類似于裝飾器，比如代碼之前做權(quán)限驗(yàn)證等。

4.1 數(shù)據(jù)庫(kù)連接

import pymysql
class DBConnection(object):
    def __init__(self,ip,user,passwd,db):
        self.ip = ip
        self.user = user
        self.passwd = passwd
        self.db = db
    def __enter__(self):
        self.conn = pymysql.connect(self.ip, user=self.user, passwd=self.passwd, db=self.db)
        self.cur = conn.cursor()
        return self.cur
    def __exit__(self, exc_type, exc_val, exc_tb):
        self.cur.close()
        self.conn.close()
with DBConnection('192.168.121.xxx', user="xxx", passwd="123456", db="xxx") as cur:
    cur.execute("select * from studnet;")
    result = cur.fetchall()
    print(result)

完成DBConnection這個(gè)類，每次連接數(shù)據(jù)庫(kù)時(shí)，只要簡(jiǎn)單的調(diào)用with語(yǔ)句即可，不需要關(guān)心數(shù)據(jù)庫(kù)的關(guān)閉、異常等

4.2上下文管理器的異常處理

class MyOpen(object):
    """自定義上下文管理類"""
    def __init__(self, file, mode):
        self._file = file
        self._mode = mode
    def __enter__(self):
        self._handle = open(self._file, self._mode)
        return self._handle
    def __exit__(self, exc_type, exc_val, exc_tb):
        # print('Type: ', exc_type)
        # print('Value:', exc_val)
        # print('TreacBack:', exc_tb)
        self._handle.close()
        print("異常已被處理")
        return True
# 讀的模式打開文件,進(jìn)行寫操作，不支持
with MyOpen('123.txt', 'r') as f:
    f.write('python')
#輸出：
異常已被處理

with 語(yǔ)法不僅可以簡(jiǎn)化資源操作的后續(xù)清除操作，還可以代替 try/finally 進(jìn)行異常處理

當(dāng)with中執(zhí)行的語(yǔ)句發(fā)生異常時(shí)，異常信息會(huì)被發(fā)送到 exit()方法的參數(shù)中， exit() 方法有如下三個(gè)參數(shù):

• exc_type : 異常類型
• exc_val : 異常值
• exc_tb : 異常回溯追蹤

這三個(gè)參數(shù)都與異常有關(guān)，with語(yǔ)句會(huì)把異常的exc_type ,exc_val 和exc_tb傳遞給 exit() 方法，它讓exit() 方法來處理異常 ，如果exit()返回的是True，那么這個(gè)異常就被忽略，并按照我們定義的方式進(jìn)行拋出。如果exit()返回的是True以外的任何東西，那么這個(gè)異常將被with語(yǔ)句拋出。