Я часто обнаруживаю, что мне нужно временно назначить некоторые переменные-члены, например
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 по мере необходимости)
Контекстные менеджеры могут быть легко использованы.
Цитирование официальных документов:
Типичное использование контекстных менеджеров включает сохранение и восстановление различных виды глобального состояния, блокировки и разблокировки ресурсов, закрытие файлы и т.д.
Кажется, что сохранение и восстановление состояния - это именно то, что мы хотим сделать здесь.
Пример:
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
Я думаю, что 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)
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
Вы также можете сделать это изначально с помощью __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__)
Решение Гуфи
>>> 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