Bin
2025-12-17 dcf780a91c16b6be28635b6e2e0e702060ee19f2
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
"""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):
    View = apps.get_model('data_manager', 'View')
    views = View.objects.all()
 
    for view in views:
        if 'hiddenColumns' in view.data:
            if 'explore' in view.data['hiddenColumns']:
                view.data['hiddenColumns']['explore'].append('tasks:inner_id')
                view.data['hiddenColumns']['explore'] = list(set(view.data['hiddenColumns']['explore']))
            if 'labeling' in view.data['hiddenColumns']:
                view.data['hiddenColumns']['labeling'].append('tasks:inner_id')
                view.data['hiddenColumns']['labeling'] = list(set(view.data['hiddenColumns']['labeling']))
 
        view.save()
 
 
def backwards(apps, schema_editor):
    View = apps.get_model('data_manager', 'View')
    views = View.objects.all()
 
    for view in views:
        if 'hiddenColumns' in view.data:
            if 'explore' in view.data['hiddenColumns']:
                view.data['hiddenColumns']['explore'].remove('tasks:inner_id')
                view.data['hiddenColumns']['explore'] = list(set(view.data['hiddenColumns']['explore']))
            if 'labeling' in view.data['hiddenColumns']:
                view.data['hiddenColumns']['labeling'].remove('tasks:inner_id')
                view.data['hiddenColumns']['labeling'] = list(set(view.data['hiddenColumns']['labeling']))
 
        view.save()
 
 
class Migration(migrations.Migration):
    dependencies = [
        ('data_manager', '0005_remove_updated_by'),
    ]
 
    operations = [
        migrations.RunPython(remove, backwards),
    ]