diff --git a/netbox/dcim/views.py b/netbox/dcim/views.py index b52e0afa5..0a12ba511 100644 --- a/netbox/dcim/views.py +++ b/netbox/dcim/views.py @@ -8,7 +8,7 @@ from django.db.models import Prefetch from django.forms import ModelMultipleChoiceField, MultipleHiddenInput, modelformset_factory from django.http import HttpResponse from django.shortcuts import get_object_or_404, redirect, render -from django.urls import reverse +from django.urls import reverse, resolve from django.utils.html import escape from django.utils.safestring import mark_safe from django.utils.translation import gettext as _ @@ -3131,6 +3131,13 @@ class CableEditView(generic.ObjectEditView): return obj + def get_extra_addanother_params(self, request): + return { + 'termination_a_device': resolve(request.GET.get('return_url')).kwargs.get('pk'), + 'a_terminations_type': request.GET.get('a_terminations_type'), + 'b_terminations_type': request.GET.get('b_terminations_type') + } + @register_model_view(Cable, 'delete') class CableDeleteView(generic.ObjectDeleteView):