Initial work on #969: Custom links

netbox/extras/admin.py

@@ -3,7 +3,7 @@
 
 from netbox.admin import admin_site
 from utilities.forms import LaxURLField
-from .models import CustomField, CustomFieldChoice, Graph, ExportTemplate, TopologyMap, Webhook
+from .models import CustomField, CustomFieldChoice, CustomLink, Graph, ExportTemplate, TopologyMap, Webhook
 
 
 def order_content_types(field):
@@ -77,6 +77,15 @@ def models(self, obj):
         return ', '.join([ct.name for ct in obj.obj_type.all()])
 
 
+#
+# Custom links
+#
+
+@admin.register(CustomLink, site=admin_site)
+class CustomLinkAdmin(admin.ModelAdmin):
+    list_display = ['name', 'content_type', 'group_name', 'weight']
+
+
 #
 # Graphs
 #

netbox/extras/constants.py

@@ -35,6 +35,24 @@
     (CF_FILTER_EXACT, 'Exact'),
 )
 
+# Custom links
+BUTTON_CLASS_DEFAULT = 'default'
+BUTTON_CLASS_PRIMARY = 'primary'
+BUTTON_CLASS_SUCCESS = 'success'
+BUTTON_CLASS_INFO = 'info'
+BUTTON_CLASS_WARNING = 'warning'
+BUTTON_CLASS_DANGER = 'danger'
+BUTTON_CLASS_LINK = 'link'
+BUTTON_CLASS_CHOICES = (
+    (BUTTON_CLASS_DEFAULT, 'Default'),
+    (BUTTON_CLASS_PRIMARY, 'Primary (blue)'),
+    (BUTTON_CLASS_SUCCESS, 'Success (green)'),
+    (BUTTON_CLASS_INFO, 'Info (aqua)'),
+    (BUTTON_CLASS_WARNING, 'Warning (orange)'),
+    (BUTTON_CLASS_DANGER, 'Danger (red)'),
+    (BUTTON_CLASS_LINK, 'None'),
+)
+
 # Graph types
 GRAPH_TYPE_INTERFACE = 100
 GRAPH_TYPE_PROVIDER = 200

netbox/extras/migrations/0022_custom_links.py

@@ -0,0 +1,32 @@
+# Generated by Django 2.2 on 2019-04-15 19:28
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('contenttypes', '0002_remove_content_type_name'),
+        ('extras', '0021_add_color_comments_changelog_to_tag'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='CustomLink',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False)),
+                ('name', models.CharField(max_length=100, unique=True)),
+                ('text', models.CharField(max_length=200)),
+                ('url', models.CharField(max_length=200)),
+                ('weight', models.PositiveSmallIntegerField(default=100)),
+                ('group_name', models.CharField(blank=True, max_length=50)),
+                ('button_class', models.CharField(max_length=30)),
+                ('new_window', models.BooleanField()),
+                ('content_type', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='contenttypes.ContentType')),
+            ],
+            options={
+                'ordering': ['group_name', 'weight', 'name'],
+            },
+        ),
+    ]

netbox/extras/models.py

@@ -300,6 +300,53 @@ def delete(self, using=None, keep_parents=False):
         CustomFieldValue.objects.filter(field__type=CF_TYPE_SELECT, serialized_value=str(pk)).delete()
 
 
+#
+# Custom links
+#
+
+class CustomLink(models.Model):
+    """
+    A custom link to an external representation of a NetBox object. The link text and URL fields accept Jinja2 template
+    code to be rendered with an object as context.
+    """
+    content_type = models.ForeignKey(
+        to=ContentType,
+        on_delete=models.CASCADE
+    )
+    name = models.CharField(
+        max_length=100,
+        unique=True
+    )
+    text = models.CharField(
+        max_length=200,
+        help_text="Template code for link text"
+    )
+    url = models.CharField(
+        max_length=200,
+        help_text="Template code for link URL"
+    )
+    weight = models.PositiveSmallIntegerField(
+        default=100
+    )
+    group_name = models.CharField(
+        max_length=50,
+        blank=True
+    )
+    button_class = models.CharField(
+        max_length=30,
+        choices=BUTTON_CLASS_CHOICES
+    )
+    new_window = models.BooleanField(
+        help_text="Force link to open in a new window"
+    )
+
+    class Meta:
+        ordering = ['group_name', 'weight', 'name']
+
+    def __str__(self):
+        return self.name
+
+
 #
 # Graphs
 #

netbox/extras/templatetags/custom_links.py

@@ -0,0 +1,68 @@
+from collections import OrderedDict
+
+from django import template
+from django.contrib.contenttypes.models import ContentType
+from django.utils.safestring import mark_safe
+from jinja2 import Environment
+
+from extras.models import CustomLink
+
+
+register = template.Library()
+
+LINK_BUTTON = '<a href="{}"{} class="btn btn-sm btn-{}">{}</a>\n'
+GROUP_BUTTON = '<div class="btn-group">\n' \
+               '<button type="button" class="btn btn-sm btn-{} dropdown-toggle" data-toggle="dropdown">\n' \
+               '{} <span class="caret"></span>\n' \
+               '</button>\n' \
+               '<ul class="dropdown-menu pull-right">\n'
+GROUP_LINK = '<li><a href="{}"{}>{}</a></li>\n'
+
+
+@register.simple_tag()
+def custom_links(obj):
+    """
+    Render all applicable links for the given object.
+    """
+    content_type = ContentType.objects.get_for_model(obj)
+    custom_links = CustomLink.objects.filter(content_type=content_type)
+    if not custom_links:
+        return ''
+
+    context = {
+        'obj': obj,
+    }
+    template_code = ''
+    group_names = OrderedDict()
+
+    # Organize custom links by group
+    for cl in custom_links:
+        if cl.group_name and cl.group_name in group_names:
+            group_names[cl.group_name].append(cl)
+        elif cl.group_name:
+            group_names[cl.group_name] = [cl]
+
+    # Add non-grouped links
+    for cl in custom_links:
+        if not cl.group_name:
+            link_target = ' target="_blank"' if cl.new_window else ''
+            template_code += LINK_BUTTON.format(
+                cl.url, link_target, cl.button_class, cl.text
+            )
+
+    # Add grouped links to template
+    for group, links in group_names.items():
+        template_code += GROUP_BUTTON.format(
+            links[0].button_class, group
+        )
+        for cl in links:
+            link_target = ' target="_blank"' if cl.new_window else ''
+            template_code += GROUP_LINK.format(
+                cl.url, link_target, cl.text
+            )
+        template_code += '</ul>\n</div>\n'
+
+    # Render template
+    rendered = Environment().from_string(source=template_code).render(**context)
+
+    return mark_safe(rendered)

netbox/templates/dcim/device.html

@@ -1,6 +1,7 @@
 {% extends '_base.html' %}
 {% load static %}
 {% load helpers %}
+{% load custom_links %}
 
 {% block title %}{{ device }}{% endblock %}
 
@@ -64,6 +65,9 @@
     </div>
     <h1>{{ device }}</h1>
     {% include 'inc/created_updated.html' with obj=device %}
+    <div class="pull-right noprint">
+        {% custom_links device %}
+    </div>
     <ul class="nav nav-tabs">
         <li role="presentation"{% if not active_tab %} class="active"{% endif %}>
             <a href="{% url 'dcim:device' pk=device.pk %}">Device</a>