Initial InfraPulse scaffold
This commit is contained in:
@@ -0,0 +1 @@
|
||||
"""Pydantic API schemas."""
|
||||
@@ -0,0 +1,16 @@
|
||||
from pydantic import BaseModel, EmailStr
|
||||
|
||||
|
||||
class TokenResponse(BaseModel):
|
||||
access_token: str
|
||||
token_type: str = "bearer"
|
||||
|
||||
|
||||
class UserRead(BaseModel):
|
||||
id: int
|
||||
email: EmailStr
|
||||
display_name: str
|
||||
role: str
|
||||
is_active: bool
|
||||
|
||||
model_config = {"from_attributes": True}
|
||||
@@ -0,0 +1,156 @@
|
||||
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}
|
||||
Reference in New Issue
Block a user