Context manager support for `contextvars.Context`

Feature or enhancement

I would like to add __enter__ and __exit__ methods to contextvars.Context such that these two are mostly equivalent:

contextvars.copy_context().run(do_something_interesting)

and:

with contextvars.copy_context():
    do_something_interesting()

Pitch

This makes it possible to combine Context enter/exit with other context managers:

import contextlib
import contextvars

@contextlib.contextmanager
def foo():
    with contextvars.copy_context():
        set_some_context_vars()
        yield

My personal motivating interest is controlling ContextVar values in pytest fixtures. For example:

import contextvars
import pytest
import module_under_test

@pytest.fixture
def clean_slate():
    with contextvars.copy_context():
        module_under_test._internal_state_context_var.set(None)
        yield

def test_foo(clean_slate):
    assert not module_under_test.is_initialized()

Previous discussion

https://stackoverflow.com/q/59264158

Linked PRs

• gh-124773
• gh-124776
• gh-99634

Bắt đầu bằng cách đọc tài liệu về contextvars.Context và các ví dụ tương đương trong issue, sau đó xem xét các PR được liên kết gh-124773, gh-124776 và gh-99634 trước khi bắt đầu. Công việc được xem là hoàn tất khi Context hỗ trợ cách sử dụng trình quản lý ngữ cảnh được yêu cầu và hành vi khớp với ví dụ run() hiện có, bao gồm cả việc kết hợp với các trình quản lý ngữ cảnh khác.