Код на салфетке

from user_app.decorators import set_template


class UserProfileView(TemplateView):  
    template_name = 'user_app/profile_page.html'  
    author_template_name = 'user_app/profile_page_author.html'  
    is_author = False  
    extended_group = 'Автор'  

    @set_template(author_template_name, 'is_author')  
    def dispatch(self, request, *args, **kwargs):  
        return super().dispatch(request, *args, **kwargs)  

    def get_context_data(self, **kwargs):  
        context = super().get_context_data(**kwargs)  
        try:  
            user = get_object_or_404(User, username=self.kwargs.get('username'))  
        except User.DoesNotExist:  
            raise Http404("Пользователь не найден")  
        context['user_profile'] = user  
        context['is_author'] = self.is_author  
        if self.is_author:  
            context['drafts'] = PostModel.objects.filter(author=user, status='ЧЕ')[:7]  
        context['user_posts'] = PostModel.post_manager.filter(author=user)[:7]  
        context['title'] = f'Профиль пользователя {user}'  
        return context

Декоратор set_template.
В директории приложения user_app создадим файл decorators.py.

В посте "Декораторы в питоне" мы описали простые декораторы и их работу. В данном случае, нужен немного более сложный, который должен принимать аргументы, а также работать с экземпляром класса в котором был вызван.

Посмотрим на код:

from functools import wraps  

from django.contrib.auth.models import User  
from django.shortcuts import get_object_or_404  


def set_template(special_template, role_field):  
    def decorator(view_func):  
        @wraps(view_func)  
        def wrapper(self, request, *args, **kwargs):  
            user = get_object_or_404(User, username=self.kwargs.get('username'))  
            if user  self.request.user and user.groups.filter(name=self.extended_group).exists():  
                self.template_name = special_template  
                setattr(self, role_field, True)  
            return view_func(self, request, *args, **kwargs)  

        return wrapper  

    return decorator

А теперь разберём, что в нём происходит:
1. Создаём функцию set_template, принимающую аргументы - имя шаблона и поле-флаг.
2. Внутри создаём функцию decorator, принимающую задекорированный метод dispatch().
3. Ещё глубже, создаём функцию wrapper, принимающую те же аргументы, что и изначальный метод dispatch(). И сразу оборачиваем её во встроенный в functools декоратор - wraps. Это позволит нам получить доступ к экземпляру класса. Если этот декоратор не указать, то ключевое слово self будет недоступно.
4. В переменную user получаем объект пользователя, на чью страницу был произведён переход.
5. В блоке if проверяем, что текущий пользователь совпадает с пользователем в переменной user, а также, что он входит в группу "Автор". Если проверка проходит, то происходят следующие изменения:
5.1. Значение поля template_name заменяется на переданное в аргументе в самой внешней функции set_template.
5.2. Значение поля-флага, имя которого передано вторым аргументом, устанавливается на True.
6. Вызывается задекорированная функция.

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

Шаблон профиля пользователя.
Откроем файл profile_page.html, расположенный в директории с шаблонами приложения user_app.

🔥2😱1

171 views10:45