wkootandManuel Hartung
aa41f8a402
* Updated imports for Netbox 3.6.1 Closes #21 and #23 * Changed Fieldtypes of Start and Expiration Date Closes #22 * Bump to NetBox v3.6.1 * Add pyproject.toml and python workflow --------- Co-authored-by: Manuel Hartung <manuel.hartung@atos.net>
79 lines
2.5 KiB
Python
79 lines
2.5 KiB
Python
from django.db.models import Q
|
|
|
|
from netbox.filtersets import NetBoxModelFilterSet
|
|
from netbox_slm.models import SoftwareProduct, SoftwareProductVersion, SoftwareProductInstallation, SoftwareLicense
|
|
|
|
|
|
class SoftwareProductFilterSet(NetBoxModelFilterSet):
|
|
"""Filter capabilities for SoftwareProduct instances."""
|
|
|
|
class Meta:
|
|
model = SoftwareProduct
|
|
fields = tuple()
|
|
|
|
def search(self, queryset, name, value):
|
|
"""Perform the filtered search."""
|
|
if not value.strip():
|
|
return queryset
|
|
qs_filter = Q(name__icontains=value) | Q(manufacturer__name__icontains=value)
|
|
return queryset.filter(qs_filter)
|
|
|
|
|
|
class SoftwareProductVersionFilterSet(NetBoxModelFilterSet):
|
|
"""Filter capabilities for SoftwareProductVersion instances."""
|
|
|
|
class Meta:
|
|
model = SoftwareProductVersion
|
|
fields = ("software_product",)
|
|
|
|
def search(self, queryset, name, value):
|
|
"""Perform the filtered search."""
|
|
if not value.strip():
|
|
return queryset
|
|
qs_filter = (
|
|
Q(name__icontains=value)
|
|
| Q(software_product__name__icontains=value)
|
|
| Q(software_product__manufacturer__name__icontains=value)
|
|
)
|
|
return queryset.filter(qs_filter)
|
|
|
|
|
|
class SoftwareProductInstallationFilterSet(NetBoxModelFilterSet):
|
|
"""Filter capabilities for SoftwareProductInstallation instances."""
|
|
|
|
class Meta:
|
|
model = SoftwareProductInstallation
|
|
fields = ("software_product",)
|
|
|
|
def search(self, queryset, name, value):
|
|
"""Perform the filtered search."""
|
|
if not value.strip():
|
|
return queryset
|
|
qs_filter = (
|
|
Q(software_product__name__icontains=value)
|
|
| Q(software_product__manufacturer__name__icontains=value)
|
|
| Q(version__name__icontains=value)
|
|
)
|
|
return queryset.filter(qs_filter)
|
|
|
|
|
|
class SoftwareLicenseFilterSet(NetBoxModelFilterSet):
|
|
"""Filter capabilities for SoftwareLicense instances."""
|
|
|
|
class Meta:
|
|
model = SoftwareLicense
|
|
fields = tuple()
|
|
|
|
def search(self, queryset, name, value):
|
|
"""Perform the filtered search."""
|
|
if not value.strip():
|
|
return queryset
|
|
qs_filter = (
|
|
Q(name__icontains=value)
|
|
| Q(software_product__name__icontains=value)
|
|
| Q(version__name__icontains=value)
|
|
| Q(installation__device__name__icontains=value)
|
|
| Q(installation__virtualmachine__name__icontains=value)
|
|
)
|
|
return queryset.filter(qs_filter)
|