mirror of
https://github.com/mblanke/ThreatHunt.git
synced 2026-03-01 14:00:20 -05:00
33 lines
822 B
Python
33 lines
822 B
Python
from pydantic import BaseModel
|
|
from typing import Optional, List, Dict, Any
|
|
from datetime import datetime
|
|
|
|
|
|
class ThreatScoreBase(BaseModel):
|
|
"""Base threat score schema"""
|
|
score: float
|
|
confidence: float
|
|
threat_type: str
|
|
description: Optional[str] = None
|
|
indicators: Optional[List[Dict[str, Any]]] = None
|
|
|
|
|
|
class ThreatScoreCreate(ThreatScoreBase):
|
|
"""Schema for creating a threat score"""
|
|
host_id: Optional[int] = None
|
|
artifact_id: Optional[int] = None
|
|
ml_model_version: Optional[str] = None
|
|
|
|
|
|
class ThreatScoreRead(ThreatScoreBase):
|
|
"""Schema for reading threat score data"""
|
|
id: int
|
|
tenant_id: int
|
|
host_id: Optional[int]
|
|
artifact_id: Optional[int]
|
|
ml_model_version: Optional[str]
|
|
created_at: datetime
|
|
|
|
class Config:
|
|
from_attributes = True
|