from django_tables2 import SingleTableView from django.views.generic.edit import UpdateView, CreateView, DeleteView from SharixAdmin.tables import CommentsTable from django.contrib.auth.mixins import UserPassesTestMixin from SharixAdmin.forms import CommentsCreateForm, ServiceTypeUpdateForm from dbsynce.models import ServiceType from django.urls import reverse from django.utils.translation import gettext as _ class CommentsCreate(UserPassesTestMixin, CreateView): model = ServiceType form_class = CommentsCreateForm template_name = "SharixAdmin/service_type_form.html" def get_context_data(self, **kwargs): context = super().get_context_data(**kwargs) context.update({ 'title': _('Комментарии'), 'object': self.object, "current_page": "comments" }) return context def get_success_url(self): return reverse('comments') def test_func(self) -> bool or None: group_names = ('METASERVICE-ADMIN') if bool(self.request.user.groups.filter(name=group_names)) or self.request.user.is_superuser: return True return False class CommentsListView(UserPassesTestMixin, SingleTableView): table_class = CommentsTable queryset = ServiceType.objects.all() template_name = 'SharixAdmin/comments.html' def get_context_data(self, **kwargs): context = super().get_context_data(**kwargs) context.update({ 'title': 'Комментарии', 'object_list': context['object_list'], "current_page": "comments" }) return context def testing(self, queryset, is_descending): queryset = queryset.annotate.order_by("-" if is_descending else "") return (queryset, True) def test_func(self) -> bool or None: group_names = ('METASERVICE-ADMIN') if bool(self.request.user.groups.filter(name=group_names)) or self.request.user.is_superuser: return True return False