mirror/netbox-community-netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2024-05-10 07:54:54 +00:00
Code Activity

Clean up migration output

Browse Source
This commit is contained in:
Jeremy Stretch
2020-07-23 14:03:19 -04:00
parent 84d4b2db77
commit 78bb2e12fe
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
netbox/virtualization/migrations/0016_replicate_interfaces.py
View File

@ -18,7 +18,7 @@ def replicate_interfaces(apps, schema_editor):
# Replicate dcim.Interface instances assigned to VirtualMachines # Replicate dcim.Interface instances assigned to VirtualMachines
original_interfaces = Interface.objects.filter(virtual_machine__isnull=False) original_interfaces = Interface.objects.filter(virtual_machine__isnull=False)
if show_output: if show_output:
print(f"\n Replicating {len(original_interfaces)} VM interfaces...", flush=True) print(f"\n Replicating {len(original_interfaces)} VM interfaces...", end='', flush=True)
for i, interface in enumerate(original_interfaces, start=1): for i, interface in enumerate(original_interfaces, start=1):
vminterface = VMInterface( vminterface = VMInterface(
virtual_machine=interface.virtual_machine, virtual_machine=interface.virtual_machine,
@ -56,8 +56,6 @@ def replicate_interfaces(apps, schema_editor):
# Verify that all interfaces have been replicated # Verify that all interfaces have been replicated
replicated_count = VMInterface.objects.count() replicated_count = VMInterface.objects.count()
assert replicated_count == original_interfaces.count(), "Replicated interfaces count does not match original 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 # Delete all interfaces not assigned to a Device
Interface.objects.filter(device__isnull=True).delete() Interface.objects.filter(device__isnull=True).delete()