access-controller/main/views.py

import logging
import os

from django.contrib.auth.decorators import login_required
from django.contrib.auth.forms import PasswordResetForm
from django.contrib.auth.mixins import LoginRequiredMixin
from django.contrib.auth.models import User
from django.contrib.auth.tokens import default_token_generator
from django.contrib.auth.views import LoginView
from django.core.exceptions import PermissionDenied
from django.http import HttpResponseRedirect
from django.shortcuts import get_list_or_404, redirect, reverse, render
from django.urls import reverse_lazy
from django.views.generic import FormView
from django_registration.views import RegistrationView
from zenpy import Zenpy
from zenpy.lib.api_objects import User as ZenpyUser

from access_controller.settings import EMAIL_HOST_USER, ZENDESK_ROLES
from main.extra_func import check_user_exist, update_profile, get_user_organization, make_engineer, make_light_agent, \
    get_users_list
from main.forms import AdminPageUsers, CustomRegistrationForm, CustomAuthenticationForm
from .models import UserProfile


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)
                return user
            else:
                raise ValueError('Непредвиденная ошибка')
        else:
            self.is_allowed = False

    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,
        'id': user_profile.id,
        'image_url': user_profile.image,
        'pagename': 'Страница профиля'
    }
    return render(request, 'pages/profile.html', context)


def auth_user(request):
    admin_creds = {
        'email': os.environ.get('ACCESS_CONTROLLER_API_EMAIL'),
        'subdomain': 'ngenix1612197338',
        'token': os.environ.get('ACCESS_CONTROLLER_API_TOKEN'),
    }
    admin = Zenpy(**admin_creds)
    zenpy_user: ZenpyUser = admin.users.search(request.user.email).values[0]
    return zenpy_user, admin


@login_required()
def work_page(request, id):
    if request.user.id == id:
        context = {
            'engineers': UserProfile.objects.filter(role="admin"),
            'agents': UserProfile.objects.filter(role="agent"),
            'pagename': 'Управление правами'
        }
        return render(request, 'pages/work.html', context)
    return redirect("login")


@login_required()
def work_hand_over(request):
    zenpy_user, admin = auth_user(request)
    if zenpy_user.role == "admin" or zenpy_user.role == "end-user":
        zenpy_user.role = "agent"
        admin.users.update(zenpy_user)
        request.user.userprofile.role = "agent"
        request.user.userprofile.save()
    return HttpResponseRedirect(reverse('work', args=(request.user.id,)))


@login_required()
def work_become_engineer(request):
    zenpy_user, admin = auth_user(request)
    if zenpy_user.role == "agent" or zenpy_user.role == "end-user":
        zenpy_user.role = "admin"
        admin.users.update(zenpy_user)
        request.user.userprofile.role = "admin"
        request.user.userprofile.save()
    return HttpResponseRedirect(reverse('work', args=(request.user.id,)))


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


class AdminPageView(FormView, LoginRequiredMixin):
    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


class CustomLoginView(LoginView):
    """
    Отображение страницы авторизации пользователя
    """
    form_class = CustomAuthenticationForm