вот мой декоратор:
def check_domain(func):
def wrapper(domain_id, *args, **kwargs):
domain = get_object_or_None(Domain, id=domain_id)
if not domain:
return None
return func(domain_id, *args, **kwargs)
return wrapper
Вот завернутая функция:
@check_domain
def collect_data(domain_id, from_date, to_date):
do_stuff(...)
Если я делаю collect_data.__name__, я получаю wrapper вместо collect_data
Любые идеи?
Вы можете использовать обертку из functools.
Вот пример, который, кажется, подходит вам:)
http://docs.python.org/library/functools.html#functools.wraps
functools.wraps не требуется! Просто используйте method.__name__
import time
def timeit(method):
def timed(*args, **kw):
ts = time.time()
result = method(*args, **kw)
te = time.time()
print('Function', method.__name__, 'time:', round((te -ts)*1000,1), 'ms')
print()
return result
return timed
@timeit
def math_harder():
[x**(x%17)^x%17 for x in range(1,5555)]
math_harder()
@timeit
def sleeper_agent():
time.sleep(1)
sleeper_agent()
Выходы:
Function math_harder time: 8.4 ms
Function sleeper_agent time: 1003.7 ms
Отъезд functools.wraps. Требуется python 2.5 или выше, если это проблема.
В дополнение к functools.wraps вы можете проверить модуль decorator, который был разработан для решения этой проблемы.
Используйте http://docs.python.org/library/functools.html#functools.wraps
Для тех, кому требуется доступ к имени украшенной функции, когда есть несколько декораторов, например:
@decorator1
@decorator2
@decorator3
def decorated_func(stuff):
return stuff
упомянутый выше functools.wraps решает, что.