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')
    ]