from django.db import migrations


MENU_SUB_FEATURES = [
    "USER_DASHBOARD",

    "ADD_CUSTOMER",
    "CUSTOMER_DETAILS",
    "CUSTOMER_DEVICE_DETAILS",

    "RESULT_LIST",

    "ADD_PRODUCT",
    "PRODUCT_LIST",
    "ADD_CATELOG_TRANSLATION",
    "CATELOG_TRANSLATION_LIST",
    "CATELOG_TRANSLATION_MATRIX",
    "ADD_CROPMAPPING",
    "CROPMAPPING_DETAILS",

    "ADD_CROP",
    "CROP_DETAILS",
    "ADD_CROP_TRANSLATION",
    "CROP_TRANSLATION_DETAILS",
    "CROP_TRANSLATION_DETAILS_MATRIX",

    "MANAGEAPPVERSION",
    "APPSETTINGS",

    "LANGUAGE_LIST",

    "ADD_SCALE",
    "ADD_SCALECROPMAPPING",
    "SCALECROPMAPPING_DETAILS",
    "BULK_UPLOAD",

    "MANAGE_PERMISSIONS",

    "ADD_USER",
    "VIEW_MY_PROFILE",
    "USER_DETAILS",
]


def fix_feature_types(apps, schema_editor):
    SubFeature = apps.get_model("usermanagement", "SubFeature")

    for name in MENU_SUB_FEATURES:
        SubFeature.objects.filter(
            feature_name__iexact=name
        ).update(feature_type="MODULE")


class Migration(migrations.Migration):

    dependencies = [
        ("usermanagement", "0023_seed_wile_permission_features"),
    ]

    operations = [
        migrations.RunPython(fix_feature_types),
    ]