Files
python-tdd/src/apps/lyric/migrations/0016_backfill_unlocked_decks.py

25 lines
695 B
Python
Raw Normal View History

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),
]