Abhimanyu Saharan 2 лет назад
Родитель
Сommit
8571f428b1
1 измененных файлов с 8 добавлено и 0 удалено
  1. 8 0
      netbox/dcim/forms/bulk_import.py

+ 8 - 0
netbox/dcim/forms/bulk_import.py

@@ -159,6 +159,14 @@ class LocationImportForm(NetBoxModelImportForm):
         model = Location
         model = Location
         fields = ('site', 'parent', 'name', 'slug', 'status', 'tenant', 'description', 'tags')
         fields = ('site', 'parent', 'name', 'slug', 'status', 'tenant', 'description', 'tags')
 
 
+    def __init__(self, data=None, *args, **kwargs):
+        super().__init__(data, *args, **kwargs)
+
+        if data:
+            # Limit location queryset by assigned site
+            params = {f"site__{self.fields['site'].to_field_name}": data.get('site')}
+            self.fields['parent'].queryset = self.fields['parent'].queryset.filter(**params)
+
 
 
 class RackRoleImportForm(NetBoxModelImportForm):
 class RackRoleImportForm(NetBoxModelImportForm):
     slug = SlugField()
     slug = SlugField()