供应商本体能力图谱 — 产能/质量/认证/风险四维供应商Ontology设计与实例化

"""
供应商本体能力图谱
功能：供应商Ontology定义/实例化/健康评分/风险识别/查询引擎
输入：供应商基础数据 + 绩效历史数据
输出：供应商本体实例 + 健康评分 + 风险图谱 + 查询结果
"""
import numpy as np
import pandas as pd
from dataclasses import dataclass, field
from typing import Optional
from datetime import datetime
import warnings
warnings.filterwarnings('ignore')


@dataclass
class CertificationTags:
    fda_registered: bool = False
    ce_certified: bool = False
    iso9001: bool = False
    rohs_compliant: bool = False
    cpsc_compliant: bool = False
    cert_expiry: dict = field(default_factory=dict)

    def cert_score(self) -> float:
        """认证完整度评分（0-1）"""
        weights = {
            "fda_registered": 0.30, "ce_certified": 0.25,
            "iso9001": 0.20, "rohs_compliant": 0.15, "cpsc_compliant": 0.10,
        }
        score = sum(w for attr, w in weights.items() if getattr(self, attr, False))
        return score

    def active_certs(self) -> list:
        """获取有效认证列表（未过期）"""
        now = datetime.now().date()
        certs = []
        for attr in ["fda_registered", "ce_certified", "iso9001", "rohs_compliant", "cpsc_compliant"]:
            if getattr(self, attr, False):
                expiry = self.cert_expiry.get(attr)
                if expiry is None or expiry > now:
                    certs.append(attr)
        return certs


@dataclass
class CapabilityTags:
    product_lines: list = field(default_factory=list)
    monthly_capacity: int = 0
    min_order_qty: int = 0
    lead_time_days: int = 30
    customization_capability: bool = False
    tech_level: str = "standard"  # entry/standard/advanced

    def can_fulfill(self, product_line: str, qty: int) -> bool:
        return (product_line in self.product_lines and qty >= self.min_order_qty
                and qty <= self.monthly_capacity)


@dataclass