from django.db import migrations


def backfill_unlocked_decks(apps, schema_editor):
    User = apps.get_model("lyric", "User")
    DeckVariant = apps.get_model("epic", "DeckVariant")
    try:
        earthman = DeckVariant.objects.get(slug="earthman")
    except DeckVariant.DoesNotExist:
        return
    for user in User.objects.filter(unlocked_decks__isnull=True):
        user.unlocked_decks.add(earthman)


class Migration(migrations.Migration):

    dependencies = [
        ("lyric", "0015_user_unlocked_decks"),
        ("epic", "0010_seed_deck_variants_and_earthman"),
    ]

    operations = [
        migrations.RunPython(backfill_unlocked_decks, migrations.RunPython.noop),
    ]