123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566 |
- from django.views.generic import DetailView
- from django.views.generic.edit import FormView
- from django.urls import reverse_lazy
- from django.shortcuts import get_object_or_404
- from django.utils.translation import gettext as _
- from django.contrib.auth.mixins import UserPassesTestMixin
- from django.contrib import messages
- from dbsynce.models import Company, Documents
- from SharixAdmin.forms import CompanyForm
- from .base import BaseView
- class PartnerDetailView(UserPassesTestMixin, BaseView, DetailView):
- model = Company
- template_name = 'SharixAdmin/partner.html'
- context_object_name = 'company'
- page_title = _('О партнере')
- page_name = 'partner'
- # Проверяем состояит ли текущий пользователь в группе PARTNER-ADMIN
- def test_func(self):
- return self.request.user.groups.filter(name='PARTNER-ADMIN').exists()
- def get_object(self, queryset=None):
- return get_object_or_404(Company, repr_id=self.request.user)
- def get_context_data(self, **kwargs):
- context = super().get_context_data(**kwargs)
- context.update({
- "company_documents": Documents.parse_requirements(self.object.requirements)
- })
- return context
-
- class PartnerEditView(UserPassesTestMixin, BaseView, FormView):
- template_name = 'SharixAdmin/partner_edit.html'
- form_class = CompanyForm
- success_url = reverse_lazy('partner_detail')
- page_title = _('Изменение данных партнера')
- page_name = 'partner'
- # Проверяем состояит ли текущий пользователь в группе PARTNER-ADMIN
- def test_func(self):
- return self.request.user.groups.filter(name='PARTNER-ADMIN').exists()
- def get_form_kwargs(self):
- kwargs = super().get_form_kwargs()
- kwargs['instance'] = get_object_or_404(Company, repr_id=self.request.user)
- return kwargs
- def form_valid(self, form):
- # Сохраняем изменения
- form.save()
- # Получаем текущий объект компании и деактивируем ее
- current_company = form.instance
- current_company.deactivate()
- # Отправляем пользователю уведомление на страницу о успехе операции
- messages.success(self.request, 'Данные успешно изменены и теперь проходят проверку!')
- return super().form_valid(form)
|