mirror of
https://github.com/netbox-community/netbox.git
synced 2024-05-10 07:54:54 +00:00
36 lines
1.0 KiB
Python
36 lines
1.0 KiB
Python
from django.db import migrations
|
|
from django.db.models import Count
|
|
|
|
import utilities.fields
|
|
|
|
|
|
def populate_virtualmachine_counts(apps, schema_editor):
|
|
VirtualMachine = apps.get_model('virtualization', 'VirtualMachine')
|
|
|
|
vms = VirtualMachine.objects.annotate(_interface_count=Count('interfaces', distinct=True))
|
|
|
|
for vm in vms:
|
|
vm.interface_count = vm._interface_count
|
|
|
|
VirtualMachine.objects.bulk_update(vms, ['interface_count'], batch_size=100)
|
|
|
|
|
|
class Migration(migrations.Migration):
|
|
dependencies = [
|
|
('virtualization', '0034_standardize_description_comments'),
|
|
]
|
|
|
|
operations = [
|
|
migrations.AddField(
|
|
model_name='virtualmachine',
|
|
name='interface_count',
|
|
field=utilities.fields.CounterCacheField(
|
|
default=0, to_field='virtual_machine', to_model='virtualization.VMInterface'
|
|
),
|
|
),
|
|
migrations.RunPython(
|
|
code=populate_virtualmachine_counts,
|
|
reverse_code=migrations.RunPython.noop
|
|
),
|
|
]
|