地缘政治风险供应链影响标签 — 贸易限制/区域冲突对供应链的实时风险量化

"""
地缘政治风险供应链影响标签引擎
功能：风险事件监测 / 影响范围识别 / Tag传播 / 应对预案触发
"""
from dataclasses import dataclass, field
from datetime import datetime
import warnings
warnings.filterwarnings('ignore')


RISK_WEIGHTS = {"tariff": 0.30, "logistics": 0.25, "supplier_country": 0.25,
                "fx": 0.10, "export_control": 0.10}

RISK_SCORES = {"CRITICAL": 1.0, "HIGH": 0.75, "MEDIUM": 0.5, "LOW": 0.25, "NONE": 0.0}


@dataclass
class GeoRiskEvent:
    event_id: str
    event_type: str        # tariff / logistics / supplier_country / fx / export_control
    description: str
    affected_regions: list
    risk_level: str        # CRITICAL / HIGH / MEDIUM / LOW
    timestamp: datetime = field(default_factory=datetime.now)
    affected_routes: list = field(default_factory=list)
    affected_materials: list = field(default_factory=list)


@dataclass
class SupplyChainEntity:
    entity_id: str
    entity_type: str       # SKU / Supplier / Shipment
    country: str
    routes: list = field(default_factory=list)
    materials: list = field(default_factory=list)
    tags: dict = field(default_factory=dict)


class GeoRiskTagEngine:

    def __init__(self):
        self.entities: dict = {}
        self.active_risks: list = []
        self.impact_log: list = []

    def register_entity(self, entity: SupplyChainEntity):
        self.entities[entity.entity_id] = entity

    def process_risk_event(self, event: GeoRiskEvent) -> list:
        """处理风险事件，更新受影响实体的Tags"""
        self.active_risks.append(event)
        impacted = []

        for entity_id, entity in self.entities.items():
            impact_score = 0.0
            tags_to_set = {}

            # 检查是否受影响
            if event.event_type == "tariff":
                if entity.country in event.affected_regions or \