Add NotificationGroup filters for users & groups

This commit is contained in:
Jeremy Stretch 2024-07-04 11:41:05 -04:00
parent 896c174088
commit a21d4fecbc

View File

@ -5,6 +5,7 @@ from django.db.models import Q
from django.utils.translation import gettext as _
from core.models import ObjectType
from extras.models import NotificationGroup
from netbox.filtersets import BaseFilterSet
from users.models import Group, ObjectPermission, Token
from utilities.filters import ContentTypeFilter
@ -32,6 +33,11 @@ class GroupFilterSet(BaseFilterSet):
queryset=ObjectPermission.objects.all(),
label=_('Permission (ID)'),
)
notification_group_id = django_filters.ModelMultipleChoiceFilter(
field_name='notification_groups',
queryset=NotificationGroup.objects.all(),
label=_('Notification group (ID)'),
)
class Meta:
model = Group
@ -67,6 +73,11 @@ class UserFilterSet(BaseFilterSet):
queryset=ObjectPermission.objects.all(),
label=_('Permission (ID)'),
)
notification_group_id = django_filters.ModelMultipleChoiceFilter(
field_name='notification_groups',
queryset=NotificationGroup.objects.all(),
label=_('Notification group (ID)'),
)
class Meta:
model = get_user_model()