Top-Button redirects to List-View

Change the redirect target of the top "Add Attachment" button from the
url you are currently at to the attachment-list-view of the object
where the attachment will/was created. This makes it consistent with for
example the "Add Component" buttons for devices.

Author: m-hau

netbox_attachments/template_content.py

@@ -57,7 +57,7 @@ def get_display_preference(app_model_name: str) -> str:
     return display_settings.get(app_model_name, default_display)
 
 
-def create_add_attachment_button(model_name: str) -> Type[PluginTemplateExtension]:
+def create_add_attachment_button(model_name: str, url_pattern_name: str) -> Type[PluginTemplateExtension]:
     """
     Creates an 'add attachment' button extension for a model.
 
@@ -72,12 +72,13 @@ class AddAttachmentButton(PluginTemplateExtension):
         models = [model_name]
 
         def buttons(self):
-            return self.render("netbox_attachments/add_attachment_button.html")
+            return self.render("netbox_attachments/add_attachment_button.html",
+                               extra_context={'object_type_attachement_list': url_pattern_name})
 
     return AddAttachmentButton
 
 
-def register_attachment_tab_view(model: Type[Model]) -> None:
+def register_attachment_tab_view(model: Type[Model]) -> str:
     """
     Creates and registers an attachment tab view for a model.
 
@@ -115,6 +116,8 @@ def get_children(self, request, parent):
 
     register_model_view(model, name=view_name, path=view_path)(AttachmentTabView)
 
+    return view_name
+
 
 def get_template_extensions() -> List[Type[PluginTemplateExtension]]:
     """
@@ -154,12 +157,14 @@ def get_template_extensions() -> List[Type[PluginTemplateExtension]]:
 
             # Handle display as additional tab
             if display_preference == "additional_tab":
+                # Register tab view
+                view_name = register_attachment_tab_view(model)
+
                 # Add button if configured
                 if should_add_button:
-                    extensions.append(create_add_attachment_button(app_model_name))
+                    url_pattern_name = f"{app_label}:{model_name}_{view_name}"
+                    extensions.append(create_add_attachment_button(app_model_name, url_pattern_name))
 
-                # Register tab view
-                register_attachment_tab_view(model)
                 continue
 
             # Create panel extension in the specified location

netbox_attachments/templates/netbox_attachments/add_attachment_button.html

@@ -1,5 +1,5 @@
 {% if perms.netbox_attachments.add_netboxattachment %}
-    <a href="{% url 'plugins:netbox_attachments:netboxattachment_add' %}?object_type={{ object|content_type_id }}&object_id={{ object.pk }}&return_url={{ object.get_absolute_url }}"
+    <a href="{% url 'plugins:netbox_attachments:netboxattachment_add' %}?object_type={{ object|content_type_id }}&object_id={{ object.pk }}&return_url={% url object_type_attachement_list pk=object.pk %}"
        class="btn btn-secondary">
         <i class="mdi mdi-paperclip" aria-hidden="true"></i>
         Add Attachment