Add GitOps Guardian application with multi-agent architecture for security and compliance reviews

week8/community_contributions/salah/gitops-guardian/models.py

@@ -0,0 +1,83 @@
+from pydantic import BaseModel, Field
+from typing import List, Optional
+from datetime import datetime
+from enum import Enum
+
+
+class RiskLevel(str, Enum):
+    SAFE = "SAFE"
+    REVIEW = "REVIEW"
+    RISKY = "RISKY"
+
+
+class SecurityIssue(BaseModel):
+    type: str
+    severity: str
+    line_number: Optional[int] = None
+    file_path: Optional[str] = None
+    description: str
+    recommendation: str
+
+
+class ComplianceViolation(BaseModel):
+    rule: str
+    severity: str
+    file_path: Optional[str] = None
+    description: str
+    suggestion: str
+
+
+class SecurityScore(BaseModel):
+    risk: float
+    issues: List[SecurityIssue] = []
+    summary: str
+    confidence: float = 0.8
+    cost: float = 0.0
+
+
+class ComplianceScore(BaseModel):
+    risk: float
+    violations: List[ComplianceViolation] = []
+    passed_checks: List[str] = []
+    summary: str
+
+
+class RiskAssessment(BaseModel):
+    overall_risk: float
+    risk_level: RiskLevel
+    security_score: SecurityScore
+    compliance_score: ComplianceScore
+    recommendation: str
+    confidence: float = 0.8
+
+
+class PullRequest(BaseModel):
+    repo: str
+    number: int
+    title: str
+    author: str
+    url: str
+    diff: str
+    files_changed: List[str] = []
+    created_at: datetime
+    labels: List[str] = []
+
+
+class ReviewResult(BaseModel):
+    pr: PullRequest
+    risk_assessment: RiskAssessment
+    reviewed_at: datetime = Field(default_factory=datetime.now)
+
+    def to_summary(self):
+        risk = self.risk_assessment
+        level_marker = "SAFE" if risk.risk_level == RiskLevel.SAFE else "REVIEW" if risk.risk_level == RiskLevel.REVIEW else "RISKY"
+
+        summary = f"""PR #{self.pr.number}: {self.pr.title}
+Overall Risk: {risk.overall_risk:.2f} - {level_marker}
+
+Security: {risk.security_score.risk:.2f} - {risk.security_score.summary}
+Compliance: {risk.compliance_score.risk:.2f} - {risk.compliance_score.summary}
+
+Recommendation: {risk.recommendation}
+"""
+        return summary.strip()