From 2a517cde9f929846f7bb806dbe0ee37fb17f0832 Mon Sep 17 00:00:00 2001
From: Jeremy Stretch <jeremy@netverity.dev>
Date: Thu, 25 Feb 2021 13:08:02 -0500
Subject: [PATCH] Add support for form fieldsets

---
 netbox/templates/generic/object_edit.html     | 49 +++++++++++++++++--
 netbox/utilities/templatetags/form_helpers.py |  8 +++
 2 files changed, 52 insertions(+), 5 deletions(-)

diff --git a/netbox/templates/generic/object_edit.html b/netbox/templates/generic/object_edit.html
index 965749a6d..7056256ec 100644
--- a/netbox/templates/generic/object_edit.html
+++ b/netbox/templates/generic/object_edit.html
@@ -30,14 +30,53 @@
                     </div>
                 {% endif %}
                 {% block form %}
-                    <div class="panel panel-default">
-                        <div class="panel-heading"><strong>{{ obj_type|capfirst }}</strong></div>
+                  {% if form.Meta.fieldsets %}
+                    {# Render grouped fields accoring to Form #}
+                    {% for group, fields in form.Meta.fieldsets %}
+                      <div class="panel panel-default">
+                          <div class="panel-heading"><strong>{{ group }}</strong></div>
+                          <div class="panel-body">
+                            {% for name in fields %}
+                              {% render_field form|getfield:name %}
+                            {% endfor %}
+                          </div>
+                      </div>
+                    {% endfor %}
+                    {% if form.custom_fields %}
+                      <div class="panel panel-default">
+                        <div class="panel-heading"><strong>Custom Fields</strong></div>
                         <div class="panel-body">
-                            {% block form_fields %}
-                                {% render_form form %}
-                            {% endblock %}
+                          {% render_custom_fields form %}
                         </div>
+                      </div>
+                    {% endif %}
+                    {% if form.tags %}
+                      <div class="panel panel-default">
+                        <div class="panel-heading"><strong>Tags</strong></div>
+                        <div class="panel-body">
+                          {% render_field form.tags %}
+                        </div>
+                      </div>
+                    {% endif %}
+                    {% if form.comments %}
+                      <div class="panel panel-default">
+                        <div class="panel-heading"><strong>Comments</strong></div>
+                        <div class="panel-body">
+                          {% render_field form.comments %}
+                        </div>
+                      </div>
+                    {% endif %}
+                  {% else %}
+                    {# Render all fields in a single group #}
+                    <div class="panel panel-default">
+                      <div class="panel-heading"><strong>{{ obj_type|capfirst }}</strong></div>
+                      <div class="panel-body">
+                        {% block form_fields %}
+                          {% render_form form %}
+                        {% endblock %}
+                      </div>
                     </div>
+                  {% endif %}
                 {% endblock %}
             </div>
         </div>
diff --git a/netbox/utilities/templatetags/form_helpers.py b/netbox/utilities/templatetags/form_helpers.py
index d3451ce86..9d46e8c37 100644
--- a/netbox/utilities/templatetags/form_helpers.py
+++ b/netbox/utilities/templatetags/form_helpers.py
@@ -4,6 +4,14 @@ from django import template
 register = template.Library()
 
 
+@register.filter()
+def getfield(form, fieldname):
+    """
+    Return the specified field of a Form.
+    """
+    return form[fieldname]
+
+
 @register.inclusion_tag('utilities/render_field.html')
 def render_field(field, bulk_nullable=False):
     """