Use functools.wraps on memoize decorator

This commit is contained in:
Kim Silkebækken 2013-02-08 17:13:37 +01:00
parent d386ae71fd
commit 6dc81ab855
1 changed files with 2 additions and 0 deletions

View File

@ -1,5 +1,6 @@
# -*- coding: utf-8 -*-
from functools import wraps
import time
@ -12,6 +13,7 @@ class memoize(object):
self.additional_key = additional_key
def __call__(self, func):
@wraps(func)
def decorated_function(*args, **kwargs):
if self.additional_key:
key = (func.__name__, args, tuple(kwargs.items()), self.additional_key())