# -*- coding: utf-8 -*-
"""
Миграция данных: перенос значений unit -> base_unit и удаление поля unit.

Этапы:
1. Создать недостающие UnitOfMeasure из старых UNIT_CHOICES
2. Для товаров без base_unit установить соответствующую единицу из справочника
3. Удалить поле unit
"""
from django.db import migrations, models
import django.db.models.deletion


# Старые UNIT_CHOICES для миграции
OLD_UNIT_CHOICES = {
    'шт': {'code': 'шт', 'name': 'Штука', 'short_name': 'шт.', 'position': 1},
    'м': {'code': 'м', 'name': 'Метр', 'short_name': 'м.', 'position': 2},
    'г': {'code': 'г', 'name': 'Грамм', 'short_name': 'г.', 'position': 3},
    'л': {'code': 'л', 'name': 'Литр', 'short_name': 'л.', 'position': 4},
    'кг': {'code': 'кг', 'name': 'Килограмм', 'short_name': 'кг.', 'position': 5},
}


def migrate_unit_to_base_unit(apps, schema_editor):
    """Перенести значения unit -> base_unit для всех товаров"""
    UnitOfMeasure = apps.get_model('products', 'UnitOfMeasure')
    Product = apps.get_model('products', 'Product')

    # 1. Создать недостающие UnitOfMeasure
    for code, data in OLD_UNIT_CHOICES.items():
        UnitOfMeasure.objects.get_or_create(
            code=code,
            defaults={
                'name': data['name'],
                'short_name': data['short_name'],
                'position': data['position'],
                'is_active': True,
            }
        )

    # 2. Получить дефолтную единицу (штука)
    default_unit = UnitOfMeasure.objects.get(code='шт')

    # 3. Для товаров без base_unit - установить из unit или дефолтную
    products_without_base_unit = Product.objects.filter(base_unit__isnull=True)
    
    for product in products_without_base_unit:
        # Получаем значение старого поля unit
        old_unit_code = getattr(product, 'unit', 'шт') or 'шт'
        
        # Находим соответствующую единицу в справочнике
        try:
            unit = UnitOfMeasure.objects.get(code=old_unit_code)
        except UnitOfMeasure.DoesNotExist:
            unit = default_unit
        
        product.base_unit = unit
        product.save(update_fields=['base_unit'])

    print(f"Миграция завершена: обновлено {products_without_base_unit.count()} товаров")


def reverse_migration(apps, schema_editor):
    """Обратная миграция не нужна - поле unit удалено"""
    pass


class Migration(migrations.Migration):
    # Отключаем атомарность, чтобы избежать конфликта с триггерами PostgreSQL
    atomic = False

    dependencies = [
        ('products', '0001_initial'),
    ]

    operations = [
        # 1. Сначала делаем base_unit nullable для data migration
        # (если он уже nullable - это ничего не изменит)
        
        # 2. Запускаем data migration
        migrations.RunPython(migrate_unit_to_base_unit, reverse_migration),
        
        # 3. Удаляем старое поле unit
        migrations.RemoveField(
            model_name='product',
            name='unit',
        ),
        
        # 4. Делаем base_unit обязательным (NOT NULL)
        migrations.AlterField(
            model_name='product',
            name='base_unit',
            field=models.ForeignKey(
                on_delete=django.db.models.deletion.PROTECT,
                related_name='products',
                to='products.unitofmeasure',
                verbose_name='Базовая единица',
                help_text='Единица хранения и закупки (банч, кг, шт). Товар принимается и хранится в этих единицах.',
            ),
        ),
    ]
