sharix-open-webapp-base/sharix_admin/views/partner.py

from dbsynce.models import Company, Documents, DocumentFile
from django.contrib import messages
from django.core.files.storage import default_storage
from django.db import transaction
from django.shortcuts import get_object_or_404, redirect
from django.urls import reverse_lazy
from django.views.generic import DetailView
from django.views.generic.edit import FormView

from sharix_admin.forms import CompanyForm, DocumentUploadForm
from sharix_admin.utils import *
from .base import BaseView

from django.contrib.auth.mixins import UserPassesTestMixin
from django_tables2 import SingleTableView
from sharix_admin.tables import PartnersTable

from django.contrib.auth.decorators import login_required
from django.http import JsonResponse
from django.utils.translation import gettext as _
from django_tables2 import SingleTableView

from sharix_admin.tables import PartnersTable
from sharix_admin.utils import group_required

class MyPartnerBaseView(BaseView):
    page_name = 'partner'

    # Проверяем состояит ли текущий пользователь в группе PARTNER-ADMIN
    def test_func(self):
        return "PARTNER-ADMIN" in self.user_groups

class PartnerDetailView(BaseView, DetailView):
    model = Company 
    template_name = 'sharix_admin/partner_detail.html'
    #template_name = 'sharix_admin/partner.html'
    context_object_name = 'company'
    page_title = 'О партнере'
    page_name = 'partner_detail'

    def detail_view(request, primary_key):
        try:
            company = Company.objects.get(pk=primary_key)
        except Company.DoesNotExist:
            raise Http404('Company does not exist')

        return render(request, template_name, context={'company': company})
    

class MyPartnerDetailView(MyPartnerBaseView, DetailView):
    model = Company
    template_name = 'sharix_admin/partner.html'
    context_object_name = 'company'
    page_title = 'О партнере'

    def get_object(self, queryset=None):
        return get_object_or_404(Company, repr=self.request.user)

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)

        if not pk:
            docs = Documents.objects.filter(
                user=pk,
                company=company.objects.filter(user=pk)
                ).prefetch_related('files').order_by('doc_type')
                #user=self.request.user,
                #company=self.object
            #).prefetch_related('files').order_by('doc_type')

            context.update({"docs": docs})
        else:
            docs = Documents.objects.filter(
                user=self.request.user,
                company=self.object
            ).prefetch_related('files').order_by('doc_type')

            context.update({"docs": docs})

        return context


class PartnerEditView(MyPartnerBaseView, FormView):
    template_name = 'sharix_admin/partner_edit.html'
    form_class = CompanyForm
    success_url = reverse_lazy('partner_detail')
    page_title = 'Изменение данных партнера'

    def get_form_kwargs(self):
        kwargs = super().get_form_kwargs()
        kwargs['instance'] = get_object_or_404(Company, repr=self.request.user)
        return kwargs

    def form_valid(self, form):
        with transaction.atomic():
            # Сохраняем изменения
            form.save()

            # Получаем текущий объект компании и деактивируем ее
            current_company = form.instance
            current_company.deactivate()

            # Выполняем пересоздание тикета на активацию партнера
            current_company.ticket_status.archive()
            create_ticket_company_activation(self.request.user, current_company)

        # Отправляем полAьзователю уведомление на страницу об успехе операции
        messages.success(self.request, 'Данные успешно изменены и теперь проходят проверку!')
        return super().form_valid(form)


class PartnerDocUploadView(MyPartnerBaseView, FormView):
    # FIXME: Загрузка новых документов должна деактивировать текущего партнера (предварительно это работа обработчиков)
    template_name = 'sharix_admin/partner/doc_upload.html'
    form_class = DocumentUploadForm
    success_url = reverse_lazy('partner_detail')

    def dispatch(self, request, *args, **kwargs):
        self.company = get_object_or_404(Company, repr=self.request.user)
        self.doc = Documents.objects.filter(
            user=self.request.user,
            company=self.company,
            doc_type=kwargs.get('doc_code')
        ).first()

        self.doc_name = self.doc.get_doc_type_display()
        self.page_title = "Изменение документа партнера: " + self.doc_name

        return super().dispatch(request, *args, **kwargs)

    def form_valid(self, form):
        with transaction.atomic():
            # Удаляем существующие файлы и записи из базы данных
            existing_files = DocumentFile.objects.filter(document=self.doc)
            for existing_file in existing_files:
                # Удаляем файл с сервера
                if default_storage.exists(existing_file.file.path):
                    default_storage.delete(existing_file.file.path)

                # Удаляем запись о файле из базы данных
                existing_file.delete()

            # Загрузка новых файлов
            doc_file = self.request.FILES.getlist("doc_file")
            for file in doc_file:
                DocumentFile.objects.create(
                    document=self.doc,
                    file=file
                )

            # Создание нового тикета и архивация старого
            if self.doc.ticket_status:
                self.doc.ticket_status.archive()

            self.doc.expire_date = self.request.POST.get('doc_expire_date') if self.request.POST[
                'doc_expire_date'] else None
            self.doc.ticket_status = create_ticket_company_docs_verification(self.request.user, self.company, self.doc)

            self.doc.save()

        # Отправляем пользователю уведомление на страницу об успехе операции
        messages.success(self.request,
                         f'Файлы документа "{self.doc_name}" успешно загружены и теперь проходят проверку!')
        return super().form_valid(form)


class PartnerDocView(BaseView, DetailView):
#class PartnerDocView(PartnerBaseView, DetailView):
    model = Documents
    template_name = 'sharix_admin/partner/doc.html'
    context_object_name = 'doc'

    def dispatch(self, request, *args, **kwargs):
        self.company = get_object_or_404(Company, repr=self.request.user)

        self.doc = Documents.objects.filter(
            user=self.request.user,
            company=self.company,
            doc_type=kwargs.get('doc_code')
        ).first()

        if not self.doc:
            return self.handle_no_permission()

        self.doc_name = self.doc.get_doc_type_display()
        self.page_title = _("Детали документа: ") + self.doc_name

        return super().dispatch(request, *args, **kwargs)

    def get_object(self, queryset=None):
        return self.doc

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        doc_files = DocumentFile.objects.filter(document=self.doc)
        context.update({"doc_files": doc_files})
        return context


class PartnersListView(BaseView, UserPassesTestMixin, SingleTableView):
    page_title = _('Партнеры')
    page_name = 'partners'
    table_class = PartnersTable
    queryset = Company.objects.all()
    template_name = 'sharix_admin/partners.html'

    def get_queryset(self):
        queryset = super().get_queryset()
        search_query = self.request.GET.get('q', '').strip()
        if search_query:
            queryset = queryset.filter(
                Q(id__icontains=search_query) |
                Q(legalname__icontains=search_query) |
                Q(inn__icontains=search_query) |
                Q(orgn__icontains=search_query) |
                Q(address__icontains=search_query)
            ).distinct()
        return queryset

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context.update({
            'object_list': context['object_list'],
        })
        return context

    def testing(self, queryset, is_descending):
        queryset = queryset.annotate.order_by("-" if is_descending else "")
        return (queryset, True)


#    def get_context_data(self, **kwargs):
#        context = super().get_context_data(**kwargs)
#        context.update(get_context(self.request, {
#            'title': _('Performers'),
#            'object_list': context['object_list'],
#        }))
#        return context

    def test_func(self) -> bool or None:
    #    group_names = ('PARTNER-ADMIN')
    #    if bool(self.request.user.groups.filter(name=group_names)) or self.request.user.is_superuser:
    #        return True
    #    return False
        return True

@login_required
@group_required('METASERVICE-ADMIN')
def change_partner_status(request):
    if request.method == 'POST':
        partners_id = request.POST.get('partners_id')
        new_status = request.POST.get('new_status')
        partners = Company.objects.get(pk=partners_id)
        partners.status = new_status
        partners.save()
        return JsonResponse({'status': 'success'})
    else:
        return JsonResponse({'status': 'error'})