Как временно назначить переменные-члены?

Я часто обнаруживаю, что мне нужно временно назначить некоторые переменные-члены, например

old_x = c.x
old_y = c.y
# keep c.z unchanged

c.x = new_x
c.y = new_y

do_something(c)

c.x = old_x
c.y = old_y

но мне хотелось бы просто написать

with c.x = new_x; c.y = new_y:
    do_something(c)

или даже

do_something(c with x = new_x; y = new_y)

Могут ли декодеры Python или другие языковые функции включить такой шаблон? (Я мог бы изменить класс c по мере необходимости)

4b9b3361

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

Цитирование официальных документов:

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

Кажется, что сохранение и восстановление состояния - это именно то, что мы хотим сделать здесь.

Пример:

from contextlib import contextmanager


@contextmanager
def temporary_change_attributes(something, **kwargs):
    previous_values = {k: getattr(something, k) for k in kwargs}
    for k, v in kwargs.items():
        setattr(something, k, v)
    try:
        yield
    finally:
        for k, v in previous_values.items():
            setattr(something, k, v)


class Something(object):
    def __init__(self, x, y):
        self.x = x
        self.y = y

    def say_hello(self):
        print("hello", self.x, self.y)


s = Something(1, 2)
s.say_hello()  # hello 1 2
with temporary_change_attributes(s, x=4, y=5):
    s.say_hello()  # hello 4 5
s.say_hello()  # hello 1 2
20
22 июля '16 в 20:19
источник

Я думаю, что contextmanager должен делать то, что вы хотите:

from contextlib import contextmanager

@contextmanager
def current_instance(c, temp_x, temp_y):
    old_x, old_y = c.x, c.y
    c.x, c.y = temp_x, temp_y
    yield c
    c.x, c.y = old_x, old_y

with current_instance(c, x, y) as c_temp:
    do_something(c_temp) 
5
22 июля '16 в 20:19
источник

mock предоставляет эту функциональность, в частности, посмотрите на использование контекстного менеджера patch.object. Он находится в основных библиотеках в python3 и доступен на pypi для более старого python.

Настройка:

>>> class C:
...     def __init__(self, x, y, z):
...         self.x = x
...         self.y = y
...         self.z = z
...         
>>> c = C(0,1,2)

Демонстрация использования:

>>> print(c.x, c.y, c.z)
0 1 2
>>> with patch.object(c, 'x', 'spam'), patch.object(c, 'y', 'eggs'):
...     print(c.x, c.y, c.z)
...     
spam eggs 2
>>> print(c.x, c.y, c.z)
0 1 2
3
22 июля '16 в 20:14
источник

Вы также можете сделать это изначально с помощью __enter__ и __exit__. Упрощенный пример:

class SomeObject(object):
    def __init__(self):
        self.a = 1
        self.b = 2
        self.c = 3

class Temporary(object):
    def __init__(self, target, **kv):
        self.target = target
        self.to_set = kv
        self.to_restore = {}

    def __enter__(self):
        self.to_restore = map(partial(getattr, self.target), filter(partial(hasattr, self.target), self.to_set.keys()))
        for k,v in self.to_set.items():
            if hasattr(self.target, k):
                self.to_restore[k] = getattr(self.target, k)
            setattr(self.target, k, v)

    def __exit__(self, *_):
        for k,v in self.to_restore.items():
            setattr(self.target, k, v)
        for k in self.to_set.keys():
            if k not in self.to_restore:
                delattr(self.target, k)

o = SomeObject()

print(o.__dict__)
with Temporary(o, a=42, d=1337):
    print(o.__dict__)
print(o.__dict__)
3
22 июля '16 в 20:25
источник

Решение Гуфи

>>> class Foo(object):
        def __init__(self):
             self._x = []
             self._y = []


        @property
        def x(self):
           return self._x[-1] or None

        @x.setter 
        def x(self, val):
           self._x.append(val)

        def reset_vals(self):
           if len(self._x) > 1:
              self._x.pop()


>>> bar = Foo()
>>> bar.x = 1
>>> bar.x
1
>>> bar.x = 2
>>> bar.x
2
>>> bar.reset_vals()
>>> bar.x
1
>>> bar.reset_vals()
>>> bar.x
1

Все еще тупой, но тем менее решение

>>> class Foo(object):
    def __init__(self):
        pass


>>> import copy
>>> bar = Foo()
>>> bar.x = 1
>>> bar.x
1
>>> bar2 = copy.copy(bar)
>>> bar2.x
1
>>> bar2.x = 5
>>> bar2.x
5
>>> bar
<__main__.Foo object at 0x0426A870>
>>> bar.x
1
2
22 июля '16 в 20:25
источник