Fix event_types field class on EventRuleImportForm

This commit is contained in:
Jeremy Stretch 2024-07-18 16:16:14 -04:00
parent 1c2b82dedd
commit 36437c644d

View File

@ -8,12 +8,13 @@ from django.utils.translation import gettext_lazy as _
from core.models import ObjectType from core.models import ObjectType
from extras.choices import * from extras.choices import *
from extras.models import * from extras.models import *
from netbox.events import get_event_type_choices
from netbox.forms import NetBoxModelImportForm from netbox.forms import NetBoxModelImportForm
from users.models import Group, User from users.models import Group, User
from utilities.forms import CSVModelForm from utilities.forms import CSVModelForm
from utilities.forms.fields import ( from utilities.forms.fields import (
CSVChoiceField, CSVContentTypeField, CSVModelChoiceField, CSVModelMultipleChoiceField, CSVMultipleContentTypeField, CSVChoiceField, CSVContentTypeField, CSVModelChoiceField, CSVModelMultipleChoiceField, CSVMultipleChoiceField,
SlugField, CSVMultipleContentTypeField, SlugField,
) )
__all__ = ( __all__ = (
@ -187,6 +188,11 @@ class EventRuleImportForm(NetBoxModelImportForm):
queryset=ObjectType.objects.with_feature('event_rules'), queryset=ObjectType.objects.with_feature('event_rules'),
help_text=_("One or more assigned object types") help_text=_("One or more assigned object types")
) )
event_types = CSVMultipleChoiceField(
choices=get_event_type_choices(),
label=_('Event types'),
help_text=_('The event type(s) which will trigger this rule')
)
action_object = forms.CharField( action_object = forms.CharField(
label=_('Action object'), label=_('Action object'),
required=True, required=True,