from django.db import migrations, models

NOITA_APP_ID = 881100
OPUS_MAGNUM_APP_ID = 558990

PATHS = {
    NOITA_APP_ID: "/noita",
    OPUS_MAGNUM_APP_ID: "/opus-magnum",
}


def set_paths(apps, schema_editor):
    Game = apps.get_model("games", "Game")
    for app_id, path in PATHS.items():
        Game.objects.filter(steam_app_id=app_id).update(path=path)


class Migration(migrations.Migration):
    dependencies = [
        ("games", "0002_seed_noita_and_opus_magnum"),
    ]

    operations = [
        migrations.AddField(
            model_name="game",
            name="path",
            field=models.CharField(default="", max_length=100),
            preserve_default=False,
        ),
        migrations.RunPython(set_paths, reverse_code=migrations.RunPython.noop),
    ]
