NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-08-26 01:06:11 -06:00
Code Issues Actions 7 Packages Projects Releases Wiki Activity

Refactor logic for finding initial component assignment on InventoryItems

Browse Source
This commit is contained in:
kkthxbye-code 2022-12-20 11:46:09 +01:00
parent 93eecfcd80
commit 9cac9f10e3
1 changed files with 10 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
netbox/dcim/forms/model_forms.py
View File

@ -1623,30 +1623,23 @@ class InventoryItemForm(DeviceComponentForm):
initial = kwargs.get('initial', {}).copy()
component_type = initial.get('component_type')
component_id = initial.get('component_id')
# Used for picking the default active tab for component selection
self.no_component = True
if instance:
if type(instance.component) is ConsolePort:
initial['consoleport'] = instance.component
elif type(instance.component) is ConsoleServerPort:
initial['consoleserverport'] = instance.component
elif type(instance.component) is FrontPort:
initial['frontport'] = instance.component
elif type(instance.component) is Interface:
initial['interface'] = instance.component
elif type(instance.component) is PowerPort:
initial['powerport'] = instance.component
elif type(instance.component) is RearPort:
initial['rearport'] = instance.component
else:
self.no_component = True
# When editing set the initial value for component selectin
for component_model in ContentType.objects.filter(MODULAR_COMPONENT_MODELS):
if type(instance.component) is component_model.model_class():
initial[component_model.model] = instance.component
self.no_component = False
break
elif component_type and component_id:
self.no_component = True
# When adding the InventoryItem from a component page
if content_type := ContentType.objects.filter(MODULAR_COMPONENT_MODELS).filter(pk=component_type).first():
if component := content_type.model_class().objects.filter(pk=component_id).first():
initial[content_type.model] = component
self.no_component = False
else:
self.no_component = True
kwargs['initial'] = initial