NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-07-29 03:46:25 -06:00
Code Issues Actions 8 Packages Projects Releases Wiki Activity

Use TreeNodeChoiceField for MPTT objects

Browse Source
This commit is contained in:
Jeremy Stretch 2019-08-14 14:03:11 -04:00
parent 81bb6a59e2
commit f0ec068650
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
netbox/extras/scripts.py
View File

@ -10,6 +10,8 @@ from django import forms
from django.conf import settings from django.conf import settings
from django.core.validators import RegexValidator from django.core.validators import RegexValidator
from django.db import transaction from django.db import transaction
from mptt.forms import TreeNodeChoiceField
from mptt.models import MPTTModel
from ipam.formfields import IPFormField from ipam.formfields import IPFormField
from utilities.exceptions import AbortTransaction from utilities.exceptions import AbortTransaction
@ -124,6 +126,10 @@ class ObjectVar(ScriptVariable):
# Queryset for field choices # Queryset for field choices
self.field_attrs['queryset'] = queryset self.field_attrs['queryset'] = queryset
# Update form field for MPTT (nested) objects
if issubclass(queryset.model, MPTTModel):
self.form_field = TreeNodeChoiceField
class IPNetworkVar(ScriptVariable): class IPNetworkVar(ScriptVariable):
""" """