PyRIGS/RIGS/views/crud.py

from PyRIGS.views import GenericListView, GenericDetailView, GenericUpdateView, GenericCreateView, ModalURLMixin
from RIGS import models


class PersonList(GenericListView):
    model = models.Person

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context['page_title'] = "People"
        context['create'] = 'person_create'
        context['edit'] = 'person_update'
        context['can_edit'] = self.request.user.has_perm('RIGS.change_person')
        context['detail'] = 'person_detail'
        return context


class PersonDetail(GenericDetailView):
    model = models.Person

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context['history_link'] = 'person_history'
        context['detail_link'] = 'person_detail'
        context['update_link'] = 'person_update'
        context['associated'] = 'partials/associated_organisations.html'
        context['associated2'] = 'partials/associated_events.html'
        context['can_edit'] = self.request.user.has_perm('RIGS.change_person')
        return context


class PersonCreate(GenericCreateView, ModalURLMixin):
    model = models.Person
    fields = ['name', 'phone', 'email', 'address', 'notes']

    def get_success_url(self):
        return self.get_close_url('person_update', 'person_detail')


class PersonUpdate(GenericUpdateView, ModalURLMixin):
    model = models.Person
    fields = ['name', 'phone', 'email', 'address', 'notes']

    def get_success_url(self):
        return self.get_close_url('person_update', 'person_detail')


class OrganisationList(GenericListView):
    model = models.Organisation

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context['create'] = 'organisation_create'
        context['edit'] = 'organisation_update'
        context['can_edit'] = self.request.user.has_perm('RIGS.change_organisation')
        context['detail'] = 'organisation_detail'
        context['union_account'] = True
        return context


class OrganisationDetail(GenericDetailView):
    model = models.Organisation

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context['history_link'] = 'organisation_history'
        context['detail_link'] = 'organisation_detail'
        context['update_link'] = 'organisation_update'
        context['associated'] = 'partials/associated_people.html'
        context['associated2'] = 'partials/associated_events.html'
        context['can_edit'] = self.request.user.has_perm('RIGS.change_organisation')
        return context


class OrganisationCreate(GenericCreateView, ModalURLMixin):
    model = models.Organisation
    fields = ['name', 'phone', 'email', 'address', 'notes', 'union_account']

    def get_success_url(self):
        return self.get_close_url('organisation_update', 'organisation_detail')


class OrganisationUpdate(GenericUpdateView, ModalURLMixin):
    model = models.Organisation
    fields = ['name', 'phone', 'email', 'address', 'notes', 'union_account']

    def get_success_url(self):
        return self.get_close_url('organisation_update', 'organisation_detail')


class VenueList(GenericListView):
    model = models.Venue

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context['create'] = 'venue_create'
        context['edit'] = 'venue_update'
        context['can_edit'] = self.request.user.has_perm('RIGS.change_venue')
        context['detail'] = 'venue_detail'
        return context


class VenueDetail(GenericDetailView):
    model = models.Venue

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context['history_link'] = 'venue_history'
        context['detail_link'] = 'venue_detail'
        context['update_link'] = 'venue_update'
        context['associated2'] = 'partials/associated_events.html'
        context['can_edit'] = self.request.user.has_perm('RIGS.change_venue')
        return context


class VenueCreate(GenericCreateView, ModalURLMixin):
    model = models.Venue
    fields = ['name', 'phone', 'email', 'address', 'notes', 'three_phase_available']

    def get_success_url(self):
        return self.get_close_url('venue_update', 'venue_detail')


class VenueUpdate(GenericUpdateView, ModalURLMixin):
    model = models.Venue
    fields = ['name', 'phone', 'email', 'address', 'notes', 'three_phase_available']

    def get_success_url(self):
        return self.get_close_url('venue_update', 'venue_detail')