Django 模板过滤器的装饰器？

Question

我有一个模板过滤器，它的功能很简单，而且运行得很好，但我想在它上面加一个装饰器。不幸的是，这个装饰器导致了一个很奇怪的Django错误，让人搞不懂……

下面是能正常工作的代码：

@register.filter(name="has_network")
def has_network(profile, network):
    hasnetworkfunc = getattr(profile, "has_%s" % network)
    return hasnetworkfunc()

加了装饰器（不工作）：

@register.filter(name="has_network")
@cache_function(30)
def has_network(profile, network):
    hasnetworkfunc = getattr(profile, "has_%s" % network)
    return hasnetworkfunc()

这里是错误信息：

TemplateSyntaxError at /

在渲染时捕获到异常： 从空列表中弹出

我试着在装饰器里面设置了断点，我相当确定它根本没有被调用……

不过以防万一，这里是装饰器（我知道会有人问）

我暂时把装饰器换成了一个什么都不做的假装饰器，但我还是得到了同样的错误

def cache_function(cache_timeout):
    def wrapper(fn):
        def decorator(*args, **kwargs):
            return fn(*args, **kwargs)
        return decorator
    return wrapper

编辑 确认: 这个问题是因为装饰器使用了 *args 和 **kwargs 吗？我猜 pop() 是为了确保过滤器至少接收一个参数？

把装饰器改成这样就解决了问题：

def cache_function(cache_timeout):
    def wrapper(fn):
        def decorator(arg1, arg2):
            return fn(arg1, arg2)
        return decorator
    return wrapper

不幸的是，这样就破坏了装饰器的通用性 :/ 现在该怎么办呢？

Answer 1

最终答案：给装饰器添加一个额外的参数，用来指明正在装饰的内容。

可能还有更优雅的方法，但这个方法是有效的。

from django.core.cache import cache
from django.db.models.query import QuerySet
try:
    from cPickle import dumps
except:
    from pickle import dumps
from hashlib import sha1

cache_miss = object()

class CantPickleAQuerySet(Exception): pass

def cache_function(cache_timeout, func_type='generic'):
    def wrapper(fn):
        def decorator(*args, **kwargs):
            try:
                cache_indentifiers = "%s%s%s%s" % (
                                         fn.__module__,
                                         fn.__name__,
                                         dumps(args),
                                         dumps(kwargs)
                                         )
            except Exception, e:
                print "Error: %s\nFailed to generate cache key: %s%s" % (e, fn.__module__, fn.__name__)
                return fn(*args, **kwargs)

            cache_key = sha1(cache_indentifiers).hexdigest()

            value = cache.get(cache_key, cache_miss)

            if value is cache_miss:
                value = fn(*args, **kwargs)

                if isinstance(value, QuerySet):
                    raise CantPickleAQuerySet("You can't cache a queryset. But you CAN cache a list! just convert your Queryset (the value you were returning) to a list like so `return list(queryset)`")

                try:
                    cache.set(cache_key, value, cache_timeout)
                except Exception, e:
                    print "Error: %s\nFailed to cache: %s\nvalue: %s" % (e, cache_indentifiers, value)

            return value

        no_arg2 = object()
        def filter_decorator(arg1, arg2=no_arg2):
            if arg2 is no_arg2:
                return decorator(arg1)
            else:
                return decorator(arg1, arg2)

        if func_type == 'generic':
            return decorator

        elif func_type == 'filter':
            return filter_decorator

    return wrapper