re2o/topologie/migrations/0052_transferports.py

# -*- coding: utf-8 -*-
# Generated by Django 1.10.7 on 2017-12-31 19:53
from __future__ import unicode_literals

from django.db import migrations


class Migration(migrations.Migration):

    dependencies = [("topologie", "0051_switchs_machine")]

    def transfer_port(apps, schema_editor):
        db_alias = schema_editor.connection.alias
        port = apps.get_model("topologie", "Port")
        switch = apps.get_model("topologie", "NewSw")
        port_list = port.objects.using(db_alias).all()
        for p in port_list:
            p.new_sw = switch.objects.filter(machine_ptr=p.switch.machine).first()
            p.save()

    def untransfer_port(apps, schema_editor):
        return

    operations = [
        migrations.AlterUniqueTogether(name="port", unique_together=set([])),
        migrations.RunPython(transfer_port, untransfer_port),
        migrations.RemoveField(model_name="port", name="switch"),
        migrations.RenameField("Port", "new_sw", "switch"),
    ]