NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-07-19 17:59:11 -06:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Fixed content type assignment within migration

Browse Source
This commit is contained in:
Jeremy Stretch 2018-10-19 10:38:15 -04:00
parent ea5121ffe1
commit d908dffab7
1 changed files with 11 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
netbox/dcim/migrations/0066_cables.py
View File

@ -9,13 +9,14 @@ def console_connections_to_cables(apps, schema_editor):
"""
Copy all existing console connections as Cables
"""
ContentType = apps.get_model('contenttypes', 'ContentType')
ConsolePort = apps.get_model('dcim', 'ConsolePort')
ConsoleServerPort = apps.get_model('dcim', 'ConsoleServerPort')
Cable = apps.get_model('dcim', 'Cable')
# Load content types
ContentType = apps.get_model('contenttypes', 'ContentType')
consoleport_type = ContentType.objects.get(app_label='dcim', model='consoleport')
consoleserverport_type = ContentType.objects.get(app_label='dcim', model='consoleserverport')
consoleport_type = ContentType.objects.get_for_model(ConsolePort)
consoleserverport_type = ContentType.objects.get_for_model(ConsoleServerPort)
# Create a new Cable instance from each console connection
for consoleport in ConsolePort.objects.filter(cs_port__isnull=False):
@ -33,13 +34,14 @@ def power_connections_to_cables(apps, schema_editor):
"""
Copy all existing power connections as Cables
"""
ContentType = apps.get_model('contenttypes', 'ContentType')
PowerPort = apps.get_model('dcim', 'PowerPort')
PowerOutlet = apps.get_model('dcim', 'PowerOutlet')
Cable = apps.get_model('dcim', 'Cable')
# Load content types
ContentType = apps.get_model('contenttypes', 'ContentType')
powerport_type = ContentType.objects.get(app_label='dcim', model='powerport')
poweroutlet_type = ContentType.objects.get(app_label='dcim', model='poweroutlet')
powerport_type = ContentType.objects.get_for_model(PowerPort)
poweroutlet_type = ContentType.objects.get_for_model(PowerOutlet)
# Create a new Cable instance from each power connection
for powerport in PowerPort.objects.filter(power_outlet__isnull=False):
@ -57,12 +59,13 @@ def interface_connections_to_cables(apps, schema_editor):
"""
Copy all InterfaceConnections as Cables
"""
ContentType = apps.get_model('contenttypes', 'ContentType')
Interface = apps.get_model('dcim', 'Interface')
InterfaceConnection = apps.get_model('dcim', 'InterfaceConnection')
Cable = apps.get_model('dcim', 'Cable')
# Load content types
ContentType = apps.get_model('contenttypes', 'ContentType')
interface_type = ContentType.objects.get(app_label='dcim', model='interface')
interface_type = ContentType.objects.get_for_model(Interface)
# Create a new Cable instance from each InterfaceConnection
for conn in InterfaceConnection.objects.all():