src/resource_inventory/migrations/0012_auto_20200103_1850.py

   1 # Generated by Django 2.2 on 2020-01-03 18:50
   2
   3 from django.db import migrations, models
   4 import django.db.models.deletion
   5
   6
   7 def pairVlanPhysicalNetworks(apps, editor):
   8     PhysicalNetwork = apps.get_model("resource_inventory", "PhysicalNetwork")
   9     Vlan = apps.get_model("resource_inventory", "Vlan")
  10     for vlan in Vlan.objects.filter(network__isnull=False):
  11         if PhysicalNetwork.objects.filter(id=vlan.network.id).exists():
  12             continue
  13         PhysicalNetwork.objects.create(id=vlan.network.id, vlan_id=vlan.vlan_id, generic_network=vlan.network)
  14
  15
  16 def deletePhysicalNetworks(apps, editor):
  17     Vlan = apps.get_model("resource_inventory", "Vlan")
  18     for vlan in Vlan.objects.all():
  19         vlan.network = None
  20     PhysicalNetwork = apps.get_model("resource_inventory", "PhysicalNetwork")
  21     PhysicalNetwork.objects.all().delete()
  22
  23
  24 class Migration(migrations.Migration):
  25
  26     dependencies = [
  27         ('resource_inventory', '0011_auto_20191106_2024'),
  28     ]
  29
  30     operations = [
  31         migrations.CreateModel(
  32             name='PhysicalNetwork',
  33             fields=[
  34                 ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
  35                 ('vlan_id', models.IntegerField()),
  36                 ('generic_network', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='resource_inventory.Network')),
  37             ],
  38             options={
  39                 'abstract': False,
  40             },
  41         ),
  42         migrations.AlterField(
  43             model_name='host',
  44             name='id',
  45             field=models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID'),
  46         ),
  47         migrations.AlterField(
  48             model_name='resourcebundle',
  49             name='id',
  50             field=models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID'),
  51         ),
  52         migrations.RunPython(pairVlanPhysicalNetworks, deletePhysicalNetworks),
  53         migrations.AlterField(
  54             model_name='vlan',
  55             name='network',
  56             field=models.ForeignKey(on_delete=django.db.models.deletion.DO_NOTHING,
  57                                     to='resource_inventory.PhysicalNetwork', null=True),
  58         ),
  59     ]