# Generated by Django 5.2.7 on 2025-11-21 07:35

from django.db import migrations


def set_default_source_type(apps, schema_editor):
    """
    Устанавливает тип "Стационарные" для всех Source, у которых не указан тип.
    """
    Source = apps.get_model('mainapp', 'Source')
    ObjectInfo = apps.get_model('mainapp', 'ObjectInfo')
    
    # Создаем или получаем тип "Стационарные"
    stationary_info, _ = ObjectInfo.objects.get_or_create(name="Стационарные")
    
    # Обновляем все Source без типа
    sources_without_type = Source.objects.filter(info__isnull=True)
    count = sources_without_type.update(info=stationary_info)
    
    print(f"Обновлено {count} источников с типом 'Стационарные'")


def reverse_set_default_source_type(apps, schema_editor):
    """
    Обратная миграция - ничего не делаем, так как это безопасная операция.
    """
    pass


class Migration(migrations.Migration):

    dependencies = [
        ('mainapp', '0009_objectownership_alter_source_info_source_ownership'),
    ]

    operations = [
        migrations.RunPython(set_default_source_type, reverse_set_default_source_type),
    ]
