Кэширование возвратов, отличных от просмотра

Question

У меня есть около дюжины запросов на просмотр разрешений, которые гарантируют, что у пользователей есть правильные разрешения для выполнения чего-либо в системе (т. Е. убедитесь, что они находятся в нужной группе, могут ли они редактировать свой профиль, являются ли они администраторами группы и т.д.).

Чек может выглядеть следующим образом:

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):
    # ...

Во-вторых, если разрешения не меняются со временем, вы можете свободно сохранять некоторую информацию в сеансе (я нашел это более удобным, чем хранение в кэше любого типа), когда пользователь входит в систему.

Но помните, что если вы измените разрешения, пользователю придется выйти из системы и снова войти в систему, чтобы работать с новыми разрешениями.