CE 마킹

# CE 마킹 적합성 평가 체크리스트 시스템
from dataclasses import dataclass, field
from typing import List, Dict, Optional
from datetime import datetime
from enum import Enum

class AssessmentRoute(Enum):
    """적합성 평가 경로"""
    SELF_ASSESSMENT = "self_assessment"      # 자체 적합성 평가
    NOTIFIED_BODY = "notified_body"          # 제3자 기관 평가

class ConformityModule(Enum):
    """EU 적합성 평가 모듈"""
    MODULE_A = "internal_production_control"  # 내부 생산 통제
    MODULE_B = "eu_type_examination"          # EU 유형 검사
    MODULE_C = "conformity_to_type"           # 유형 적합성
    MODULE_D = "quality_assurance"            # 품질 보증
    MODULE_H = "full_quality_assurance"       # 완전 품질 보증

@dataclass
class CEMarkingRequirement:
    """CE 마킹 요구사항"""
    id: str
    category: str
    description: str
    eu_ai_act_article: str
    status: str = "not_started"  # not_started, in_progress, completed
    evidence_document: Optional[str] = None

@dataclass
class CEMarkingProcess:
    """CE 마킹 획득 프로세스 관리"""
    ai_system_name: str
    provider_name: str
    high_risk_category: str
    assessment_route: AssessmentRoute
    notified_body: Optional[str] = None
    requirements: List[CEMarkingRequirement] = field(default_factory=list)

    # 마킹 정보
    ce_marking_date: Optional[datetime] = None
    declaration_number: Optional[str] = None
    registration_id: Optional[str] = None

    def __post_init__(self):
        self._initialize_requirements()

    def _initialize_requirements(self):
        """EU AI Act 기반 CE 마킹 요구사항"""
        reqs = [
            CEMarkingRequirement(
                id="CE-01",
                category="위험관리",
                description="위험 관리 시스템 수립 및 문서화",
                eu_ai_act_article="Article 9"
            ),
            CEMarkingRequirement(
                id="CE-02",
                category="데이터 거버넌스",
                description="학습/검증/테스트 데이터 품질 관리",
                eu_ai_act_article="Article 10"
            ),
            CEMarkingRequirement(
                id="CE-03",
                category="기술문서",
                description="기술 문서 작성 (부속서 IV 준수)",
                eu_ai_act_article="Article 11"
            ),
            CEMarkingRequirement(
                id="CE-04",
                category="기록보관",
                description="자동 로깅 기능 구현",
                eu_ai_act_article="Article 12"
            ),
            CEMarkingRequirement(
                id="CE-05",
                category="투명성",
                description="사용자 대상 정보 제공 체계",
                eu_ai_act_article="Article 13"
            ),
            CEMarkingRequirement(
                id="CE-06",
                category="인간감독",
                description="인간 감독 메커니즘 구현",
                eu_ai_act_article="Article 14"
            ),
            CEMarkingRequirement(
                id="CE-07",
                category="정확성/견고성",
                description="정확성, 견고성, 사이버보안 확보",
                eu_ai_act_article="Article 15"
            ),
            CEMarkingRequirement(
                id="CE-08",
                category="품질관리",
                description="품질관리시스템(QMS) 구축",
                eu_ai_act_article="Article 17"
            ),
        ]
        self.requirements.extend(reqs)

    def check_conformity_readiness(self) -> Dict:
        """적합성 평가 준비 상태 확인"""
        total = len(self.requirements)
        completed = len([r for r in self.requirements if r.status == "completed"])
        in_progress = len([r for r in self.requirements if r.status == "in_progress"])

        return {
            "ai_system": self.ai_system_name,
            "provider": self.provider_name,
            "assessment_route": self.assessment_route.value,
            "notified_body": self.notified_body,
            "progress": {
                "total_requirements": total,
                "completed": completed,
                "in_progress": in_progress,
                "not_started": total - completed - in_progress,
                "completion_rate": completed / total * 100
            },
            "ready_for_assessment": completed == total,
            "pending_items": [
                {"id": r.id, "description": r.description}
                for r in self.requirements if r.status != "completed"
            ]
        }

    def generate_declaration_of_conformity(self) -> Dict:
        """EU 적합성 선언서 생성"""
        if not self.check_conformity_readiness()["ready_for_assessment"]:
            raise ValueError("모든 요구사항 충족 전 선언서 생성 불가")

        return {
            "document_type": "EU DECLARATION OF CONFORMITY",
            "declaration_number": self.declaration_number or f"DoC-{datetime.now().strftime('%Y%m%d')}",
            "provider": {
                "name": self.provider_name,
                "address": "...",  # 실제 구현 시 입력
                "contact": "..."
            },
            "ai_system": {
                "name": self.ai_system_name,
                "high_risk_category": self.high_risk_category,
                "version": "...",
                "unique_identifier": "..."
            },
            "conformity_assessment": {
                "route": self.assessment_route.value,
                "notified_body": self.notified_body,
                "certificate_number": "..." if self.notified_body else None
            },
            "legal_references": [
                "Regulation (EU) 2024/1689 (AI Act)",
            ],
            "declaration": "본 AI 시스템은 위 규정의 관련 요구사항을 준수합니다.",
            "signed_date": datetime.now().isoformat(),
            "authorized_signatory": "..."
        }

# 사용 예시
ce_process = CEMarkingProcess(
    ai_system_name="AI 신용평가 시스템 v2.0",
    provider_name="KAITRUST Corp",
    high_risk_category="Essential Services - Credit Scoring",
    assessment_route=AssessmentRoute.SELF_ASSESSMENT
)

# 요구사항 완료 처리
for req in ce_process.requirements:
    req.status = "completed"
    req.evidence_document = f"DOC-{req.id}-evidence.pdf"

print(ce_process.check_conformity_readiness())
print(ce_process.generate_declaration_of_conformity())