Временное изменение значения переменной в Python

Основываясь на ответе @arthaigo, более краткая версия:

import contextlib

@contextlib.contextmanager
def temporary_assignment(object, new_value):
  old_value = eval(object)
  globals()[object] = new_value
  yield
  globals()[object] = old_value

Я знаю, что этот вопрос довольно старый, но когда я столкнулся с той же проблемой, вот мое решение:

class test_context_manager():
    def __init__(self, old_object, new_object):
        self.new = new_object
        self.old = old_object
        self.old_code = eval(old_object)
    def __enter__(self):
        globals()[self.old] = self.new
    def __exit__(self, type, value, traceback):
        globals()[self.old] = self.old_code

Это не красиво, так как сильно использует глобальные переменные, но, похоже, работает.

Например:

x = 5
print(x)
with test_context_manager("x", 7):
    print(x)

print(x)

Результат:

5
7
5

или с функциями:

def func1():
    print("hi")

def func2():
    print("bye")

x = 5
func1()
with test_context_manager("func1", func2):
    func1()

func1()

Результат:

hi
bye
hi

Я только что нашел еще один умный способ сделать это, используя unittest.mock, задокументированный здесь.

import unittest.mock
a = b = 1
with unittest.mock.patch.dict(locals(), a=2, b=3):
  print(a, b)  # shows 2, 3
print(a, b)  # shows 1, 1

with unittest.mock.patch.dict(locals(), c=4):
  assert 'c' in locals()
assert 'c' not in locals()

Обычно я использую этот пользовательский менеджер контекста attr_as:

from contextlib import contextmanager

@contextmanager
def attr_as(obj, field:str, value) -> None:
    old_value = getattr(obj, field)
    setattr(obj, field, value)
    yield
    setattr(obj, field, old_value)

Затем вы можете использовать его с тем же набором аргументов, что и для setattr в attr_as.

class Foo:
    def __init__(self):
        self.x = 1

foo = Foo()
with attr_as(foo, 'x', 2):
    print(foo.x)

bar = 3
with attr_as(sys.modules[__name__], 'bar', 4):
    print(bar) 

Обратите внимание: если вам нужно сохранить существование/отсутствие атрибута, а не только значение, это также возможно, добавив еще несколько строк:

from contextlib import contextmanager

@contextmanager
def attr_as(obj, field:str, value) -> None:
    old_exists = hasattr(obj, field)
    if old_exists:
        old_value = getattr(obj, field)
    setattr(obj, field, value)
    yield
    if old_exists:
        setattr(obj, field, old_value)
    else:
        delattr(obj, field)

Как насчет closure?

i.e.

#!/usr/bin/python

def create_closure(arg):
    var = arg

    def hide_me():
        return var

    return hide_me


x = 10
clo = create_closure(x)

x *= 2
x2 = clo()
x += 1

print "x: {0}".format(x)
print "x2: {0}".format(x2)

Это дает:

x: 21
x2: 10

И поскольку это замыкание, его можно существенно расширить, чтобы сохранить целую кучу других переменных и состояний, а затем либо использовать их исключительно в форме замыкания в дальнейшем, либо использовать в качестве доски, с которой можно восстановить состояние позже, когда вы закончите. Выполнено.