76 lines
1.7 KiB
Python
76 lines
1.7 KiB
Python
from django.contrib.auth.mixins import PermissionRequiredMixin, LoginRequiredMixin
|
|
from django.forms import inlineformset_factory
|
|
from django.views import generic
|
|
|
|
from access.forms import TeamForm
|
|
from access.mixin import *
|
|
from access.models import *
|
|
|
|
|
|
|
|
class IndexView(PermissionRequiredMixin, OrganizationPermission, generic.ListView):
|
|
permission_required = 'access.view_organization'
|
|
template_name = 'access/index.html.j2'
|
|
context_object_name = "organization_list"
|
|
|
|
|
|
def get_queryset(self):
|
|
|
|
if self.request.user.is_superuser:
|
|
|
|
return Organization.objects.filter()
|
|
|
|
else:
|
|
|
|
return Organization.objects.filter(pk__in=self.user_organizations())
|
|
|
|
|
|
|
|
class View(LoginRequiredMixin, OrganizationPermission, generic.UpdateView):
|
|
model = Organization
|
|
permission_required = 'access.view_organization'
|
|
template_name = "access/organization.html.j2"
|
|
fields = ["name", 'id']
|
|
|
|
|
|
def get_success_url(self, **kwargs):
|
|
return f"/organization/{self.kwargs['pk']}/"
|
|
|
|
def get_queryset(self):
|
|
|
|
return Organization.objects.filter(pk=self.kwargs['pk'])
|
|
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
|
|
organization = Organization.objects.get(pk=self.kwargs['pk'])
|
|
|
|
context['organization'] = organization
|
|
|
|
TeamsForm = inlineformset_factory(Organization, Team, fields=["team_name", 'id'], fk_name='organization', extra=1)
|
|
formset = TeamsForm(instance=organization)
|
|
|
|
context['formset'] = formset
|
|
|
|
return context
|
|
|
|
|
|
|
|
class Change(LoginRequiredMixin, OrganizationPermission, generic.DetailView):
|
|
pass
|
|
|
|
|
|
|
|
class Delete(LoginRequiredMixin, OrganizationPermission, generic.DetailView):
|
|
pass
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|