"""Seed the MasterAnnualCTC table with common CTC ranges.

Run: python master_seeder/seed_master_annual_ctc.py
"""

import django
import os
import sys

CTC_RANGES = [
    "0–1 LPA",
    "1–2 LPA",
    "2–3 LPA",
    "3–5 LPA",
    "5–7 LPA",
    "6–8 LPA",
    "8–10 LPA",
    "10–12 LPA",
    "12–15 LPA",
    "15–20 LPA",
    "20–25 LPA",
    "25–30 LPA",
    "30+ LPA",
]


def seed_master_annual_ctc():
    from django.contrib.auth import get_user_model
    from apps.master_annual_ctc.models import MasterAnnualCTC

    User = get_user_model()
    admin = User.objects.filter(role="ADMIN").order_by("id").first() or User.objects.order_by("id").first()
    if not admin:
        print("ERROR: no user found to set created_by/updated_by.")
        sys.exit(1)

    existing = {n.lower() for n in MasterAnnualCTC.objects.values_list("name", flat=True)}
    created = 0
    for name in CTC_RANGES:
        if name.lower() in existing:
            continue
        MasterAnnualCTC.objects.create(name=name, status="ACTIVE", created_by=admin, updated_by=admin)
        created += 1

    print(f"Seeded {created} new CTC range(s); {MasterAnnualCTC.objects.count()} total in master_annual_ctcs.")


if __name__ == "__main__":
    sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
    os.environ.setdefault("DJANGO_SETTINGS_MODULE", "config.settings.dev")
    django.setup()
    seed_master_annual_ctc()