# Generated by Django 3.1 on 2020-09-07 07:54

from django.db import migrations

from etebase_server.django.models import AccessLevels


def change_access_level_to_int(apps, schema_editor):
    CollectionMember = apps.get_model("django_etebase", "CollectionMember")
    CollectionInvitation = apps.get_model("django_etebase", "CollectionInvitation")

    for member in CollectionMember.objects.all():
        if member.accessLevelOld == "adm":
            member.accessLevel = AccessLevels.ADMIN
        elif member.accessLevelOld == "rw":
            member.accessLevel = AccessLevels.READ_WRITE
        elif member.accessLevelOld == "ro":
            member.accessLevel = AccessLevels.READ_ONLY
        member.save()

    for invitation in CollectionInvitation.objects.all():
        if invitation.accessLevelOld == "adm":
            invitation.accessLevel = AccessLevels.ADMIN
        elif invitation.accessLevelOld == "rw":
            invitation.accessLevel = AccessLevels.READ_WRITE
        elif invitation.accessLevelOld == "ro":
            invitation.accessLevel = AccessLevels.READ_ONLY
        invitation.save()


class Migration(migrations.Migration):

    dependencies = [
        ("django_etebase", "0027_auto_20200907_0752"),
    ]

    operations = [
        migrations.RunPython(change_access_level_to_int),
    ]