mirror of
https://github.com/netbox-community/netbox.git
synced 2024-05-10 07:54:54 +00:00
60 lines
1.7 KiB
Python
60 lines
1.7 KiB
Python
from django import forms
|
|
from django.db.models import Count
|
|
|
|
from extras.forms import CustomFieldForm, CustomFieldBulkEditForm, CustomFieldFilterForm
|
|
from utilities.forms import BootstrapMixin, BulkImportForm, CommentField, CSVDataField, FilterChoiceField, SlugField
|
|
|
|
from .models import Tenant, TenantGroup
|
|
|
|
|
|
#
|
|
# Tenant groups
|
|
#
|
|
|
|
class TenantGroupForm(forms.ModelForm, BootstrapMixin):
|
|
slug = SlugField()
|
|
|
|
class Meta:
|
|
model = TenantGroup
|
|
fields = ['name', 'slug']
|
|
|
|
|
|
#
|
|
# Tenants
|
|
#
|
|
|
|
class TenantForm(BootstrapMixin, CustomFieldForm):
|
|
slug = SlugField()
|
|
comments = CommentField()
|
|
|
|
class Meta:
|
|
model = Tenant
|
|
fields = ['name', 'slug', 'group', 'description', 'comments']
|
|
|
|
|
|
class TenantFromCSVForm(forms.ModelForm):
|
|
group = forms.ModelChoiceField(TenantGroup.objects.all(), required=False, to_field_name='name',
|
|
error_messages={'invalid_choice': 'Group not found.'})
|
|
|
|
class Meta:
|
|
model = Tenant
|
|
fields = ['name', 'slug', 'group', 'description']
|
|
|
|
|
|
class TenantImportForm(BulkImportForm, BootstrapMixin):
|
|
csv = CSVDataField(csv_form=TenantFromCSVForm)
|
|
|
|
|
|
class TenantBulkEditForm(BootstrapMixin, CustomFieldBulkEditForm):
|
|
pk = forms.ModelMultipleChoiceField(queryset=Tenant.objects.all(), widget=forms.MultipleHiddenInput)
|
|
group = forms.ModelChoiceField(queryset=TenantGroup.objects.all(), required=False)
|
|
|
|
class Meta:
|
|
nullable_fields = ['group']
|
|
|
|
|
|
class TenantFilterForm(BootstrapMixin, CustomFieldFilterForm):
|
|
model = Tenant
|
|
group = FilterChoiceField(queryset=TenantGroup.objects.annotate(filter_count=Count('tenants')),
|
|
to_field_name='slug', null_option=(0, 'None'))
|