access-controller/main/views.py

from django.contrib.auth.forms import PasswordResetForm
from django.contrib.auth.tokens import default_token_generator
from django.contrib.contenttypes.models import ContentType
from django.shortcuts import render, get_list_or_404
from django.urls import reverse_lazy
from django.views.generic import FormView

from access_controller.settings import EMAIL_HOST_USER
from main.extra_func import check_user_exist, update_profile, get_user_organization, \
    make_engineer, make_light_agent, get_users_list

from django.contrib.auth.models import User, Permission
from main.models import UserProfile
from main.forms import CustomRegistrationForm, AdminPageUsers
from django_registration.views import RegistrationView
from django.contrib.auth.decorators import login_required
from django.contrib.auth.mixins import LoginRequiredMixin, PermissionRequiredMixin
from django.core.exceptions import PermissionDenied

import logging

from access_controller.settings import ZENDESK_ROLES


class CustomRegistrationView(RegistrationView):
    """
    Отображение и логика работы страницы регистрации пользователя
    """
    form_class = CustomRegistrationForm
    template_name = 'django_registration/registration_form.html'
    success_url = reverse_lazy('django_registration_complete')
    is_allowed = True

    def register(self, form):
        self.is_allowed = True
        if check_user_exist(form.data['email']) and get_user_organization(form.data['email']) == 'SYSTEM':
            forms = PasswordResetForm(self.request.POST)
            if forms.is_valid():
                opts = {
                    'use_https': self.request.is_secure(),
                    'token_generator': default_token_generator,
                    'from_email': EMAIL_HOST_USER,
                    'email_template_name': 'registration/password_reset_email.html',
                    'subject_template_name': 'registration/password_reset_subject.txt',
                    'request': self.request,
                    'html_email_template_name': None,
                    'extra_email_context': None,
                }
                user = User.objects.create_user(
                    username=form.data['email'],
                    email=form.data['email'],
                    password=User.objects.make_random_password(length=50)
                )
                forms.save(**opts)
                update_profile(user.userprofile)
                self.set_permission(user)
                return user
            else:
                raise ValueError('Непредвиденная ошибка')
        else:
            self.is_allowed = False

    @staticmethod
    def set_permission(user) -> None:
        """
        Дает разрешение на просмотр страница администратора, если пользователь имеет роль admin
        """

        content_type = ContentType.objects.get_for_model(UserProfile)
        permission, created = Permission.objects.get_or_create(
            codename='has_control_access',
            content_type=content_type,
        )
        if user.userprofile.role == 'admin':
            user.user_permissions.add(permission)

    def get_success_url(self, user=None):
        """
        Возвращает url-адрес страницы, куда нужно перейти после успешной/неуспешной регистрации
        Используется самой django-registration
        """
        if self.is_allowed:
            return reverse_lazy('password_reset_done')
        else:
            return reverse_lazy('django_registration_disallowed')


@login_required()
def profile_page(request):
    """
    Отображение страницы профиля

    :param request: объект с деталями запроса
    :type request: :class:`django.http.HttpResponse`
    :return: объект ответа сервера с HTML-кодом внутри
    """
    user_profile = request.user.userprofile
    update_profile(user_profile)

    context = {
        'email': user_profile.user.email,
        'name': user_profile.name,
        'role': user_profile.role,
        'image_url': user_profile.image,
        'pagename': 'Страница профиля'
    }
    return render(request, 'pages/profile.html', context)


def main_page(request):
    logger = logging.getLogger('main.index')
    logger.info('Index page opened')
    return render(request, 'pages/index.html')


class AdminPageView(FormView, LoginRequiredMixin, PermissionRequiredMixin):
    permission_required = 'main.has_control_access'
    template_name = 'pages/adm_ruleset.html'
    form_class = AdminPageUsers
    success_url = '/control/'

    def form_valid(self, form):
        if 'engineer' in self.request.POST:
            self.make_engineers(form.cleaned_data['users'])
        elif 'light_agent' in self.request.POST:
            self.make_light_agents(form.cleaned_data['users'])
        return super().form_valid(form)

    @staticmethod
    def make_engineers(users):
        [make_engineer(user) for user in users]

    @staticmethod
    def make_light_agents(users):
        [make_light_agent(user) for user in users]

    @staticmethod
    def count_users(users):  # TODO: this func counts users from all zendesk instead of just from a model
        engineers, light_agents = 0, 0
        for user in users:
            if user.custom_role_id == ZENDESK_ROLES['engineer']:
                engineers += 1
            elif user.custom_role_id == ZENDESK_ROLES['light_agent']:
                light_agents += 1
        return engineers, light_agents

    def get_context_data(self, **kwargs):
        if self.request.user.userprofile.role != 'admin':
            raise PermissionDenied
        context = super().get_context_data(**kwargs)
        context['users'] = get_list_or_404(
            UserProfile, role='agent')
        context['engineers'], context['light_agents'] = self.count_users(get_users_list())
        return context  # TODO: need to get profile page url