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 set initial counts for Device in migration

Browse Source
This commit is contained in:
Arthur 2023-06-26 14:16:01 -07:00
parent 72088adb5c
commit a1a926f69e
1 changed files with 19 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -4,6 +4,21 @@ from django.db import migrations
import utilities.fields
def recalculate_device_counts(apps, schema_editor):
Device = apps.get_model("dcim", "Device")
for device in Device.objects.all():
device._console_port_count = device.consoleports.count()
device._console_server_port_count = device.consoleserverports.count()
device._interface_count = device.interfaces.count()
device._front_port_count = device.frontports.count()
device._rear_port_count = device.rearports.count()
device._device_bay_count = device.devicebays.count()
device._inventory_item_count = device.inventoryitems.count()
device._power_port_count = device.powerports.count()
device._power_outlet_count = device.poweroutlets.count()
device.save()
class Migration(migrations.Migration):
dependencies = [
('dcim', '0174_rack_starting_unit'),
@ -55,4 +70,8 @@ class Migration(migrations.Migration):
name='_rear_port_count',
field=utilities.fields.CounterCacheField(default=0),
),
migrations.RunPython(
recalculate_device_counts,
reverse_code=migrations.RunPython.noop
),
]