"""This file and its contents are licensed under the Apache License 2.0. Please see the included NOTICE for copyright information and LICENSE for a copy of the license.
"""
from django.db import migrations


def remove(apps, schema_editor):
    db_alias = schema_editor.connection.alias
    View = apps.get_model('data_manager', 'View')
    views = View.objects.using(db_alias).all()

    for view in views:
        if 'hiddenColumns' in view.data:
            if 'explore' in view.data['hiddenColumns']:
                view.data['hiddenColumns']['explore'].append('tasks:predictions_model_versions')
                view.data['hiddenColumns']['explore'] = list(set(view.data['hiddenColumns']['explore']))
            if 'labeling' in view.data['hiddenColumns']:
                view.data['hiddenColumns']['labeling'].append('tasks:predictions_model_versions')
                view.data['hiddenColumns']['labeling'] = list(set(view.data['hiddenColumns']['labeling']))

        view.save(using=db_alias)


class Migration(migrations.Migration):
    dependencies = [
        ('data_manager', '0002_remove_annotations_ids'),
    ]

    operations = [
        migrations.RunPython(remove),
    ]