nautobot 2.4.10__py3-none-any.whl → 2.4.12__py3-none-any.whl

nautobot/dcim/filters/__init__.py

@@ -74,6 +74,7 @@ from nautobot.dcim.models import (
     Module,
     ModuleBay,
     ModuleBayTemplate,
+    ModuleFamily,
     ModuleType,
     Platform,
     PowerFeed,
@@ -137,6 +138,7 @@ __all__ = (
     "ManufacturerFilterSet",
     "ModuleBayFilterSet",
     "ModuleBayTemplateFilterSet",
+    "ModuleFamilyFilterSet",
     "ModuleFilterSet",
     "ModuleTypeFilterSet",
     "PathEndpointFilterSet",
@@ -959,6 +961,12 @@ class DeviceFilterSet(
         field_name="controller_managed_device_group__wireless_networks",
         label="Has wireless networks",
     )
+    controller = NaturalKeyOrPKMultipleChoiceFilter(
+        field_name="controller_managed_device_group__controller",
+        queryset=Controller.objects.all(),
+        to_field_name="name",
+        label="Controller (name or ID)",
+    )
 
     def filter_ip_addresses(self, queryset, name, value):
         pk_values = set(item for item in value if is_uuid(item))
@@ -2072,6 +2080,12 @@ class ModuleFilterSet(
         to_field_name="model",
         label="Module type (model or ID)",
     )
+    module_family = NaturalKeyOrPKMultipleChoiceFilter(
+        field_name="module_type__module_family",
+        queryset=ModuleFamily.objects.all(),
+        to_field_name="name",
+        label="Module family (name or ID)",
+    )
     parent_module_bay = django_filters.ModelMultipleChoiceFilter(
         queryset=ModuleBay.objects.all(),
         label="Parent Module Bay",
@@ -2082,6 +2096,13 @@ class ModuleFilterSet(
         label="Device (name or ID)",
         method="filter_device",
     )
+    compatible_with_module_bay = extend_schema_field({"type": "string", "format": "uuid"})(
+        django_filters.ModelChoiceFilter(
+            queryset=ModuleBay.objects.all(),
+            method="filter_module_bay",
+            label="Compatible with module bay (ID)",
+        )
+    )
 
     def _construct_device_filter_recursively(self, field_name, value):
         recursion_depth = MODULE_RECURSION_DEPTH_LIMIT
@@ -2107,6 +2128,12 @@ class ModuleFilterSet(
         params = self.generate_query_filter_device(value)
         return queryset.filter(params)
 
+    def filter_module_bay(self, queryset, name, value):
+        """Filter modules based on a module bay's module family."""
+        if value and value.module_family:
+            return queryset.filter(module_type__module_family=value.module_family)
+        return queryset
+
     class Meta:
         model = Module
         fields = "__all__"
@@ -2133,6 +2160,23 @@ class ModuleTypeFilterSet(DeviceTypeModuleTypeCommonFiltersMixin, NautobotFilter
             },
         },
     )
+    manufacturer = NaturalKeyOrPKMultipleChoiceFilter(
+        queryset=Manufacturer.objects.all(),
+        to_field_name="name",
+        label="Manufacturer (name or ID)",
+    )
+    module_family = NaturalKeyOrPKMultipleChoiceFilter(
+        queryset=ModuleFamily.objects.all(),
+        to_field_name="name",
+        label="Module family (name or ID)",
+    )
+    compatible_with_module_bay = extend_schema_field({"type": "string", "format": "uuid"})(
+        django_filters.ModelChoiceFilter(
+            queryset=ModuleBay.objects.all(),
+            method="filter_module_bay",
+            label="Installable in module bay (ID)",
+        )
+    )
     has_modules = RelatedMembershipBooleanFilter(
         field_name="modules",
         label="Has module instances",
@@ -2142,6 +2186,12 @@ class ModuleTypeFilterSet(DeviceTypeModuleTypeCommonFiltersMixin, NautobotFilter
         model = ModuleType
         fields = "__all__"
 
+    def filter_module_bay(self, queryset, name, value):
+        """Filter module types based on a module bay's module family."""
+        if value and value.module_family:
+            return queryset.filter(module_family=value.module_family)
+        return queryset
+
 
 class ModuleBayTemplateFilterSet(ModularDeviceComponentTemplateModelFilterSetMixin, NautobotFilterSet):
     q = SearchFilter(
@@ -2172,6 +2222,14 @@ class ModuleBayTemplateFilterSet(ModularDeviceComponentTemplateModelFilterSetMix
             },
         }
     )
+    module_family = NaturalKeyOrPKMultipleChoiceFilter(
+        queryset=ModuleFamily.objects.all(),
+        to_field_name="name",
+        label="Module family (name or ID)",
+    )
+    requires_first_party_modules = django_filters.BooleanFilter(
+        label="Requires first-party modules",
+    )
 
     class Meta:
         model = ModuleBayTemplate
@@ -2228,6 +2286,15 @@ class ModuleBayFilterSet(NautobotFilterSet):
         field_name="installed_module",
         label="Has installed module",
     )
+    module_family = NaturalKeyOrPKMultipleChoiceFilter(
+        queryset=ModuleFamily.objects.all(),
+        to_field_name="name",
+        label="Module family (name or ID)",
+    )
+    requires_first_party_modules = django_filters.BooleanFilter(
+        field_name="requires_first_party_modules",
+        label="Requires first-party modules",
+    )
 
     class Meta:
         model = ModuleBay
@@ -2362,3 +2429,52 @@ class InterfaceVDCAssignmentFilterSet(NautobotFilterSet):
             "interface",
             "virtual_device_context",
         ]
+
+
+class ModuleFamilyFilterSet(NautobotFilterSet):
+    """FilterSet for ModuleFamily objects."""
+
+    q = SearchFilter(
+        filter_predicates={
+            "name": {
+                "lookup_expr": "icontains",
+                "preprocessor": str.strip,
+            },
+            "description": {
+                "lookup_expr": "icontains",
+                "preprocessor": str.strip,
+            },
+            "module_types__model": {
+                "lookup_expr": "icontains",
+                "preprocessor": str.strip,
+            },
+            "module_types__manufacturer__name": {
+                "lookup_expr": "icontains",
+                "preprocessor": str.strip,
+            },
+        }
+    )
+
+    module_types = NaturalKeyOrPKMultipleChoiceFilter(
+        queryset=ModuleType.objects.all(),
+        to_field_name="model",
+        label="Module types (model or ID)",
+    )
+
+    module_bay_id = extend_schema_field({"type": "array", "items": {"type": "string", "format": "uuid"}})(
+        django_filters.ModelMultipleChoiceFilter(
+            queryset=ModuleBay.objects.all(),
+            label="Module bay (ID)",
+        )
+    )
+
+    class Meta:
+        model = ModuleFamily
+        fields = [
+            "id",
+            "name",
+            "description",
+            "module_types",
+            "module_bay_id",
+            "tags",
+        ]

nautobot/dcim/filters/mixins.py

@@ -39,6 +39,7 @@ from nautobot.dcim.models import (
     RearPortTemplate,
 )
 from nautobot.extras.filters import CustomFieldModelFilterSetMixin
+from nautobot.extras.filters.mixins import RelationshipModelFilterSetMixin
 
 
 class CableTerminationModelFilterSetMixin(django_filters.FilterSet):
@@ -101,7 +102,7 @@ class DeviceComponentModelFilterSetMixin(CustomFieldModelFilterSetMixin):
     )
 
 
-class ModularDeviceComponentModelFilterSetMixin(DeviceComponentModelFilterSetMixin):
+class ModularDeviceComponentModelFilterSetMixin(DeviceComponentModelFilterSetMixin, RelationshipModelFilterSetMixin):
     module = NaturalKeyOrPKMultipleChoiceFilter(
         queryset=Module.objects.all(),
         to_field_name="module_type__model",