Djhoward12 naaps 124 child tabs

changes/124.added

@@ -0,0 +1 @@
+Added child floor plan tab to display child location floor plans in a list

changes/135.fixed

@@ -0,0 +1 @@
+Changed floor_plan column on FloorPlanTable to orderable=False to fix bug

nautobot_floor_plan/tables.py

@@ -13,7 +13,7 @@ class FloorPlanTable(BaseTable):
     """Table for list view."""
 
     pk = ToggleColumn()
-    floor_plan = tables.Column(empty_values=[])
+    floor_plan = tables.Column(empty_values=[], orderable=False)
     location = tables.Column(linkify=True)
     x_origin_seed = tables.Column()
     y_origin_seed = tables.Column()

nautobot_floor_plan/template_content.py

@@ -43,18 +43,38 @@ def buttons(self):
 </a>"""
 
     def detail_tabs(self):
-        """Add a "Floor Plan" tab to the Location detail view if the Location has a FloorPlan associated to it."""
+        """Add a "Floor Plan" tab to the Location detail view or a "Child Floor Plan" tab if the Location has Children with FloorPlans associated to them."""
         location = self.context["object"]
-        try:
-            location.floor_plan  # pylint: disable=pointless-statement
-            return [
+
+        # Initialize the tabs list
+        tabs = []
+
+        # Determine conditions
+        has_floor_plan = getattr(location, "floor_plan", None) is not None
+        has_children = location.children.exists()
+        has_child_floor_plans = location.children.filter(floor_plan__isnull=False).exists()
+
+        # Add "Floor Plan" tab if applicable
+        if has_floor_plan:
+            tabs.append(
                 {
                     "title": "Floor Plan",
                     "url": reverse("plugins:nautobot_floor_plan:location_floor_plan_tab", kwargs={"pk": location.pk}),
                 }
-            ]
-        except ObjectDoesNotExist:
-            return []
+            )
+
+        # Add "Child Floor Plan(s)" tab if applicable
+        if has_children and has_child_floor_plans:
+            tabs.append(
+                {
+                    "title": "Child Floor Plan(s)",
+                    "url": reverse(
+                        "plugins:nautobot_floor_plan:location_child_floor_plan_tab", kwargs={"pk": location.pk}
+                    ),
+                }
+            )
+
+        return tabs
 
 
 template_extensions = (LocationFloorPlanTab,)

nautobot_floor_plan/templates/nautobot_floor_plan/floorplan_retrieve.html

@@ -59,9 +59,13 @@
                 <td>{{ object.y_axis_step }}</td>
             </tr>
             <tr>
-                <td>Tiles</td>
+                <td>Tile(s)</td>
                 <td><a href="{% url 'plugins:nautobot_floor_plan:floorplantile_list' %}?floor_plan={{ object.pk }}">{{ object.tiles.count }}</a></td>
             </tr>
+            <tr>
+                <td>Child Floor Plan(s)</td>
+                <td><a href="{% url 'dcim:location' object.location.pk %}">{{ object.location|count_children_floor_plans }}</a></td>
+            </tr>
         </table>
     </div>
 {% endblock content_left_page %}

nautobot_floor_plan/templates/nautobot_floor_plan/location_child_floor_plan.html

@@ -0,0 +1,16 @@
+{% extends "dcim/location.html" %}
+{% load helpers %}
+{% load static %}
+{% load seed_helpers %}
+{% block content %}
+<div class="panel panel-default">
+    <div class="panel-heading">
+        <strong>Child Floor Plans</strong>
+        <div class="pull-right noprint">
+            {% table_config_button_small children_table %}
+        </div>
+    </div>
+    {% include 'inc/table.html' with table=children_table %}
+</div>
+{% table_config_form children_table %}
+{% endblock content %}

nautobot_floor_plan/templatetags/seed_helpers.py

@@ -29,3 +29,10 @@ def grid_location_conversion(floor_plan_tile, axis):
         grid = utils.grid_number_to_letter(grid)
 
     return f"{grid}"
+
+
+@register.filter()
+def count_children_floor_plans(location):
+    """Returns count of Children with FloorPlans for a given location."""
+    count = location.children.filter(floor_plan__isnull=False).count()
+    return count

nautobot_floor_plan/urls.py

@@ -13,6 +13,11 @@
 router.register("floor-plan-tiles", views.FloorPlanTileUIViewSet)
 urlpatterns = [
     path("locations/<uuid:pk>/floor_plan/", views.LocationFloorPlanTab.as_view(), name="location_floor_plan_tab"),
+    path(
+        "locations/<uuid:pk>/child_floor_plan/",
+        views.ChildLocationFloorPlanTab.as_view(),
+        name="location_child_floor_plan_tab",
+    ),
     path("docs/", RedirectView.as_view(url=static("nautobot_floor_plan/docs/index.html")), name="docs"),
 ]
 urlpatterns += router.urls

nautobot_floor_plan/views.py

@@ -1,5 +1,6 @@
 """Views for FloorPlan."""
 
+from django_tables2 import RequestConfig
 from nautobot.apps.views import (
     NautobotUIViewSet,
     ObjectChangeLogViewMixin,
@@ -10,6 +11,7 @@
     ObjectNotesViewMixin,
     ObjectView,
 )
+from nautobot.core.views.paginator import EnhancedPaginator, get_paginate_count
 from nautobot.dcim.models import Location
 
 from nautobot_floor_plan import filters, forms, models, tables
@@ -36,6 +38,36 @@ class LocationFloorPlanTab(ObjectView):
     template_name = "nautobot_floor_plan/location_floor_plan.html"
 
 
+class ChildLocationFloorPlanTab(ObjectView):
+    """Add a "Child Floor Plan" tab to the Location detail view."""
+
+    queryset = Location.objects.without_tree_fields().all()
+    template_name = "nautobot_floor_plan/location_child_floor_plan.html"
+
+    def get_extra_context(self, request, instance):
+        """Return child locations that have floor plans."""
+        children = (
+            Location.objects.restrict(request.user, "view")
+            .without_tree_fields()
+            .filter(parent=instance, floor_plan__isnull=False)
+            .select_related("parent", "location_type")
+        )
+
+        children_table = tables.FloorPlanTable(models.FloorPlan.objects.filter(location__in=children.all()))
+
+        paginate = {
+            "paginator_class": EnhancedPaginator,
+            "per_page": get_paginate_count(request),
+        }
+        RequestConfig(request, paginate).configure(children_table)
+
+        return {
+            "children_table": children_table,
+            "active_tab": "nautobot_floor_plan:2",
+            **super().get_extra_context(request, instance),
+        }
+
+
 class FloorPlanTileUIViewSet(
     ObjectDetailViewMixin,
     ObjectListViewMixin,