from django.contrib.auth import decorators as auth_decorator from django.urls import reverse from django.utils.decorators import method_decorator from core.forms.comment import AddNoteForm from core.models.notes import Notes from core.views.common import AddView, ChangeView, DeleteView, IndexView from itim.forms.ports import DetailForm, PortForm from itim.models.services import Port, Service from settings.models.user_settings import UserSettings class Add(AddView): form_class = PortForm model = Port permission_required = [ 'itim.add_port', ] def get_initial(self): initial: dict = { 'organization': UserSettings.objects.get(user = self.request.user).default_organization } if 'pk' in self.kwargs: if self.kwargs['pk']: initial.update({'parent': self.kwargs['pk']}) self.model.parent.field.hidden = True return initial def get_success_url(self, **kwargs): return reverse('Settings:_ports') def get_context_data(self, **kwargs): context = super().get_context_data(**kwargs) context['content_title'] = 'New Port' return context class Change(ChangeView): context_object_name = "item" form_class = PortForm model = Port permission_required = [ 'itim.change_port', ] def get_context_data(self, **kwargs): context = super().get_context_data(**kwargs) context['content_title'] = str(self.object) return context def get_success_url(self, **kwargs): return reverse('Settings:_port_view', args=(self.kwargs['pk'],)) class Delete(DeleteView): model = Port permission_required = [ 'itim.delete_port', ] def get_context_data(self, **kwargs): context = super().get_context_data(**kwargs) context['content_title'] = 'Delete ' + str(self.object) return context def get_success_url(self, **kwargs): return reverse('Settings:_ports') class Index(IndexView): context_object_name = "items" model = Port paginate_by = 10 permission_required = [ 'itim.view_port' ] template_name = 'itim/port_index.html.j2' def get_context_data(self, **kwargs): context = super().get_context_data(**kwargs) context['model_docs_path'] = self.model._meta.app_label + self.model._meta.model_name context['content_title'] = 'Ports' return context class View(ChangeView): context_object_name = "item" form_class = DetailForm model = Port permission_required = [ 'itim.view_port', ] template_name = 'itim/port.html.j2' def get_context_data(self, **kwargs): context = super().get_context_data(**kwargs) context['services'] = Service.objects.filter(port=self.kwargs['pk']).order_by('name', 'organization') context['notes_form'] = AddNoteForm(prefix='note') context['notes'] = Notes.objects.filter(config_group=self.kwargs['pk']) context['model_pk'] = self.kwargs['pk'] context['model_name'] = self.model._meta.model_name context['model_delete_url'] = reverse('Settings:_port_delete', args=(self.kwargs['pk'],)) context['content_title'] = self.object return context @method_decorator(auth_decorator.permission_required("itim.change_service", raise_exception=True)) def post(self, request, *args, **kwargs): item = Port.objects.get(pk=self.kwargs['pk']) notes = AddNoteForm(request.POST, prefix='note') if notes.is_bound and notes.is_valid() and notes.instance.note != '': notes.instance.organization = item.organization notes.save() def get_success_url(self, **kwargs): return reverse('Settings:_port_view', args=(self.kwargs['pk'],))