from pydantic import BaseModel, Field

from core.schemas.status import AuditSeverity, AuditStatus


class AuditFindings(BaseModel):
    name: str = Field()
    description: str = Field("")
    severity: AuditSeverity = Field()


class AuditResults(BaseModel):
    status: AuditStatus = Field()
    findings: list[AuditFindings] = Field()
    risk_level: float = Field(
        description="value from 0.0 to 1.0, indicating how dangerous the auditresults are."
    )