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

from django.db import migrations


def copy_tower_settings(apps, schema_editor):
    TowerSettings = apps.get_model('main', 'TowerSettings')
    Setting = apps.get_model('conf', 'Setting')
    for tower_setting in TowerSettings.objects.all().iterator():
        try:
            value = tower_setting.value
            # LICENSE is stored as a string; convert it to a dict.
            if tower_setting.key == 'LICENSE':
                value = json.loads(value)
            setting, created = Setting.objects.get_or_create(
                key=tower_setting.key,
                user=tower_setting.user,
                created=tower_setting.created,
                modified=tower_setting.modified,
                defaults=dict(value=value),
            )
            if not created and setting.value != value:
                setting.value = value
                setting.save(update_fields=['value'])
        except Setting.MultipleObjectsReturned:
            pass


def revert_tower_settings(apps, schema_editor):
    TowerSettings = apps.get_model('main', 'TowerSettings')
    Setting = apps.get_model('conf', 'Setting')
    for setting in Setting.objects.all().iterator():
        value = setting.value
        # LICENSE is stored as a JSON object; convert it back to a string.
        if setting.key == 'LICENSE':
            value = json.dumps(value)
        defaults = dict(
            value=value,
            value_type='string',
            description='',
            category='',
        )
        try:
            tower_setting, created = TowerSettings.objects.get_or_create(
                key=setting.key,
                user=setting.user,
                defaults=defaults,
            )
            if not created:
                update_fields = []
                for k, v in defaults.items():
                    if getattr(tower_setting, k) != v:
                        setattr(tower_setting, k, v)
                        update_fields.append(k)
                if update_fields:
                    tower_setting.save(update_fields=update_fields)
        except TowerSettings.MultipleObjectsReturned:
            pass


class Migration(migrations.Migration):

    dependencies = [
        ('conf', '0001_initial'),
        ('main', '0004_squashed_v310_release'),
    ]

    run_before = [
        ('main', '0005_squashed_v310_v313_updates'),
    ]

    operations = [
        migrations.RunPython(copy_tower_settings, revert_tower_settings),
    ]