"""企查查 VO 组装器。"""

from __future__ import annotations

from datetime import UTC
from typing import Any

from fastapi_modules.fastapi_leaudit.domian.vo.qichachaVo import QichachaCompanyInfoVO
from fastapi_modules.fastapi_leaudit.models.qichachaCompanyInfo import QichachaCompanyInfo


class QichachaVoAssembler:
    """企查查 VO 组装器。"""

    @classmethod
    def BuildCompanyInfo(cls, record: QichachaCompanyInfo) -> QichachaCompanyInfoVO:
        """组装企业信息 VO。"""
        dishonesty = record.dishonesty if isinstance(record.dishonesty, dict) else None
        data_list = dishonesty.get("Data") if dishonesty else []
        has_dishonesty = bool(dishonesty and int(dishonesty.get("VerifyResult") or 0) == 1)
        dishonesty_count = len(data_list) if isinstance(data_list, list) else 0
        return QichachaCompanyInfoVO(
            searchKey=record.searchKey,
            creditCode=record.creditCode,
            companyName=record.companyName,
            enterprise=record.enterprise if isinstance(record.enterprise, dict) else None,
            dishonesty=dishonesty,
            hasDishonesty=has_dishonesty,
            dishonestyCount=dishonesty_count,
            updatedAt=cls.FormatDatetime(record.updated_at),
        )

    @classmethod
    def FormatDatetime(cls, value: Any) -> str | None:
        """格式化时间。"""
        if value is None:
            return None
        if getattr(value, "tzinfo", None) is None:
            value = value.replace(tzinfo=UTC)
        return value.isoformat()