from django.db import migrations, models
import django.db.models.deletion


def vincular_horario(apps, schema_editor):
    HorarioLaboral = apps.get_model('ventas', 'HorarioLaboral')
    DiaLaboral = apps.get_model('ventas', 'DiaLaboral')
    horario = HorarioLaboral.objects.first()
    if not horario:
        horario = HorarioLaboral.objects.create()
    DiaLaboral.objects.filter(horario__isnull=True).update(horario=horario)


class Migration(migrations.Migration):

    dependencies = [
        ('ventas', '0014_dialaboral_remove_horariolaboral_domingo_and_more'),
    ]

    operations = [
        migrations.AddField(
            model_name='dialaboral',
            name='horario',
            field=models.ForeignKey(
                null=True, on_delete=django.db.models.deletion.CASCADE,
                related_name='dias', to='ventas.horariolaboral',
            ),
        ),
        migrations.RunPython(vincular_horario, reverse_code=migrations.RunPython.noop),
        migrations.AlterField(
            model_name='dialaboral',
            name='horario',
            field=models.ForeignKey(
                on_delete=django.db.models.deletion.CASCADE,
                related_name='dias', to='ventas.horariolaboral',
            ),
        ),
        migrations.AlterField(
            model_name='dialaboral',
            name='dia',
            field=models.PositiveSmallIntegerField(choices=[
                (0, 'Lunes'), (1, 'Martes'), (2, 'Miércoles'), (3, 'Jueves'),
                (4, 'Viernes'), (5, 'Sábado'), (6, 'Domingo'),
            ]),
        ),
        migrations.AlterUniqueTogether(
            name='dialaboral',
            unique_together={('horario', 'dia')},
        ),
    ]