offline-db/users/migrations/0011_seed_marathoner.py

# Generated by Django 5.1.6

from django.db import migrations

def create_marathoner_badge(apps, schema_editor):
    Badge = apps.get_model('users', 'Badge')
    badge_data = {
        'slug': 'marathoner',
        'name': 'Marathoner',
        'description': 'Watched 50 episodes in total',
        'icon_url': '/static/badges/marathoner.png'
    }
    if not Badge.objects.filter(slug=badge_data['slug']).exists():
        Badge.objects.create(**badge_data)

def remove_marathoner_badge(apps, schema_editor):
    Badge = apps.get_model('users', 'Badge')
    Badge.objects.filter(slug='marathoner').delete()

class Migration(migrations.Migration):

    dependencies = [
        ('users', '0010_seed_season_completist'),
    ]

    operations = [
        migrations.RunPython(create_marathoner_badge, remove_marathoner_badge),
    ]