缓存非视图返回

Question

我对视图进行了十几个左右的权限查找，以确保用户拥有在系统上执行某些操作的正确权限（即确保他们位于正确的组中，如果他们可以编辑他们的个人资料，如果他们是组管理员， ETC）。

检查可能如下所示：

from django.contrib.auth.decorators import user_passes_test

test_canvote = lambda u: u.has_perm('polls.can_vote')

@user_passes_test(test_canvote)
def my_view(request):
    # ...

这实际上是 Django 教程中的代码（我的有点难看）。有时，检查对于数据库来说是非常密集的，会引发多个查询。随着大量用户访问权限检查页面，事情很快就会变得非常慢。

我的问题是，我可以（在您的帮助下）为 user_passes_test 装饰器构建一个包装器（或替换），该装饰器在缓存中搜索键 'TESTCACHE' + user.pk + 'testname' ，如果它不存在，执行测试并保存其结果。

我以前从未编写过装饰器，但我想它看起来与 user_passes_test 几乎相同，只是将测试作为字符串传递：

@cached_user_passes_test('test_canvote')
def my_view(request):
   # ...

和以往一样，如果我生气了或者如果Django 已经为我做到了这一点（所以我在其他地方遇到了问题）。

编辑：标准装饰器可以在这里找到： http: //code.djangoproject.com/browser/django/trunk/django/contrib/auth/decorators.py

我认为替换 user_passes_test 可能比包装它更容易，所以这是起点。当然，如果您觉得我的说法不正确，请告诉我：

try:
    from functools import update_wrapper, wraps
except ImportError:
    from django.utils.functional import update_wrapper, wraps  # Python 2.3, 2.4 fallback.

from django.contrib.auth import REDIRECT_FIELD_NAME
from django.http import HttpResponseRedirect
from django.utils.http import urlquote
from django.utils.decorators import auto_adapt_to_methods

def user_passes_test(test_func, login_url=None, redirect_field_name=REDIRECT_FIELD_NAME):
    """
    Decorator for views that checks that the user passes the given test,
    redirecting to the log-in page if necessary. The test should be a callable
    that takes the user object and returns True if the user passes.
    """
    if not login_url:
        from django.conf import settings
        login_url = settings.LOGIN_URL

    def decorator(view_func):
        def _wrapped_view(request, *args, **kwargs):
            if test_func(request.user):
                return view_func(request, *args, **kwargs)
            path = urlquote(request.get_full_path())
            tup = login_url, redirect_field_name, path
            return HttpResponseRedirect('%s?%s=%s' % tup)
        return wraps(view_func)(_wrapped_view)
    return auto_adapt_to_methods(decorator)

Answer 1

您可能需要序列化该函数（当我使用它作为缓存的密钥时我没有这样做），但类似这样的事情应该可以工作：

from django.core.cache import cache

def cached_user_passes_test(test_func, login_url=None, redirect_field_name=REDIRECT_FIELD_NAME):
    if not login_url:
        from django.conf import settings
        login_url = settings.LOGIN_URL

    def decorator(view_func):
        def _wrapped_view(request, *args, **kwargs):
            key = str(test_func) + str(request.user)
            cached_test_result = cache.get(key)
            if cached_test_result != None:
                test_result = cached_test_result
            else:
                test_result = test_func(request.user)
                cache.set(key, test_result, 60)       

            if test_result:
                return view_func(request, *args, **kwargs)
            path = urlquote(request.get_full_path())
            tup = login_url, redirect_field_name, path
            return HttpResponseRedirect('%s?%s=%s' % tup)
        return wraps(view_func)(_wrapped_view)
    return auto_adapt_to_methods(decorator)

Answer 2

首先你可以简单地写：

from django.contrib.auth.decorators import permission_required

@permission_required('polls.can_vote')
def my_view(request):
    # ...

其次，如果权限没有及时改变，当用户登录时，你可以自由地在会话中存储一些信息（我发现它比存储在任何类型的缓存中更方便）但

请记住，如果您更改权限，用户必须注销并重新登录才能使用新权限。