from regulations.models import Angka, Huruf, Pasal, Bab, Bagian, Paragraf

def update_uu_relations():
    # Mengaitkan Angka ke UU berdasarkan Pasal
    for angka in Angka.objects.all():
        if not angka.uu and angka.pasal:
            angka.uu = angka.pasal.uu
            angka.save()

    # Mengaitkan Huruf ke UU berdasarkan Angka, Pasal, atau Ayat
    for huruf in Huruf.objects.all():
        if not huruf.uu:
            if huruf.angka:
                huruf.uu = huruf.angka.uu
            elif huruf.pasal:
                huruf.uu = huruf.pasal.uu
            elif huruf.ayat:
                huruf.uu = huruf.ayat.pasal.uu
            huruf.save()

    # Mengaitkan Pasal ke UU berdasarkan Bab
    for pasal in Pasal.objects.all():
        if not pasal.uu and pasal.bab:
            pasal.uu = pasal.bab.uu
            pasal.save()

    # Mengaitkan Bagian ke UU berdasarkan Bab
    for bagian in Bagian.objects.all():
        if not bagian.uu and bagian.bab:
            bagian.uu = bagian.bab.uu
            bagian.save()

    # Mengaitkan Paragraf ke UU berdasarkan Bab
    for paragraf in Paragraf.objects.all():
        if not paragraf.uu and paragraf.bab:
            paragraf.uu = paragraf.bab.uu
            paragraf.save()

    # Mengaitkan Bab ke UU berdasarkan Buku (jika ada)
    for bab in Bab.objects.all():
        if not bab.uu and bab.buku:
            bab.uu = bab.buku.uu
            bab.save()

# Jalankan fungsi untuk mengaitkan relasi ke UU
update_uu_relations()