mirror of
https://github.com/netbox-community/netbox.git
synced 2024-05-10 07:54:54 +00:00
78 lines
2.7 KiB
Python
78 lines
2.7 KiB
Python
import sys
|
|
|
|
from django.db import migrations
|
|
|
|
|
|
def replicate_interfaces(apps, schema_editor):
|
|
show_output = 'test' not in sys.argv
|
|
|
|
ContentType = apps.get_model('contenttypes', 'ContentType')
|
|
TaggedItem = apps.get_model('extras', 'TaggedItem')
|
|
Interface = apps.get_model('dcim', 'Interface')
|
|
IPAddress = apps.get_model('ipam', 'IPAddress')
|
|
VMInterface = apps.get_model('virtualization', 'VMInterface')
|
|
|
|
interface_ct = ContentType.objects.get_for_model(Interface)
|
|
vminterface_ct = ContentType.objects.get_for_model(VMInterface)
|
|
|
|
# Replicate dcim.Interface instances assigned to VirtualMachines
|
|
original_interfaces = Interface.objects.filter(virtual_machine__isnull=False)
|
|
if show_output:
|
|
print(f"\n Replicating {len(original_interfaces)} VM interfaces...", flush=True)
|
|
for i, interface in enumerate(original_interfaces, start=1):
|
|
vminterface = VMInterface(
|
|
virtual_machine=interface.virtual_machine,
|
|
name=interface.name,
|
|
enabled=interface.enabled,
|
|
mac_address=interface.mac_address,
|
|
mtu=interface.mtu,
|
|
mode=interface.mode,
|
|
description=interface.description,
|
|
untagged_vlan=interface.untagged_vlan,
|
|
)
|
|
vminterface.save()
|
|
|
|
# Copy tagged VLANs
|
|
vminterface.tagged_vlans.set(interface.tagged_vlans.all())
|
|
|
|
# Reassign tags to the new instance
|
|
TaggedItem.objects.filter(
|
|
content_type=interface_ct, object_id=interface.pk
|
|
).update(
|
|
content_type=vminterface_ct, object_id=vminterface.pk
|
|
)
|
|
|
|
# Update any assigned IPAddresses
|
|
IPAddress.objects.filter(assigned_object_id=interface.pk).update(
|
|
assigned_object_type=vminterface_ct,
|
|
assigned_object_id=vminterface.pk
|
|
)
|
|
|
|
# Progress counter
|
|
if show_output and not i % 250:
|
|
percentage = int(i / len(original_interfaces) * 100)
|
|
print(f" {i}/{len(original_interfaces)} ({percentage}%)", flush=True)
|
|
|
|
# Verify that all interfaces have been replicated
|
|
replicated_count = VMInterface.objects.count()
|
|
assert replicated_count == original_interfaces.count(), "Replicated interfaces count does not match original count!"
|
|
if show_output:
|
|
print(f"\n Replicated {replicated_count} interfaces ", end='', flush=True)
|
|
|
|
# Delete all interfaces not assigned to a Device
|
|
Interface.objects.filter(device__isnull=True).delete()
|
|
|
|
|
|
class Migration(migrations.Migration):
|
|
|
|
dependencies = [
|
|
('ipam', '0037_ipaddress_assignment'),
|
|
('virtualization', '0015_vminterface'),
|
|
]
|
|
|
|
operations = [
|
|
migrations.RunPython(
|
|
code=replicate_interfaces
|
|
),
|
|
]
|