mirror of
https://github.com/netbox-community/netbox.git
synced 2025-12-19 03:42:25 -06:00
131 lines
4.4 KiB
Python
131 lines
4.4 KiB
Python
from django import forms
|
|
from django.contrib.contenttypes.models import ContentType
|
|
from django.core.exceptions import ObjectDoesNotExist, ValidationError
|
|
from django.utils.translation import gettext_lazy as _
|
|
|
|
from dcim.constants import LOCATION_SCOPE_TYPES
|
|
from dcim.models import Site
|
|
from utilities.forms import get_field_value
|
|
from utilities.forms.fields import (
|
|
ContentTypeChoiceField, CSVContentTypeField, DynamicModelChoiceField,
|
|
)
|
|
from utilities.templatetags.builtins.filters import bettertitle
|
|
from utilities.forms.widgets import HTMXSelect
|
|
|
|
__all__ = (
|
|
'ScopedBulkEditForm',
|
|
'ScopedForm',
|
|
'ScopedImportForm',
|
|
)
|
|
|
|
|
|
class ScopedForm(forms.Form):
|
|
scope_type = ContentTypeChoiceField(
|
|
queryset=ContentType.objects.filter(model__in=LOCATION_SCOPE_TYPES),
|
|
widget=HTMXSelect(),
|
|
required=False,
|
|
label=_('Scope type')
|
|
)
|
|
scope = DynamicModelChoiceField(
|
|
label=_('Scope'),
|
|
queryset=Site.objects.none(), # Initial queryset
|
|
required=False,
|
|
disabled=True,
|
|
selector=True
|
|
)
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
instance = kwargs.get('instance')
|
|
initial = kwargs.get('initial', {})
|
|
|
|
if instance is not None and instance.scope:
|
|
initial['scope'] = instance.scope
|
|
kwargs['initial'] = initial
|
|
|
|
super().__init__(*args, **kwargs)
|
|
self._set_scoped_values()
|
|
|
|
def clean(self):
|
|
super().clean()
|
|
|
|
scope = self.cleaned_data.get('scope')
|
|
scope_type = self.cleaned_data.get('scope_type')
|
|
if scope_type and not scope:
|
|
raise ValidationError({
|
|
'scope': _(
|
|
"Please select a {scope_type}."
|
|
).format(scope_type=scope_type.model_class()._meta.model_name)
|
|
})
|
|
|
|
# Assign the selected scope (if any)
|
|
self.instance.scope = scope
|
|
|
|
def _set_scoped_values(self):
|
|
if scope_type_id := get_field_value(self, 'scope_type'):
|
|
try:
|
|
scope_type = ContentType.objects.get(pk=scope_type_id)
|
|
model = scope_type.model_class()
|
|
self.fields['scope'].queryset = model.objects.all()
|
|
self.fields['scope'].widget.attrs['selector'] = model._meta.label_lower
|
|
self.fields['scope'].disabled = False
|
|
self.fields['scope'].label = _(bettertitle(model._meta.verbose_name))
|
|
except ObjectDoesNotExist:
|
|
pass
|
|
|
|
if self.instance and scope_type_id != self.instance.scope_type_id:
|
|
self.initial['scope'] = None
|
|
|
|
else:
|
|
# Clear the initial scope value if scope_type is not set
|
|
self.initial['scope'] = None
|
|
|
|
|
|
class ScopedBulkEditForm(forms.Form):
|
|
scope_type = ContentTypeChoiceField(
|
|
queryset=ContentType.objects.filter(model__in=LOCATION_SCOPE_TYPES),
|
|
widget=HTMXSelect(method='post', attrs={'hx-select': '#form_fields'}),
|
|
required=False,
|
|
label=_('Scope type')
|
|
)
|
|
scope = DynamicModelChoiceField(
|
|
label=_('Scope'),
|
|
queryset=Site.objects.none(), # Initial queryset
|
|
required=False,
|
|
disabled=True,
|
|
selector=True
|
|
)
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
|
|
if scope_type_id := get_field_value(self, 'scope_type'):
|
|
try:
|
|
scope_type = ContentType.objects.get(pk=scope_type_id)
|
|
model = scope_type.model_class()
|
|
self.fields['scope'].queryset = model.objects.all()
|
|
self.fields['scope'].widget.attrs['selector'] = model._meta.label_lower
|
|
self.fields['scope'].disabled = False
|
|
self.fields['scope'].label = _(bettertitle(model._meta.verbose_name))
|
|
except ObjectDoesNotExist:
|
|
pass
|
|
|
|
|
|
class ScopedImportForm(forms.Form):
|
|
scope_type = CSVContentTypeField(
|
|
queryset=ContentType.objects.filter(model__in=LOCATION_SCOPE_TYPES),
|
|
required=False,
|
|
label=_('Scope type (app & model)')
|
|
)
|
|
|
|
def clean(self):
|
|
super().clean()
|
|
|
|
scope_id = self.cleaned_data.get('scope_id')
|
|
scope_type = self.cleaned_data.get('scope_type')
|
|
if scope_type and not scope_id:
|
|
raise ValidationError({
|
|
'scope_id': _(
|
|
"Please select a {scope_type}."
|
|
).format(scope_type=scope_type.model_class()._meta.model_name)
|
|
})
|