Explorar o código

#6732 - Fix hiding of ASN field in Site creation form

Daniel Sheppard %!s(int64=4) %!d(string=hai) anos
pai
achega
0f68ecda78
Modificáronse 1 ficheiros con 9 adicións e 7 borrados
  1. 9 7
      netbox/dcim/forms/models.py

+ 9 - 7
netbox/dcim/forms/models.py

@@ -166,13 +166,15 @@ class SiteForm(BootstrapMixin, TenancyForm, CustomFieldModelForm):
     def __init__(self, instance, *args, **kwargs):
         super(SiteForm, self).__init__(instance=instance, *args, **kwargs)
         if instance is None or (instance and (instance.asn is None or instance.asn == '')):
-            site_fieldset = list(self.Meta.fieldsets[0][1])
-            site_fieldset.pop(6)
-            self.Meta.fieldsets = (
-                ('Site', tuple(site_fieldset)),
-                self.Meta.fieldsets[1],
-                self.Meta.fieldsets[2],
-            )
+            if 'asn' in self.Meta.fieldsets[0][1]:
+                site_fieldset = list(self.Meta.fieldsets[0][1])
+                index = site_fieldset.index('asn')
+                site_fieldset.pop(index)
+                self.Meta.fieldsets = (
+                    ('Site', tuple(site_fieldset)),
+                    self.Meta.fieldsets[1],
+                    self.Meta.fieldsets[2],
+                )
             del self.fields['asn']