法规变更影响传播引擎 — 新规 → 受影响SKU/市场的Tag传播与行动触发

"""
法规变更影响传播引擎
功能：法规解析 / 受影响实体识别 / Tag传播 / Action触发 / 合规时间线管理
输入：法规变更事件 + 产品知识图谱
输出：受影响SKU列表 + Tag更新 + 行动计划 + 合规倒计时
"""
import numpy as np
import pandas as pd
from dataclasses import dataclass, field
from datetime import datetime, timedelta
from collections import deque
import warnings
warnings.filterwarnings('ignore')


@dataclass
class RegulatoryChange:
    """法规变更事件"""
    reg_id: str
    name: str
    jurisdiction: list       # 适用司法管辖区
    effective_date: datetime
    categories: list         # 适用品类
    materials: list          # 适用材料/成分
    impact_type: str         # IMMEDIATE / GRADUAL / PHASE_IN
    penalty_type: str        # DELISTING / FINE / RECALL
    grace_period_days: int = 0
    compliance_actions: list = field(default_factory=list)


@dataclass
class ProductNode:
    """产品图谱节点"""
    sku_id: str
    name: str
    markets: list
    categories: list
    materials: list
    supplier_ids: list
    tags: dict = field(default_factory=dict)


@dataclass
class ImpactResult:
    """影响评估结果"""
    sku_id: str
    impact_level: str      # CRITICAL / HIGH / MEDIUM / LOW
    impact_dimensions: list
    days_to_comply: int
    affected_inventory: int
    tags_applied: dict
    required_actions: list


class RegulatoryImpactEngine:
    """法规变更影响传播引擎"""

    def __init__(self):
        self.products: dict = {}        # sku_id → ProductNode
        self.suppliers: dict = {}       # supplier_id → {certs, materials}