Python上下文管理器怎么使用

這篇文章主要介紹了Python上下文管理器怎么使用的相關(guān)知識(shí)，內(nèi)容詳細(xì)易懂，操作簡(jiǎn)單快捷，具有一定借鑒價(jià)值，相信大家閱讀完這篇Python上下文管理器怎么使用文章都會(huì)有所收獲，下面我們一起來(lái)看看吧。

什么是上下文管理器？

即使你沒有聽說(shuō)過(guò) Python 的上下文管理器，根據(jù)介紹，你也已經(jīng)知道，它是try/finally塊的替代品。它是使用打開文件時(shí)常用的語(yǔ)句with來(lái)實(shí)現(xiàn)的。與try/finally相同，引入此模式是為了保證在塊末尾執(zhí)行某些操作，即使發(fā)生異?；虺绦蚪K止。

從表面上看，上下文管理協(xié)議只是圍繞with代碼塊的語(yǔ)句。實(shí)際上，它包含 2 個(gè)特殊的 ( dunder ) 方法 -__enter__和__exit__組成，分別有助于啟動(dòng)和停止。

當(dāng)代碼中遇到with語(yǔ)句時(shí)，將觸發(fā)__enter__方法并將其返回值放入as限定符后面的變量中。with塊體執(zhí)行完畢后，調(diào)用__exit__方法進(jìn)行停止——完成finally塊的作用。

# Using try/finally
import time

start = time.perf_counter()  # Setup
try:  # Actual body
    time.sleep(3)
finally:  # Teardown
    end = time.perf_counter()
    elapsed = end - start

print(elapsed)

# Using Context Manager
with Timer() as t:
    time.sleep(3)

print(t.elapsed)

上面的代碼顯示了使用try/finally的版本和使用with語(yǔ)句來(lái)實(shí)現(xiàn)簡(jiǎn)單的計(jì)時(shí)器的更優(yōu)雅的版本。如上所述，實(shí)現(xiàn)這樣的上下文管理器需要__enter__和__exit__，但是我們將如何創(chuàng)建它們呢？我們看一下這個(gè)Timer類的代碼：

# Implementation of above context manager
class Timer:
    def __init__(self):
        self._start = None
        self.elapsed = 0.0

    def start(self):
        if self._start is not None:
            raise RuntimeError('Timer already started...')
        self._start = time.perf_counter()

    def stop(self):
        if self._start is None:
            raise RuntimeError('Timer not yet started...')
        end = time.perf_counter()
        self.elapsed += end - self._start
        self._start = None

    def __enter__(self):  # Setup
        self.start()
        return self

    def __exit__(self, *args):  # Teardown
        self.stop()

此代碼片段顯示了實(shí)現(xiàn)__enter__和__exit__方法的Timer類。__enter__方法僅啟動(dòng)計(jì)時(shí)器并返回self，self將在with ....中作為some_var賦值， with語(yǔ)句體完成后，將使用 3 個(gè)參數(shù)調(diào)用__exit__方法 - 異常類型、異常值和回溯。如果with語(yǔ)句正文中一切順利，則這些都等于None。如果引發(fā)異常，這些將填充異常數(shù)據(jù)，我們可以在__exit__方法中處理這些數(shù)據(jù)。在這種情況下，我們省略了異常處理，只是停止計(jì)時(shí)器并計(jì)算經(jīng)過(guò)的時(shí)間，并將其存儲(chǔ)在上下文管理器的屬性中。

我們已經(jīng)在這里看到了with語(yǔ)句的實(shí)現(xiàn)和示例用法，但是為了更直觀地了解實(shí)際發(fā)生的情況，讓我們看看如何在沒有 Python 語(yǔ)法糖的情況下調(diào)用這些特殊方法：

manager = Timer()
manager.__enter__()  # Setup
time.sleep(3)  # Body
manager.__exit__(None, None, None)  # Teardown
print(manager.elapsed)

現(xiàn)在我們已經(jīng)確定了什么是上下文管理器，它是如何工作的以及如何實(shí)現(xiàn)它，讓我們看看使用它的好處——只是為了有更多的動(dòng)力從try/finally切換到with語(yǔ)句。

第一個(gè)好處是整個(gè)啟動(dòng)和停止都在上下文管理器對(duì)象的控制下進(jìn)行。這可以防止錯(cuò)誤并減少樣板代碼，從而使 API 更安全、更易于使用。使用它的另一個(gè)原因是with塊突出了關(guān)鍵部分并鼓勵(lì)你減少該部分中的代碼量，這通常也是一個(gè)好習(xí)慣。最后——最后但并非最不重要的一點(diǎn)——它是一個(gè)很好的重構(gòu)工具，它可以將常見的啟動(dòng)和停止代碼分解出來(lái)，并將其移動(dòng)到一個(gè)位置——即__enter__和__exit__方法。

話雖如此，我希望我能說(shuō)服你開始使用上下文管理器，而不是try/finally，即使你以前沒有使用過(guò)它們。那么，現(xiàn)在讓我們看看一些很酷且有用的上下文管理器，你應(yīng)該開始將它們包含在你的代碼中！

@contextmanager

在上一節(jié)中，我們探討了如何使用__enter__和__exit__方法實(shí)現(xiàn)上下文管理器。這很簡(jiǎn)單，但我們可以使用contextlib，更具體地說(shuō)，使用@contextmanager，使其更簡(jiǎn)單。

@contextmanager是一個(gè)裝飾器，可用于編寫自包含的上下文管理函數(shù)。因此，我們不需要?jiǎng)?chuàng)建整個(gè)類并實(shí)現(xiàn)__enter__和__exit__方法，我們只需要?jiǎng)?chuàng)建一個(gè)生成器：

from contextlib import contextmanager
from time import time, sleep

@contextmanager
def timed(label):
    start = time()  # Setup - __enter__
    print(f"{label}: Start at {start}")
    try:  
        yield  # yield to body of `with` statement
    finally:  # Teardown - __exit__
        end = time()
        print(f"{label}: End at {end} ({end - start} elapsed)")

with timed("Counter"):
    sleep(3)

# Counter: Start at 1599153092.4826472
# Counter: End at 1599153095.4854734 (3.00282621383667 elapsed)

此代碼段實(shí)現(xiàn)了與上一節(jié)中的Timer類非常相似的上下文管理器。然而，這一次，我們需要的代碼要少得多。這段代碼分為兩個(gè)部分，一部分是在yield之前，另一部分是yield之后。yield之前的代碼承擔(dān)了__enter__方法的工作，而yield本身是__enter__方法的return語(yǔ)句。yield之后的都是__exit__方法的一部分。

正如你在上面看到的，像這樣使用單個(gè)函數(shù)創(chuàng)建上下文管理器需要使用使用try/finally語(yǔ)句，因?yàn)槿绻谡Z(yǔ)句withy體中發(fā)生異常，它將在yield行被引發(fā)，我們需要在對(duì)應(yīng)于__exit__方法的finally塊中處理它。

正如我已經(jīng)提到的，這可以用于自包含的上下文管理器。但是，它不適合需要成為對(duì)象一部分的上下文管理器，例如連接或鎖。

盡管使用單個(gè)函數(shù)構(gòu)建上下文管理器會(huì)迫使你使用try/finally，并且只能用于更簡(jiǎn)單的用例，但在我看來(lái)，它仍然是構(gòu)建更精簡(jiǎn)的上下文管理器的優(yōu)雅而實(shí)用的選擇。

現(xiàn)實(shí)生活中的例子

現(xiàn)在讓我們從理論轉(zhuǎn)向?qū)嵱们矣杏玫纳舷挛墓芾砥?，你可以自己?gòu)建它。

記錄上下文管理器

當(dāng)需要嘗試查找代碼中的一些bug時(shí)，你可能會(huì)首先查看日志以找到問(wèn)題的根本原因。但是，這些日志可能默認(rèn)設(shè)置為錯(cuò)誤或警告級(jí)別，這可能不足以用于調(diào)試。更改整個(gè)程序的日志級(jí)別應(yīng)該很容易，但更改特定代碼部分的日志級(jí)別可能會(huì)更復(fù)雜 - 不過(guò)，這可以通過(guò)以下上下文管理器輕松解決：

import logging
from contextlib import contextmanager

@contextmanager
def log(level):
    logger = logging.getLogger()
    current_level = logger.getEffectiveLevel()
    logger.setLevel(level)
    try:
        yield
    finally:
        logger.setLevel(current_level)

def some_function():
    logging.debug("Some debug level information...")
    logging.error('Serious error...')
    logging.warning('Some warning message...')

with log(logging.DEBUG):
    some_function()

# DEBUG:root:Some debug level information...
# ERROR:root:Serious error...
# WARNING:root:Some warning message...

超時(shí)上下文管理器

在本文的開頭，我們正在使用計(jì)時(shí)代碼塊。我們?cè)谶@里嘗試的是將超時(shí)設(shè)置為with語(yǔ)句包圍的塊：

import signal
from time import sleep

class timeout:
    def __init__(self, seconds, *, timeout_message=""):
        self.seconds = int(seconds)
        self.timeout_message = timeout_message

    def _timeout_handler(self, signum, frame):
        raise TimeoutError(self.timeout_message)

    def __enter__(self):
        signal.signal(signal.SIGALRM, self._timeout_handler)  # Set handler for SIGALRM
        signal.alarm(self.seconds)  # start countdown for SIGALRM to be raised

    def __exit__(self, exc_type, exc_val, exc_tb):
        signal.alarm(0)  # Cancel SIGALRM if it's scheduled
        return exc_type is TimeoutError  # Suppress TimeoutError


with timeout(3):
    # Some long running task...
    sleep(10)

上面的代碼為這個(gè)上下文管理器聲明了一個(gè)名為timeout的類，因?yàn)檫@個(gè)任務(wù)不能在單個(gè)函數(shù)中完成。為了能夠?qū)崿F(xiàn)這種超時(shí)，我們還需要使用信號(hào)-更具體地說(shuō)是SIGALRM。我們首先使用signal.signal(...)將處理程序設(shè)置為SIGALRM，這意味著當(dāng)內(nèi)核引發(fā)SIGALRM時(shí)，將調(diào)用處理程序函數(shù)。對(duì)于這個(gè)處理程序函數(shù)（_timeout_handler），它所做的只是引發(fā)TimeoutError，如果沒有及時(shí)完成，它將停止with語(yǔ)句體中的執(zhí)行。處理程序就位后，我們還需要以指定的秒數(shù)開始倒計(jì)時(shí)，這由signal.alarm(self.seconds)完成。

對(duì)于__exit__方法，如果上下文管理器的主體設(shè)法在時(shí)間到期之前完成，SIGALRM則將被取消，而signal.alarm(0)和程序可以繼續(xù)。另一方面 - 如果由于超時(shí)而引發(fā)信號(hào)，那么_timeout_handler將引發(fā)TimeoutError，這將__exit__被捕獲和抑制，with語(yǔ)句主體將被中斷，其余代碼可以繼續(xù)執(zhí)行。

使用已有的

除了上面的上下文管理器，標(biāo)準(zhǔn)庫(kù)或其他常用庫(kù)（如request或sqlite3）中已經(jīng)有很多有用的上下文管理程序。那么，讓我們看看我們可以在那里找到什么。

臨時(shí)更改小數(shù)精度

如果你正在執(zhí)行大量數(shù)學(xué)運(yùn)算并需要特定的精度，那么你可能會(huì)遇到需要臨時(shí)更改十進(jìn)制數(shù)精度的情況：

from decimal import getcontext, Decimal, setcontext, localcontext, Context

# Bad
old_context = getcontext().copy()
getcontext().prec = 40
print(Decimal(22) / Decimal(7))
setcontext(old_context)

# Good
with localcontext(Context(prec=50)):
    print(Decimal(22) / Decimal(7))  # 3.1428571428571428571428571428571428571428571428571

print(Decimal(22) / Decimal(7))      # 3.142857142857142857142857143

上面的代碼演示了不帶和帶上下文管理器的選項(xiàng)。第二個(gè)選項(xiàng)顯然更短，更具可讀性。它還考慮了臨時(shí)上下文，使其不易出錯(cuò)。

contextlib

在使用@contextmanager時(shí)，我們已經(jīng)窺探了contextlib，但我們可以使用更多的東西——作為第一個(gè)示例，讓我們看看redirect_stdout和redirect redirect_stderr：

import sys
from contextlib import redirect_stdout

# Bad
with open("help.txt", "w") as file:
    stdout = sys.stdout
    sys.stdout = file
    try:
        help(int)
    finally:
        sys.stdout = stdout

# Good
with open("help.txt", "w") as file:
    with redirect_stdout(file):
        help(int)

如果你有一個(gè)工具或函數(shù)，默認(rèn)情況下將所有數(shù)據(jù)輸出到stdout或stderr，但你希望它將數(shù)據(jù)輸出到其他地方——例如文件。那么這兩個(gè)上下文管理器可能非常有用。與前面的示例一樣，這大大提高了代碼的可讀性，并消除了不必要的視覺干擾。

contextlib的另一個(gè)方便的方法是suppress上下文管理器，它將抑制任何不需要的異常和錯(cuò)誤：

import os
from contextlib import suppress

try:
    os.remove('file.txt')
except FileNotFoundError:
    pass


with suppress(FileNotFoundError):
    os.remove('file.txt')

當(dāng)然，正確處理異常是更好的，但有時(shí)你只需要消除令人討厭的DeprecationWarning警告，這個(gè)上下文管理器至少會(huì)使它可讀。

我將提到的contextlib中的最后一個(gè)實(shí)際上是我最喜歡的，它叫做closing：

# Bad
try:
    page = urlopen(url)
    ...
finally:
    page.close()

# Good
from contextlib import closing

with closing(urlopen(url)) as page:
    ...

此上下文管理器將關(guān)閉作為參數(shù)傳遞給它的任何資源（在上面的示例中），即page對(duì)象。至于在后臺(tái)實(shí)際發(fā)生的情況，上下文管理器實(shí)際上只是強(qiáng)制調(diào)用頁(yè)面對(duì)象的.close()方法，與使用try/finally選項(xiàng)的方式相同。

用于更好測(cè)試的上下文管理器

若你們想讓人們使用、閱讀或維護(hù)你們所寫的測(cè)試，你們必須讓他們可讀，易于理解和模仿。mock.patch上下文管理器可以幫助你：

# Bad
import requests
from unittest import mock
from unittest.mock import Mock

r = Mock()
p = mock.patch('requests.get', return_value=r)
mock_func = p.start()
requests.get(...)
# ... do some asserts
p.stop()

# Good
r = Mock()
with mock.patch('requests.get', return_value=r):
    requests.get(...)
    # ... do some asserts

使用mock.patch上下文管理器可以讓你擺脫不必要的.start()和.stop()調(diào)用，并幫助你定義此特定模擬的明確范圍。這個(gè)測(cè)試的好處是它可以與unittest以及pytest一起使用，即使它是標(biāo)準(zhǔn)庫(kù)的一部分（因此也是unittest）。

說(shuō)到pytest，讓我們也展示一下這個(gè)庫(kù)中至少一個(gè)非常有用的上下文管理器：

import pytest, os

with pytest.raises(FileNotFoundError, message="Expecting FileNotFoundError"):
    os.remove('file.txt')

這個(gè)例子展示了pytest.raises的非常簡(jiǎn)單的用法，它斷言代碼塊引發(fā)提供的異常。如果沒有，則測(cè)試失敗。這對(duì)于測(cè)試預(yù)期會(huì)引發(fā)異常或失敗的代碼路徑非常方便。

跨請(qǐng)求持久化會(huì)話

從pytest轉(zhuǎn)到另一個(gè)偉大的庫(kù)——requests。通常，你可能需要在HTTP請(qǐng)求之間保留cookie，需要保持TCP連接活動(dòng)，或者只想對(duì)同一主機(jī)執(zhí)行多個(gè)請(qǐng)求。requests提供了一個(gè)很好的上下文管理器來(lái)幫助應(yīng)對(duì)這些挑戰(zhàn)，即管理會(huì)話：

import requests

with requests.Session() as session:
    session.request(method=method, url=url, **kwargs)

除了解決上述問(wèn)題之外，這個(gè)上下文管理器還可以幫助提高性能，因?yàn)樗鼘⒅赜玫讓舆B接，因此避免為每個(gè)請(qǐng)求/響應(yīng)對(duì)打開新連接。

管理 SQLite 事務(wù)

最后但同樣重要的是，還有用于管理SQLite事務(wù)的上下文管理器。除了使代碼更干凈之外，此上下文管理器還提供了在異常情況下回滾更改的能力，以及在with語(yǔ)句體成功完成時(shí)自動(dòng)提交的能力：

import sqlite3
from contextlib import closing

# Bad
connection = sqlite3.connect(":memory:")
try:
    connection.execute("INSERT INTO employee(firstname, lastname) values (?, ?)", ("John", "Smith",))
except sqlite3.IntegrityError:
    ...

connection.close()

# Good
with closing(sqlite3.connect(":memory:")) as connection:
    with connection:
        connection.execute("INSERT INTO employee(firstname, lastname) values (?, ?)", ("John", "Smith",))

在本例中，你還可以看到closing上下文管理器的良好使用，它有助于處理不再使用的連接對(duì)象，這進(jìn)一步簡(jiǎn)化了代碼，并確保我們不會(huì)讓任何連接掛起。

關(guān)于“Python上下文管理器怎么使用”這篇文章的內(nèi)容就介紹到這里，感謝各位的閱讀！相信大家對(duì)“Python上下文管理器怎么使用”知識(shí)都有一定的了解，大家如果還想學(xué)習(xí)更多知識(shí)，歡迎關(guān)注億速云行業(yè)資訊頻道。