NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-08-25 08:46:10 -06:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

6347 optimize migrations

Browse Source
This commit is contained in:
Arthur 2023-07-19 11:55:34 +07:00
parent 34ca646fba
commit b0aa8c15c8
2 changed files with 19 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
netbox/dcim/migrations/0175_device__console_port_count_and_more.py
View File

@ -8,27 +8,27 @@ from django.db.models import Count
def recalculate_device_counts(apps, schema_editor): def recalculate_device_counts(apps, schema_editor):
Device = apps.get_model("dcim", "Device") Device = apps.get_model("dcim", "Device")
devices = list(Device.objects.all().annotate( devices = list(Device.objects.all().annotate(
console_port_count=Count('consoleports'), console_port_count=Count('consoleports', distinct=True),
console_server_port_count=Count('consoleserverports'), console_server_port_count=Count('consoleserverports', distinct=True),
interface_count=Count('interfaces'), interface_count=Count('interfaces', distinct=True),
front_port_count=Count('frontports'), front_port_count=Count('frontports', distinct=True),
rear_port_count=Count('rearports'), rear_port_count=Count('rearports', distinct=True),
device_bay_count=Count('devicebays'), device_bay_count=Count('devicebays', distinct=True),
inventory_item_count=Count('inventoryitems'), inventory_item_count=Count('inventoryitems', distinct=True),
power_port_count=Count('powerports'), power_port_count=Count('powerports', distinct=True),
power_outlet_count=Count('poweroutlets'), power_outlet_count=Count('poweroutlets', distinct=True),
)) ))
for device in devices: for device in devices:
device._console_port_count = console_port_count device._console_port_count = device.console_port_count
device._console_server_port_count = console_server_port_count device._console_server_port_count = device.console_server_port_count
device._interface_count = interface_count device._interface_count = device.interface_count
device._front_port_count = front_port_count device._front_port_count = device.front_port_count
device._rear_port_count = rear_port_count device._rear_port_count = device.rear_port_count
device._device_bay_count = device_bay_count device._device_bay_count = device.device_bay_count
device._inventory_item_count = inventory_item_count device._inventory_item_count = device.inventory_item_count
device._power_port_count = power_port_count device._power_port_count = device.power_port_count
device._power_outlet_count = power_outlet_count device._power_outlet_count = device.dpower_outlet_count
Device.objects.bulk_update( Device.objects.bulk_update(
devices, devices,

2
netbox/virtualization/migrations/0035_virtualmachine__interface_count.py
View File

@ -8,7 +8,7 @@ from django.db.models import Count
def populate_virtualmachine_counts(apps, schema_editor): def populate_virtualmachine_counts(apps, schema_editor):
VirtualMachine = apps.get_model('virtualization', 'VirtualMachine') VirtualMachine = apps.get_model('virtualization', 'VirtualMachine')
vms = list(VirtualMachine.objects.annotate(interface_count=Count('interfaces'))) vms = list(VirtualMachine.objects.annotate(interface_count=Count('interfaces', distinct=True)))
for vm in vms: for vm in vms:
vm._interface_count = vm.interface_count vm._interface_count = vm.interface_count