from datetime import datetime
from pydantic import BaseModel


class CVE(BaseModel):
    cve_id: str
    summary: str | None = None
    cvss: float | None = None
    cvss_version: int | str | None = None
    cvss_v2: float | None = None
    cvss_v3: float | None = None
    epss: float | None = None
    ranking_epss: float | None = None
    kev: bool = False
    propose_action: str | None = None
    ransomware_campaign: str | None = None
    references: list[str] = []
    published_time: datetime | None = None
    vendor: str | None = None
    product: str | None = None
    version: str | None = None


class CVEWithCPEs(CVE):
    cpes: list[str] = []