Контекстный менеджер Python для временного назначения переменных

Нет, потому что диспетчер контекста не может так назначать переменные в области действия вызывающего объекта. (Любой, кто думает, что вы можете сделать это с помощью locals или inspect, попробуйте использовать диспетчер контекста, который вы придумали внутри функции. Это не сработает.)

Существуют существуют утилиты для работы с вещами, не являющимися локальными переменными, такими как глобальные переменные модулей, другие атрибуты объектов и словари... но они unittest.mock.patch и связанные с ним функции, поэтому вам следует тщательно рассмотреть другие альтернативы, прежде чем использовать их в не- контекст тестирования. Такие операции, как «временно изменить эту вещь, а затем восстановить ее», как правило, приводят к запутанному коду и могут указывать на то, что вы используете слишком много глобального состояния.

Простой ответ на ваш вопрос:

Содержит ли стандартная библиотека Python такой менеджер контекста?

это «Нет, это не так».

Моя ошибка, вместо этого, возможно, что-то вроде этого, это не встроено:

class ContextTester(object):
    """Initialize context environment and replace variables when completed"""

    def __init__(self, locals_reference):
        self.prev_local_variables = locals_reference.copy()
        self.locals_reference = locals_reference

    def __enter__(self):
        pass

    def __exit__(self, exception_type, exception_value, traceback):
        self.locals_reference.update(self.prev_local_variables)



a = 5
def do_some_work():
    global a
    print(a)
    a = 8
print("Before context tester: {}".format(a))
with ContextTester(locals()) as context:
    print("In context tester before assignment: {}".format(a))
    a = 6
    do_some_work()
    print("In context tester after assignment: {}".format(a))
print("After context tester: {}".format(a))

Выход:

Before context tester: 5
In context tester before assignment: 5
6
In context tester after assignment: 8
After context tester: 5

Для ясности, чтобы вы знали, что он действительно что-то делает:

class ContextTester(object):
    """Initialize context environment and replace variables when completed"""

    def __init__(self, locals_reference):
        self.prev_local_variables = locals_reference.copy()
        self.locals_reference = locals_reference

    def __enter__(self):
        pass

    def __exit__(self, exception_type, exception_value, traceback):
        #self.locals_reference.update(self.prev_local_variables)
        pass

a = 5
def do_some_work():
    global a
    print(a)
    a = 8
print("Before context tester: {}".format(a))
with ContextTester(locals()) as context:
    print("In context tester before assignment: {}".format(a))
    a = 6
    do_some_work()
    print("In context tester after assignment: {}".format(a))
print("After context tester: {}".format(a))
a = 5
print("Before context tester: {}".format(a))
with ContextTester(locals()) as context:
    print("In context tester before assignment: {}".format(a))
    a = 6
    print("In context tester after assignment: {}".format(a))
print("After context tester: {}".format(a))

Выход:

Before context tester: 5
In context tester before assignment: 5
6
In context tester after assignment: 8
After context tester: 8

Before context tester: 5
In context tester before assignment: 5
In context tester after assignment: 6
After context tester: 6

Вы также можете сделать это:

def wrapper_function(func, *args, **kwargs):
    prev_globals = globals().copy()
    func(*args, **kwargs)
    globals().update(prev_globals)

Следует отметить, что если вы попытаетесь использовать оператор with внутри функции, вы захотите использовать globals() в качестве ссылки на локальные, и это может иметь непредвиденные последствия, в любом случае все еще может.

Я бы не рекомендовал делать это вообще, но должно работать.