# Generated by Django 4.2.30 on 2026-10-20 13:21

from django.db import migrations, models
from django.utils.text import slugify


def populate_kunyas_from_existing_records(apps, schema_editor):
    Kunya = apps.get_model('hadis', 'Kunya')
    Transmitters = apps.get_model('hadis', 'Transmitters')

    # Cache kunyas by normalized text key -> Kunya instance
    kunya_by_text = {}
    for k in Kunya.objects.all():
        if isinstance(k.title, list):
            for entry in k.title:
                if isinstance(entry, dict):
                    txt = (entry.get('text') or entry.get('title') or '').strip().lower()
                    if txt:
                        kunya_by_text[txt] = k

    def get_or_create_kunya(json_list):
        if not json_list or not isinstance(json_list, list):
            return None
        cleaned = []
        lookup_keys = []
        for entry in json_list:
            if isinstance(entry, dict):
                lang = (entry.get('language_code') or '').strip()
                txt = (entry.get('text') or entry.get('title') or '').strip()
                if lang and txt:
                    cleaned.append({'language_code': lang, 'text': txt, 'title': txt})
                    lookup_keys.append(txt.lower())
        if not cleaned:
            return None

        for key in lookup_keys:
            if key in kunya_by_text:
                existing = kunya_by_text[key]
                # Merge missing languages into existing kunya
                existing_langs = {
                    item.get('language_code'): item
                    for item in (existing.title or [])
                    if isinstance(item, dict) and item.get('language_code')
                }
                updated = False
                for item in cleaned:
                    if item['language_code'] not in existing_langs:
                        existing_langs[item['language_code']] = item
                        updated = True
                if updated:
                    existing.title = list(existing_langs.values())
                    existing.save(update_fields=['title'])
                for k in lookup_keys:
                    kunya_by_text[k] = existing
                return existing

        first_text = cleaned[0]['text']
        slug_val = slugify(first_text, allow_unicode=True)[:240] or 'kunya'
        kunya_obj = Kunya.objects.create(title=cleaned, slug=slug_val)
        for k in lookup_keys:
            kunya_by_text[k] = kunya_obj
        return kunya_obj

    for tr in Transmitters.objects.all().iterator():
        if tr.kunya and isinstance(tr.kunya, list):
            k_obj = get_or_create_kunya(tr.kunya)
            if k_obj:
                tr.kunyas.add(k_obj)


class Migration(migrations.Migration):

    dependencies = [
        ('hadis', '0058_rename_hadis_booka_date_ty_9a1e3c_idx_hadis_booka_date_ty_99a484_idx_and_more'),
    ]

    operations = [
        migrations.CreateModel(
            name='Kunya',
            fields=[
                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
                ('title', models.JSONField(default=list, verbose_name='Title')),
                ('slug', models.SlugField(allow_unicode=True, blank=True, max_length=255, null=True, verbose_name='slug')),
                ('created_at', models.DateTimeField(auto_now_add=True, verbose_name='created at')),
                ('updated_at', models.DateTimeField(auto_now=True, verbose_name='updated at')),
            ],
            options={
                'verbose_name': 'Kunya',
                'verbose_name_plural': 'Kunyas',
                'ordering': ('-created_at', '-id'),
            },
        ),
        migrations.AddField(
            model_name='transmitters',
            name='kunyas',
            field=models.ManyToManyField(blank=True, related_name='transmitters', to='hadis.kunya', verbose_name='Kunyas'),
        ),
        migrations.RunPython(
            populate_kunyas_from_existing_records,
            reverse_code=migrations.RunPython.noop,
        ),
    ]