from django.core.management.base import BaseCommand
import pandas as pd
from resumes.models import ResumeBasic, ResumeDetail
from django.db import transaction


class Command(BaseCommand):
    help = "导入护理类简历详情数据到 ResumeDetail 模型"

    def add_arguments(self, parser):
        parser.add_argument("filepath", type=str, help="Excel 文件路径")

    def handle(self, *args, **options):
        filepath = options["filepath"]
        df = pd.read_excel(filepath)

        success, skipped, errors = 0, 0, []

        for _, row in df.iterrows():
            raw_resume_id = row.get("resume_id")
            phone = row.get("phone")
            email = row.get("email")

            if pd.isna(raw_resume_id):
                skipped += 1
                continue

            resume_id = None
            try:
                resume_id = int(str(raw_resume_id).strip().split(".")[0])  # 去掉小数点尾巴等干扰
            except (ValueError, TypeError):
                errors.append(f"resume_id 非法格式: {raw_resume_id}")
                continue

            try:
                basic = ResumeBasic.objects.get(resume_id=resume_id)
                ResumeDetail.objects.update_or_create(
                    resume=basic,
                    defaults={
                        "unlinked_resume_id": None,
                        "phone": str(phone).strip() if not pd.isna(phone) else "",
                        "email": str(email).strip() if not pd.isna(email) else ""
                    }
                )
                success += 1
            except ResumeBasic.DoesNotExist:
                ResumeDetail.objects.update_or_create(
                    unlinked_resume_id=resume_id,
                    defaults={
                        "resume": None,
                        "phone": str(phone).strip() if not pd.isna(phone) else "",
                        "email": str(email).strip() if not pd.isna(email) else ""
                    }
                )
                success += 1
                errors.append(f"resume_id={resume_id} 无对应 ResumeBasic，已记录至 unlinked_resume_id")

        self.stdout.write(self.style.SUCCESS(f"成功导入 {success} 条，跳过 {skipped} 条"))
        if errors:
            self.stdout.write(self.style.WARNING("以下数据未关联基础简历："))
            for msg in errors:
                self.stdout.write(f" - {msg}")