Fix #10094 - Contact Assignments Create and Add Another

netbox/netbox/views/generic/object_views.py

@@ -327,6 +327,12 @@ def alter_object(self, obj, request, url_args, url_kwargs):
         """
         return obj
 
+    def get_extra_addanother_params(self, request, params):
+        """
+        Return a QueryDict of extra params to use on the Add Another button.
+        """
+        return params
+
     #
     # Request handlers
     #
@@ -399,6 +405,7 @@ def post(self, request, *args, **kwargs):
 
                     # If cloning is supported, pre-populate a new instance of the form
                     params = prepare_cloned_fields(obj)
+                    params = self.get_extra_addanother_params(request, params)
                     if params:
                         if 'return_url' in request.GET:
                             params['return_url'] = request.GET.get('return_url')

netbox/tenancy/views.py

@@ -1,4 +1,5 @@
 from django.contrib.contenttypes.models import ContentType
+from django.http import QueryDict
 from django.shortcuts import get_object_or_404
 
 from circuits.models import Circuit
@@ -365,6 +366,15 @@ def alter_object(self, instance, request, args, kwargs):
             instance.object = get_object_or_404(content_type.model_class(), pk=request.GET.get('object_id'))
         return instance
 
+    def get_extra_addanother_params(self, request, params: dict):
+        if not params:
+            params = QueryDict(mutable=True)
+
+        params['content_type'] = request.GET.get('content_type')
+        params['object_id'] = request.GET.get('object_id')
+
+        return params
+
 
 class ContactAssignmentDeleteView(generic.ObjectDeleteView):
     queryset = ContactAssignment.objects.all()

netbox/utilities/utils.py

@@ -285,7 +285,7 @@ def prepare_cloned_fields(instance):
     """
     # Generate the clone attributes from the instance
     if not hasattr(instance, 'clone'):
-        return QueryDict()
+        return QueryDict(mutable=True)
     attrs = instance.clone()
 
     # Prepare querydict parameters