10769 allow filtering of cables if have terminations

This commit is contained in:
Arthur 2023-10-02 13:59:09 -07:00
parent 786f0cc7f3
commit d56b29e502
2 changed files with 35 additions and 1 deletions

View File

@ -1745,6 +1745,14 @@ class CableFilterSet(TenancyFilterSet, NetBoxModelFilterSet):
method='filter_by_cable_end_b',
field_name='terminations__termination_id'
)
has_a_terminations = django_filters.BooleanFilter(
method='_has_a_terminations',
label=_('Has a terminations'),
)
has_b_terminations = django_filters.BooleanFilter(
method='_has_b_terminations',
label=_('Has b terminations'),
)
type = django_filters.MultipleChoiceFilter(
choices=CableTypeChoices
)
@ -1812,6 +1820,18 @@ class CableFilterSet(TenancyFilterSet, NetBoxModelFilterSet):
# Filter by termination id and cable_end type
return self.filter_by_cable_end(queryset, name, value, CableEndChoices.SIDE_B)
def _has_a_terminations(self, queryset, name, value):
if value:
return queryset.filter(terminations__cable_end=CableEndChoices.SIDE_A)
else:
return queryset.exclude(terminations__cable_end=CableEndChoices.SIDE_A)
def _has_b_terminations(self, queryset, name, value):
if value:
return queryset.filter(terminations__cable_end=CableEndChoices.SIDE_B)
else:
return queryset.exclude(terminations__cable_end=CableEndChoices.SIDE_B)
class CableTerminationFilterSet(BaseFilterSet):
termination_type = ContentTypeFilter()

View File

@ -908,7 +908,7 @@ class VirtualChassisFilterForm(TenancyFilterForm, NetBoxModelFilterSetForm):
class CableFilterForm(TenancyFilterForm, NetBoxModelFilterSetForm):
model = Cable
fieldsets = (
(None, ('q', 'filter_id', 'tag')),
(None, ('q', 'filter_id', 'has_a_terminations', 'has_b_terminations', 'tag')),
(_('Location'), ('site_id', 'location_id', 'rack_id', 'device_id')),
(_('Attributes'), ('type', 'status', 'color', 'length', 'length_unit')),
(_('Tenant'), ('tenant_group_id', 'tenant_id')),
@ -979,6 +979,20 @@ class CableFilterForm(TenancyFilterForm, NetBoxModelFilterSetForm):
choices=add_blank_choice(CableLengthUnitChoices),
required=False
)
has_a_terminations = forms.NullBooleanField(
label=_('Has a terminations'),
required=False,
widget=forms.Select(
choices=BOOLEAN_WITH_BLANK_CHOICES
)
)
has_b_terminations = forms.NullBooleanField(
label=_('Has b terminations'),
required=False,
widget=forms.Select(
choices=BOOLEAN_WITH_BLANK_CHOICES
)
)
tag = TagFilterField(model)