134 lines
2.9 KiB
Python
134 lines
2.9 KiB
Python
from django.contrib.auth import decorators as auth_decorator
|
|
from django.urls import reverse
|
|
from django.utils.decorators import method_decorator
|
|
|
|
from itam.forms.device_model import DetailForm, DeviceModelForm
|
|
from itam.models.device_models import DeviceModel
|
|
|
|
from core.views.common import AddView, ChangeView, DeleteView
|
|
|
|
from settings.models.user_settings import UserSettings
|
|
|
|
|
|
|
|
class Change(ChangeView):
|
|
|
|
form_class = DeviceModelForm
|
|
|
|
context_object_name = "device_model"
|
|
|
|
model = DeviceModel
|
|
|
|
permission_required = [
|
|
'itam.change_devicemodel',
|
|
]
|
|
|
|
template_name = 'form.html.j2'
|
|
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
|
|
context['content_title'] = self.object.name
|
|
|
|
return context
|
|
|
|
def get_success_url(self, **kwargs):
|
|
|
|
return reverse('Settings:_device_model_view', args=(self.kwargs['pk'],))
|
|
|
|
|
|
|
|
class View(ChangeView):
|
|
|
|
form_class = DetailForm
|
|
|
|
context_object_name = "device_model"
|
|
|
|
model = DeviceModel
|
|
|
|
permission_required = [
|
|
'itam.view_devicemodel',
|
|
]
|
|
|
|
template_name = 'itam/device_model.html.j2'
|
|
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
|
|
context['model_pk'] = self.kwargs['pk']
|
|
context['model_name'] = self.model._meta.verbose_name.replace(' ', '')
|
|
|
|
context['model_delete_url'] = reverse('Settings:_device_model_delete', args=(self.kwargs['pk'],))
|
|
|
|
context['content_title'] = self.object.name
|
|
|
|
return context
|
|
|
|
def get_success_url(self, **kwargs):
|
|
|
|
return reverse('Settings:_device_model_view', args=(self.kwargs['pk'],))
|
|
|
|
|
|
@method_decorator(auth_decorator.permission_required("itam.change_devicemodel", raise_exception=True))
|
|
def post(self, request, *args, **kwargs):
|
|
|
|
return super().post(request, *args, **kwargs)
|
|
|
|
|
|
|
|
class Add(AddView):
|
|
|
|
form_class = DeviceModelForm
|
|
|
|
model = DeviceModel
|
|
|
|
permission_required = [
|
|
'itam.add_devicemodel',
|
|
]
|
|
|
|
template_name = 'form.html.j2'
|
|
|
|
|
|
def get_initial(self):
|
|
|
|
return {
|
|
'organization': UserSettings.objects.get(user = self.request.user).default_organization
|
|
}
|
|
|
|
|
|
def get_success_url(self, **kwargs):
|
|
|
|
return reverse('Settings:_device_models')
|
|
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
|
|
context['content_title'] = 'Add Device Model'
|
|
|
|
return context
|
|
|
|
|
|
|
|
class Delete(DeleteView):
|
|
model = DeviceModel
|
|
permission_required = [
|
|
'itam.delete_devicemodel',
|
|
]
|
|
template_name = 'form.html.j2'
|
|
|
|
|
|
def get_success_url(self, **kwargs):
|
|
|
|
return reverse('Settings:_device_models')
|
|
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
|
|
context['content_title'] = 'Delete ' + self.object.name
|
|
|
|
return context
|