Доступ к атрибутам на украшенном вызываемом классе
У меня есть вызываемый класс:
class CallMeMaybe:
__name__ = 'maybe'
def __init__(self):
self.n_calls = 0
def __call__(self):
self.n_calls += 1
raise Exception
Это, кажется, работает как рекламируется:
>>> f = CallMeMaybe()
>>> f.n_calls
0
>>> for i in range(7):
... try:
... f()
... except Exception:
... pass
...
>>> f.n_calls
7
Я хочу украсить его экспоненциальным отступлением:
from backoff import on_exception, expo
dec = on_exception(expo, Exception, max_tries=3, on_backoff=print)
f = CallMeMaybe()
f2 = dec(f)
Теперь похоже, что доступ к атрибутам перестал работать:
>>> f2.n_calls
0
>>> f2()
{'target': <__main__.CallMeMaybe object at 0xcafef00d>, 'args': (), 'kwargs': {}, 'tries': 1, 'elapsed': 2.1e-05, 'wait': 0.4843249208229148}
{'target': <__main__.CallMeMaybe object at 0xcafef00d>, 'args': (), 'kwargs': {}, 'tries': 2, 'elapsed': 0.484935, 'wait': 1.6524016553598126}
---------------------------------------------------------------------------
Exception Traceback (most recent call last)
... blah blah blah
>>> f2.n_calls
0
Мой вопрос: кто копировал n_calls
имя в f2
Пространство имен, а почему? Теперь он содержит устаревшее значение - правильное значение должно быть 3:
>>> f2.__wrapped__.n_calls
3
1 ответ
backoff
Модуль в своей реализации используетfunctools.wraps
какие звонки functools.update_wrapper
и вы можете видеть из исходного кода, что по умолчанию он обновляет __dict__
из обертки:
WRAPPER_ASSIGNMENTS = ('__module__', '__name__', '__qualname__', '__doc__',
'__annotations__')
WRAPPER_UPDATES = ('__dict__',)
def update_wrapper(wrapper,
wrapped,
assigned = WRAPPER_ASSIGNMENTS,
updated = WRAPPER_UPDATES):
"""Update a wrapper function to look like the wrapped function
wrapper is the function to be updated
wrapped is the original function
assigned is a tuple naming the attributes assigned directly
from the wrapped function to the wrapper function (defaults to
functools.WRAPPER_ASSIGNMENTS)
updated is a tuple naming the attributes of the wrapper that
are updated with the corresponding attribute from the wrapped
function (defaults to functools.WRAPPER_UPDATES)
"""
for attr in assigned:
try:
value = getattr(wrapped, attr)
except AttributeError:
pass
else:
setattr(wrapper, attr, value)
for attr in updated:
getattr(wrapper, attr).update(getattr(wrapped, attr, {}))
+
−# Issue #17482: set __wrapped__ last so we don't inadvertently copy it
# from the wrapped function when updating __dict__
wrapper.__wrapped__ = wrapped
# Return the wrapper so this can be used as a decorator via partial()
return wrapper
def wraps(wrapped,
assigned = WRAPPER_ASSIGNMENTS,
updated = WRAPPER_UPDATES):
"""Decorator factory to apply update_wrapper() to a wrapper function
Returns a decorator that invokes update_wrapper() with the decorated
function as the wrapper argument and the arguments to wraps() as the
remaining arguments. Default arguments are as for update_wrapper().
This is a convenience function to simplify applying partial() to
update_wrapper().
"""
return partial(update_wrapper, wrapped=wrapped,
assigned=assigned, updated=updated)
К сожалению, кажется невозможным добиться того, что вы хотите. backoff
модуль может позволить опционально assigned
/updated
списки атрибутов для передачи wraps
чтобы избежать копирования атрибута. Однако это действительно решит проблему, потому что в этот момент у вас не будет доступа к n_calls
,
Вы, вероятно, должны использовать изменяемый объект вместо простого int
,