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