# Generated by Django 5.1.4 on 2026-02-21 05:06

from django.db import migrations


def deduplicate_workout_equipment(apps, schema_editor):
    """Remove duplicate WorkoutEquipment rows before adding unique constraint."""
    WorkoutEquipment = apps.get_model('equipment', 'WorkoutEquipment')
    seen = set()
    to_delete = []
    for we in WorkoutEquipment.objects.all().order_by('id'):
        key = (we.exercise_id, we.equipment_id)
        if key in seen:
            to_delete.append(we.id)
        else:
            seen.add(key)
    if to_delete:
        WorkoutEquipment.objects.filter(id__in=to_delete).delete()


class Migration(migrations.Migration):

    dependencies = [
        ('equipment', '0002_workoutequipment'),
        ('exercise', '0010_alter_exercise_complexity_rating_and_more'),
    ]

    operations = [
        migrations.RunPython(deduplicate_workout_equipment, migrations.RunPython.noop),
        migrations.AlterUniqueTogether(
            name='workoutequipment',
            unique_together={('exercise', 'equipment')},
        ),
    ]