mirror/netbox-community-netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2024-05-10 07:54:54 +00:00
Code Activity

Use TreeNodeChoiceField for MPTT objects

Browse Source
This commit is contained in:
Jeremy Stretch
2019-08-14 14:03:11 -04:00
parent 7f65e009a8
commit 8bd1fad7d0
1 changed files with 6 additions and 0 deletions
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):
""" """