# -*- coding: utf-8 -*-
from __future__ import unicode_literals

from django.db import migrations, models

def transform_autor(apps, schema_editor):
    print
    Organizator = apps.get_model('seminar', 'Organizator')

    # preorgovani oprav
    Oprava = apps.get_model('korektury', 'Oprava')
    for oprava in Oprava.objects.all():
        jmeno = oprava.autor.split()
        if len(jmeno) == 2:
            try:
                org = Organizator.objects.get(user__first_name=jmeno[0],
                                                 user__last_name=jmeno[1])
                oprava.autor_org = org
                oprava.save()
            except:
                print("Org nenalezen -- mažu korekturu")
#                oprava.delete()
        else:
            print("Org nenalezen -- mažu korekturu")
            oprava.delete()

    # preorgovani komentaru
    Komentar = apps.get_model('korektury', 'Komentar')
    for komentar in Komentar.objects.all():
        jmeno = komentar.autor.split()
        if len(jmeno) == 2:
            try:
                org = Organizator.objects.get(user__first_name=jmeno[0],
                                                 user__last_name=jmeno[1])
                komentar.autor_org = org
                komentar.save()
            except:
                print("Org nenalezen -- mažu korekturu")
#                oprava.delete()
        else:
            print("Org nenalezen -- mažu korekturu")
            komentar.delete()

def back(apps, schema_editor):
    pass


class Migration(migrations.Migration):

    dependencies = [
        ('korektury', '0010_Pridani_odkazu_na_organizatora'),
    ]

    operations = [
            migrations.RunPython(transform_autor, back),
    ]