mirror of
https://github.com/netbox-community/netbox.git
synced 2024-05-10 07:54:54 +00:00
33 lines
1.2 KiB
Python
33 lines
1.2 KiB
Python
from __future__ import unicode_literals
|
|
|
|
from django.contrib.auth.mixins import PermissionRequiredMixin
|
|
from django.shortcuts import get_object_or_404
|
|
|
|
from utilities.views import ObjectDeleteView, ObjectEditView
|
|
from .forms import ImageAttachmentForm
|
|
from .models import ImageAttachment
|
|
|
|
|
|
class ImageAttachmentEditView(PermissionRequiredMixin, ObjectEditView):
|
|
permission_required = 'extras.change_imageattachment'
|
|
model = ImageAttachment
|
|
model_form = ImageAttachmentForm
|
|
|
|
def alter_obj(self, imageattachment, request, args, kwargs):
|
|
if not imageattachment.pk:
|
|
# Assign the parent object based on URL kwargs
|
|
model = kwargs.get('model')
|
|
imageattachment.parent = get_object_or_404(model, pk=kwargs['object_id'])
|
|
return imageattachment
|
|
|
|
def get_return_url(self, request, imageattachment):
|
|
return imageattachment.parent.get_absolute_url()
|
|
|
|
|
|
class ImageAttachmentDeleteView(PermissionRequiredMixin, ObjectDeleteView):
|
|
permission_required = 'extras.delete_imageattachment'
|
|
model = ImageAttachment
|
|
|
|
def get_return_url(self, request, imageattachment):
|
|
return imageattachment.parent.get_absolute_url()
|