from django.db import migrations, models


def seed_billboard_applets(apps, schema_editor):
    Applet = apps.get_model("applets", "Applet")
    for slug, name, cols, rows in [
        ("billboard-my-scrolls",  "My Scrolls", 4, 3),
        ("billboard-my-contacts", "Contacts",   4, 3),
        ("billboard-most-recent", "Most Recent", 8, 6),
    ]:
        Applet.objects.get_or_create(
            slug=slug,
            defaults={"name": name, "grid_cols": cols, "grid_rows": rows, "context": "billboard"},
        )


def remove_billboard_applets(apps, schema_editor):
    Applet = apps.get_model("applets", "Applet")
    Applet.objects.filter(slug__in=[
        "billboard-my-scrolls",
        "billboard-my-contacts",
        "billboard-most-recent",
    ]).delete()


class Migration(migrations.Migration):

    dependencies = [
        ("applets", "0005_gameboard_applet_heights"),
    ]

    operations = [
        migrations.AlterField(
            model_name="applet",
            name="context",
            field=models.CharField(
                choices=[
                    ("dashboard", "Dashboard"),
                    ("gameboard", "Gameboard"),
                    ("wallet", "Wallet"),
                    ("billboard", "Billboard"),
                ],
                default="dashboard",
                max_length=20,
            ),
        ),
        migrations.RunPython(seed_billboard_applets, remove_billboard_applets),
    ]