Trình quản lý bối cảnh Python tự trả về

Trong Python, trình quản lý bối cảnh là một đối tượng có thể được sử dụng trong câu lệnh

with Timer("doing stuff"):
    for i in range(1000000):
        pass

with Timer("doing stuff"):
    for i in range(1000000):
        pass

import time

class Timer(object):
    def __init__(self, msg):
        self._msg = msg

    def __enter__(self):
        # time.monotonic() requires Python >= 3.3
        self._start = time.monotonic()

    def __exit__(self, exc_type, exc_value, exc_traceback):
        if exc_type:
            print('Failed: {}: {}'.format(self._msg, exc_value))
        else:
            print('{}: {} s'.format(self._msg, time.monotonic() - self._start))

Chúng ta có thể sử dụng nó như thế này

with Timer("doing stuff"):
    for i in range(1000000):
        pass

cái nào tạo ra đầu ra này

doing stuff: 0.04281306266784668 s

Xử lý ngoại lệ

Trình quản lý ngữ cảnh cũng có thể xử lý các ngoại lệ từ bên trong khối mà chúng bảo vệ.

with Timer("doing stuff"):
    for i in range(1000000):
        pass

with Timer("doing stuff"):
    raise ValueError('ack')

chúng tôi nhận được

Failed: doing stuff: ack
Traceback (most recent call last):
  File "test.py", line 20, in <module>
    raise ValueError('ack')
ValueError: ack

Bởi vì

with Timer("doing stuff"):
    for i in range(1000000):
        pass

with Timer("doing stuff"):
    for i in range(1000000):
        pass

with Timer("doing stuff"):
    for i in range(1000000):
        pass

with Timer("doing stuff"):
    for i in range(1000000):
        pass

with Timer("doing stuff"):
    for i in range(1000000):
        pass

with Timer("doing stuff"):
    for i in range(1000000):
        pass

Bối cảnh từ trình quản lý bối cảnh

Giả sử chúng ta muốn trình quản lý bối cảnh trả về một số thông tin, có lẽ là mức độ chi tiết của bộ đếm thời gian. Chúng ta có thể sửa đổi

doing stuff: 0.04281306266784668 s

def __enter__(self):
    # time functions require Python >= 3.3
    self._start = time.monotonic()
    return time.clock_getres(time.CLOCK_MONOTONIC)

Bây giờ chúng ta có thể làm

with Timer("doing stuff") as resolution:
    print('Resolution: {}'.format(resolution))
    for i in range(1000000):
        pass

sản xuất

Resolution: 1e-09
doing stuff: 0.043778783998277504 s

Người trang trí doing stuff: 0.04281306266784668 s 2

Viết trình quản lý bối cảnh liên quan đến một số bản soạn sẵn. các phương thức

doing stuff: 0.04281306266784668 s

with Timer("doing stuff"):
    for i in range(1000000):
        pass

doing stuff: 0.04281306266784668 s

doing stuff: 0.04281306266784668 s

with Timer("doing stuff"):
    for i in range(1000000):
        pass

import contextlib
import time

@contextlib.contextmanager
def Timer(msg):
    start = time.monotonic()
    try:
        yield time.clock_getres(time.CLOCK_MONOTONIC)
    except BaseException as e:
        print('Failed: {}: {}'.format(msg, e))
        raise
    else:
        print('{}: {} s'.format(msg, time.monotonic() - start))

Bây giờ

with Timer("doing stuff"):
    for i in range(1000000):
        pass

with Timer("doing stuff"):
    for i in range(1000000):
        pass

with Timer("doing stuff"):
    raise ValueError('ack')

tự quản lý

Một cách sử dụng cho trình quản lý bối cảnh là đảm bảo rằng một đối tượng (tệp, kết nối mạng, xử lý cơ sở dữ liệu) được đóng khi rời khỏi một khối.

with Timer("doing stuff"):
    raise ValueError('ack')

with Timer("doing stuff"):
    for i in range(1000000):
        pass

with Timer("doing stuff"):
    raise ValueError('ack')

import contextlib

with contextlib.closing(open("/etc/passwd")) as fh:
    lines = fh.readlines()

Nhưng điều đó có vẻ hơi dài dòng. Một cách tiếp cận tốt hơn. nhiều đối tượng triển khai

doing stuff: 0.04281306266784668 s

with Timer("doing stuff"):
    for i in range(1000000):
        pass

with Timer("doing stuff"):
    raise ValueError('ack')

with Timer("doing stuff"):
    raise ValueError('ack')

Chức năng __ nhập __ trong Python là gì?

Điều gì nên __ thoát __ trở lại trong Python?

Những hành động nào được đảm bảo bởi trình quản lý bối cảnh Python?

Sự khác biệt giữa trình trang trí và trình quản lý ngữ cảnh trong Python là gì?