Кэширование возвратов без просмотра - PullRequest
Новая
       8

Кэширование возвратов без просмотра

5 голосов
/ 18 января 2010

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

Чек может выглядеть так: 

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)

Ответы [ 2 ]

1 голос
/ 18 января 2010

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

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)
0 голосов
/ 20 января 2010

Сначала вы можете просто написать: 

from django.contrib.auth.decorators import permission_required

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

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

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

...