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

from django_tables2 import SingleTableView
from sharix_admin.utils import group_required
from sharix_admin.tables import ServiceTable
from dbsynce.models import Service
from django.contrib.auth.decorators import login_required
from django.contrib.auth.mixins import UserPassesTestMixin
from django.utils.translation import gettext as _

class ServiceListView(UserPassesTestMixin, SingleTableView):
    table_class = ServiceTable
    queryset = Service.objects.all()
    template_name = 'sharix_admin/service.html'

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context.update({
            'title': _('Rates'),
            'object_list': context['object_list']
        })
        return context
    
    def test_func(self) -> bool or None:
        group_names = ('PROVIDER')
        if bool(self.request.user.groups.filter(name=group_names)) or self.request.user.is_superuser:
            return True
        return False

@login_required
@group_required('PROVIDER')
def change_service_status(request):
    if request.method == 'POST':
        service_id = request.POST.get('service_id')
        new_status = request.POST.get('new_status')
        
        service = Service.objects.get(pk=service_id)
        service.status = new_status
        service.save()