NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-07-31 04:46:26 -06:00
Code Issues Actions 8 Packages Projects Releases Wiki Activity

Add policy (name) filter to VLANTranslationRuleFilterSet

Browse Source
This commit is contained in:
Brian Tiemann 2024-10-28 15:45:08 -04:00
parent 04fb92fa98
commit 94d7015c5e
2 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
netbox/ipam/filtersets.py
View File

@ -1127,10 +1127,16 @@ class VLANTranslationRuleFilterSet(NetBoxModelFilterSet):
queryset=VLANTranslationPolicy.objects.all(), queryset=VLANTranslationPolicy.objects.all(),
label=_('VLAN Translation Policy (ID)'), label=_('VLAN Translation Policy (ID)'),
) )
policy = django_filters.ModelMultipleChoiceFilter(
field_name='policy__name',
queryset=VLANTranslationPolicy.objects.all(),
to_field_name='name',
label=_('VLAN Translation Policy (name)'),
)
class Meta: class Meta:
model = VLANTranslationRule model = VLANTranslationRule
fields = ('id', 'policy_id', 'local_vid', 'remote_vid', 'description') fields = ('id', 'policy_id', 'policy', 'local_vid', 'remote_vid', 'description')
def search(self, queryset, name, value): def search(self, queryset, name, value):
if not value.strip(): if not value.strip():

5
netbox/ipam/tests/test_filtersets.py
View File

@ -1976,6 +1976,11 @@ class VLANTranslationRuleTestCase(TestCase, ChangeLoggedFilterSetTests):
params = {'policy_id': [policies[0].pk]} params = {'policy_id': [policies[0].pk]}
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2) self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
def test_policy(self):
policies = VLANTranslationPolicy.objects.all()[:2]
params = {'policy': [policies[0].name]}
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
def test_local_vid(self): def test_local_vid(self):
params = {'local_vid': [100]} params = {'local_vid': [100]}
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2) self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)