Keith Smith
157 lines
4.6 KiB
Python
157 lines
4.6 KiB
Python
from datetime import datetime
|
|
from typing import Any
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class AssetCreate(BaseModel):
|
|
name: str = Field(min_length=1, max_length=160)
|
|
asset_type: str = Field(min_length=1, max_length=64)
|
|
address: str | None = Field(default=None, max_length=255)
|
|
metadata: dict[str, Any] = Field(default_factory=dict)
|
|
|
|
|
|
class AssetUpdate(BaseModel):
|
|
name: str | None = Field(default=None, min_length=1, max_length=160)
|
|
asset_type: str | None = Field(default=None, min_length=1, max_length=64)
|
|
address: str | None = Field(default=None, max_length=255)
|
|
status: str | None = Field(default=None, max_length=32)
|
|
metadata: dict[str, Any] | None = None
|
|
|
|
|
|
class AssetRead(AssetCreate):
|
|
id: int
|
|
status: str
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
|
|
model_config = {"from_attributes": True}
|
|
|
|
|
|
class MonitorCreate(BaseModel):
|
|
name: str = Field(min_length=1, max_length=160)
|
|
monitor_type: str = Field(default="http", max_length=64)
|
|
target: str = Field(min_length=1, max_length=512)
|
|
asset_id: int | None = None
|
|
config: dict[str, Any] = Field(default_factory=dict)
|
|
interval_seconds: int = Field(default=60, ge=10)
|
|
|
|
|
|
class MonitorUpdate(BaseModel):
|
|
name: str | None = Field(default=None, min_length=1, max_length=160)
|
|
monitor_type: str | None = Field(default=None, max_length=64)
|
|
target: str | None = Field(default=None, min_length=1, max_length=512)
|
|
asset_id: int | None = None
|
|
config: dict[str, Any] | None = None
|
|
interval_seconds: int | None = Field(default=None, ge=10)
|
|
status: str | None = Field(default=None, max_length=32)
|
|
|
|
|
|
class MonitorRead(MonitorCreate):
|
|
id: int
|
|
status: str
|
|
last_checked_at: datetime | None
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
|
|
model_config = {"from_attributes": True}
|
|
|
|
|
|
class WebsiteMonitorCreate(BaseModel):
|
|
name: str = Field(min_length=1, max_length=160)
|
|
url: str = Field(min_length=1, max_length=512)
|
|
expected_status: int = Field(default=200, ge=100, le=599)
|
|
expected_text: str | None = None
|
|
unexpected_text: str | None = None
|
|
timeout_seconds: int = Field(default=10, ge=1, le=120)
|
|
interval_seconds: int = Field(default=60, ge=10)
|
|
create_asset: bool = True
|
|
alert_enabled: bool = True
|
|
alert_severity: str = "critical"
|
|
failure_threshold: int = Field(default=3, ge=1, le=20)
|
|
|
|
|
|
class CheckResultRead(BaseModel):
|
|
id: int
|
|
monitor_id: int
|
|
status: str
|
|
response_time_ms: int | None
|
|
message: str | None
|
|
observed_at: datetime
|
|
|
|
model_config = {"from_attributes": True}
|
|
|
|
|
|
class AlertRuleCreate(BaseModel):
|
|
monitor_id: int
|
|
name: str = Field(min_length=1, max_length=160)
|
|
severity: str = "warning"
|
|
condition: dict[str, Any] = Field(default_factory=dict)
|
|
failure_threshold: int = Field(default=3, ge=1)
|
|
cooldown_seconds: int = Field(default=300, ge=0)
|
|
is_enabled: bool = True
|
|
|
|
|
|
class AlertRuleUpdate(BaseModel):
|
|
monitor_id: int | None = None
|
|
name: str | None = Field(default=None, min_length=1, max_length=160)
|
|
severity: str | None = None
|
|
condition: dict[str, Any] | None = None
|
|
failure_threshold: int | None = Field(default=None, ge=1)
|
|
cooldown_seconds: int | None = Field(default=None, ge=0)
|
|
is_enabled: bool | None = None
|
|
|
|
|
|
class AlertRuleRead(AlertRuleCreate):
|
|
id: int
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
|
|
model_config = {"from_attributes": True}
|
|
|
|
|
|
class IncidentRead(BaseModel):
|
|
id: int
|
|
asset_id: int | None
|
|
monitor_id: int | None
|
|
alert_rule_id: int | None
|
|
title: str
|
|
severity: str
|
|
status: str
|
|
opened_at: datetime
|
|
resolved_at: datetime | None
|
|
acknowledged_at: datetime | None
|
|
silenced_until: datetime | None
|
|
details: dict[str, Any]
|
|
|
|
model_config = {"from_attributes": True}
|
|
|
|
|
|
class NotificationChannelCreate(BaseModel):
|
|
name: str = Field(min_length=1, max_length=160)
|
|
channel_type: str = Field(min_length=1, max_length=64)
|
|
settings: dict[str, Any] = Field(default_factory=dict)
|
|
secret: str | None = None
|
|
is_enabled: bool = True
|
|
|
|
|
|
class NotificationChannelUpdate(BaseModel):
|
|
name: str | None = Field(default=None, min_length=1, max_length=160)
|
|
channel_type: str | None = Field(default=None, min_length=1, max_length=64)
|
|
settings: dict[str, Any] | None = None
|
|
secret: str | None = None
|
|
is_enabled: bool | None = None
|
|
|
|
|
|
class NotificationChannelRead(BaseModel):
|
|
id: int
|
|
name: str
|
|
channel_type: str
|
|
settings: dict[str, Any]
|
|
has_secret: bool
|
|
is_enabled: bool
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
|
|
model_config = {"from_attributes": True}
|