remove common function for update

This commit is contained in:
Arthur
2025-12-12 09:20:04 -08:00
parent 451ea1dbfb
commit 7ad3e9daf3
3 changed files with 45 additions and 13 deletions

View File

@@ -33,7 +33,6 @@ from utilities.tracking import TrackingModelMixin
from .device_components import *
from .mixins import RenderConfigMixin
from .modules import Module
from ..utils import update_device_components
__all__ = (
@@ -958,6 +957,12 @@ class Device(
if cf_defaults := CustomField.objects.get_defaults_for_model(model):
for component in components:
component.custom_field_data = cf_defaults
# Set denormalized references (_site, _location, _rack) before bulk_create
# since bulk_create bypasses the save() method
for component in components:
component._site = self.site
component._location = self.location
component._rack = self.rack
components = model.objects.bulk_create(components)
# Prefetch related objects to minimize queries needed during post_save
prefetch_fields = get_prefetchable_fields(model)
@@ -1013,8 +1018,6 @@ class Device(
self._instantiate_components(self.device_type.inventoryitemtemplates.all(), bulk_create=False)
# Interface bridges have to be set after interface instantiation
update_interface_bridges(self, self.device_type.interfacetemplates.all())
# Update denormalized fields for all components
update_device_components(self)
# Update Site and Rack assignment for any child Devices
devices = Device.objects.filter(parent_bay__device=self)

View File

@@ -15,7 +15,6 @@ from netbox.models.mixins import WeightMixin
from utilities.jsonschema import validate_schema
from utilities.string import title
from .device_components import *
from ..utils import update_device_components
__all__ = (
'Module',
@@ -316,6 +315,13 @@ class Module(PrimaryModel, ConfigContextModel):
for component in create_instances:
component.custom_field_data = cf_defaults
# Set denormalized references (_site, _location, _rack) before bulk_create
# since bulk_create bypasses the save() method
for component in create_instances:
component._site = self.device.site
component._location = self.device.location
component._rack = self.device.rack
if component_model is not ModuleBay:
component_model.objects.bulk_create(create_instances)
# Emit the post_save signal for each newly created object
@@ -348,5 +354,3 @@ class Module(PrimaryModel, ConfigContextModel):
# Interface bridges have to be set after interface instantiation
update_interface_bridges(self.device, self.module_type.interfacetemplates, self)
# Update denormalized fields for all components
update_device_components(self.device)

View File

@@ -6,11 +6,25 @@ from django.dispatch import receiver
from dcim.choices import CableEndChoices, LinkStatusChoices
from virtualization.models import VMInterface
from .models import (
Cable, CablePath, CableTermination, Device, FrontPort, Interface, PathEndpoint, PowerPanel, Rack, Location,
Cable, CablePath, CableTermination, ConsolePort, ConsoleServerPort, Device, DeviceBay, FrontPort, Interface,
InventoryItem, ModuleBay, PathEndpoint, PowerOutlet, PowerPanel, PowerPort, Rack, RearPort, Location,
VirtualChassis,
)
from .models.cables import trace_paths
from .utils import create_cablepath, rebuild_paths, update_device_components
from .utils import create_cablepath, rebuild_paths
COMPONENT_MODELS = (
ConsolePort,
ConsoleServerPort,
DeviceBay,
FrontPort,
Interface,
InventoryItem,
ModuleBay,
PowerOutlet,
PowerPort,
RearPort,
)
#
@@ -31,8 +45,9 @@ def handle_location_site_change(instance, created, **kwargs):
PowerPanel.objects.filter(location__in=locations).update(site=instance.site)
CableTermination.objects.filter(_location__in=locations).update(_site=instance.site)
# Update component models for devices in these locations
for device in Device.objects.filter(location__in=locations):
update_device_components(device)
# (since Device.objects.update() doesn't trigger post_save signals)
for model in COMPONENT_MODELS:
model.objects.filter(device__location__in=locations).update(_site=instance.site)
@receiver(post_save, sender=Rack)
@@ -43,8 +58,13 @@ def handle_rack_site_change(instance, created, **kwargs):
if not created:
Device.objects.filter(rack=instance).update(site=instance.site, location=instance.location)
# Update component models for devices in this rack
for device in Device.objects.filter(rack=instance):
update_device_components(device)
# (since Device.objects.update() doesn't trigger post_save signals)
for model in COMPONENT_MODELS:
model.objects.filter(device__rack=instance).update(
_site=instance.site,
_location=instance.location,
_rack=instance,
)
@receiver(post_save, sender=Device)
@@ -53,7 +73,12 @@ def handle_device_site_change(instance, created, **kwargs):
Update child components to update the parent Site, Location, and Rack when a Device is saved.
"""
if not created:
update_device_components(instance)
for model in COMPONENT_MODELS:
model.objects.filter(device=instance).update(
_site=instance.site,
_location=instance.location,
_rack=instance.rack,
)
#