mirror of
https://github.com/netbox-community/netbox.git
synced 2024-05-10 07:54:54 +00:00
2aa51d0d94
* adds contact assignment bulk import #11307 * Remove unsupported tags field added by NetBoxModelImportForm --------- Co-authored-by: Jeremy Stretch <jstretch@netboxlabs.com>
110 lines
2.8 KiB
Python
110 lines
2.8 KiB
Python
from django.contrib.contenttypes.models import ContentType
|
|
from django.utils.translation import gettext as _
|
|
from netbox.forms import NetBoxModelImportForm
|
|
from tenancy.models import *
|
|
from utilities.forms.fields import CSVContentTypeField, CSVModelChoiceField, SlugField
|
|
|
|
__all__ = (
|
|
'ContactAssignmentImportForm',
|
|
'ContactImportForm',
|
|
'ContactGroupImportForm',
|
|
'ContactRoleImportForm',
|
|
'TenantImportForm',
|
|
'TenantGroupImportForm',
|
|
)
|
|
|
|
|
|
#
|
|
# Tenants
|
|
#
|
|
|
|
class TenantGroupImportForm(NetBoxModelImportForm):
|
|
parent = CSVModelChoiceField(
|
|
queryset=TenantGroup.objects.all(),
|
|
required=False,
|
|
to_field_name='name',
|
|
help_text=_('Parent group')
|
|
)
|
|
slug = SlugField()
|
|
|
|
class Meta:
|
|
model = TenantGroup
|
|
fields = ('name', 'slug', 'parent', 'description', 'tags')
|
|
|
|
|
|
class TenantImportForm(NetBoxModelImportForm):
|
|
slug = SlugField()
|
|
group = CSVModelChoiceField(
|
|
queryset=TenantGroup.objects.all(),
|
|
required=False,
|
|
to_field_name='name',
|
|
help_text=_('Assigned group')
|
|
)
|
|
|
|
class Meta:
|
|
model = Tenant
|
|
fields = ('name', 'slug', 'group', 'description', 'comments', 'tags')
|
|
|
|
|
|
#
|
|
# Contacts
|
|
#
|
|
|
|
class ContactGroupImportForm(NetBoxModelImportForm):
|
|
parent = CSVModelChoiceField(
|
|
queryset=ContactGroup.objects.all(),
|
|
required=False,
|
|
to_field_name='name',
|
|
help_text=_('Parent group')
|
|
)
|
|
slug = SlugField()
|
|
|
|
class Meta:
|
|
model = ContactGroup
|
|
fields = ('name', 'slug', 'parent', 'description', 'tags')
|
|
|
|
|
|
class ContactRoleImportForm(NetBoxModelImportForm):
|
|
slug = SlugField()
|
|
|
|
class Meta:
|
|
model = ContactRole
|
|
fields = ('name', 'slug', 'description')
|
|
|
|
|
|
class ContactImportForm(NetBoxModelImportForm):
|
|
group = CSVModelChoiceField(
|
|
queryset=ContactGroup.objects.all(),
|
|
required=False,
|
|
to_field_name='name',
|
|
help_text=_('Assigned group')
|
|
)
|
|
|
|
class Meta:
|
|
model = Contact
|
|
fields = ('name', 'title', 'phone', 'email', 'address', 'link', 'group', 'description', 'comments', 'tags')
|
|
|
|
|
|
class ContactAssignmentImportForm(NetBoxModelImportForm):
|
|
content_type = CSVContentTypeField(
|
|
queryset=ContentType.objects.all(),
|
|
help_text=_("One or more assigned object types")
|
|
)
|
|
contact = CSVModelChoiceField(
|
|
queryset=Contact.objects.all(),
|
|
to_field_name='name',
|
|
help_text=_('Assigned contact')
|
|
)
|
|
role = CSVModelChoiceField(
|
|
queryset=ContactRole.objects.all(),
|
|
to_field_name='name',
|
|
help_text=_('Assigned role')
|
|
)
|
|
|
|
# Remove the tags field added by NetBoxModelImportForm (unsupported by ContactAssignment)
|
|
tags = None
|
|
|
|
class Meta:
|
|
model = ContactAssignment
|
|
fields = ('content_type', 'object_id', 'contact', 'priority', 'role')
|