c0mrade
9a2aea038a
- Add pyproject.toml with uv and ruff configuration - Pin Python version to 3.13 via .python-version - Add Makefile commands: lint, format, fix - Apply ruff formatting to entire codebase - Remove unused imports (base64 in yookassa/simple_subscription) - Update .gitignore for new config files
387 lines
8.0 KiB
Python
387 lines
8.0 KiB
Python
"""Schemas for Ban System integration in cabinet."""
|
|
|
|
from datetime import datetime
|
|
from typing import Any
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
# === Status ===
|
|
|
|
|
|
class BanSystemStatusResponse(BaseModel):
|
|
"""Ban System integration status."""
|
|
|
|
enabled: bool
|
|
configured: bool
|
|
|
|
|
|
# === Stats ===
|
|
|
|
|
|
class BanSystemStatsResponse(BaseModel):
|
|
"""Overall Ban System statistics."""
|
|
|
|
total_users: int = 0
|
|
active_users: int = 0
|
|
users_over_limit: int = 0
|
|
total_requests: int = 0
|
|
total_punishments: int = 0
|
|
active_punishments: int = 0
|
|
nodes_online: int = 0
|
|
nodes_total: int = 0
|
|
agents_online: int = 0
|
|
agents_total: int = 0
|
|
panel_connected: bool = False
|
|
uptime_seconds: int | None = None
|
|
|
|
|
|
# === Users ===
|
|
|
|
|
|
class BanUserIPInfo(BaseModel):
|
|
"""User IP address information."""
|
|
|
|
ip: str
|
|
first_seen: datetime | None = None
|
|
last_seen: datetime | None = None
|
|
node: str | None = None
|
|
request_count: int = 0
|
|
country_code: str | None = None
|
|
country_name: str | None = None
|
|
city: str | None = None
|
|
|
|
|
|
class BanUserRequestLog(BaseModel):
|
|
"""User request log entry."""
|
|
|
|
timestamp: datetime
|
|
source_ip: str
|
|
destination: str | None = None
|
|
dest_port: int | None = None
|
|
protocol: str | None = None
|
|
action: str | None = None
|
|
node: str | None = None
|
|
|
|
|
|
class BanUserListItem(BaseModel):
|
|
"""User in the list."""
|
|
|
|
email: str
|
|
unique_ip_count: int = 0
|
|
total_requests: int = 0
|
|
limit: int | None = None
|
|
is_over_limit: bool = False
|
|
blocked_count: int = 0
|
|
last_seen: datetime | None = None
|
|
|
|
|
|
class BanUsersListResponse(BaseModel):
|
|
"""Paginated list of users."""
|
|
|
|
users: list[BanUserListItem] = []
|
|
total: int = 0
|
|
offset: int = 0
|
|
limit: int = 50
|
|
|
|
|
|
class BanUserDetailResponse(BaseModel):
|
|
"""Detailed user information."""
|
|
|
|
email: str
|
|
unique_ip_count: int = 0
|
|
total_requests: int = 0
|
|
limit: int | None = None
|
|
is_over_limit: bool = False
|
|
blocked_count: int = 0
|
|
ips: list[BanUserIPInfo] = []
|
|
recent_requests: list[BanUserRequestLog] = []
|
|
network_type: str | None = None # wifi, mobile, mixed
|
|
|
|
|
|
# === Punishments (Bans) ===
|
|
|
|
|
|
class BanPunishmentItem(BaseModel):
|
|
"""Punishment/ban entry."""
|
|
|
|
id: int | None = None
|
|
user_id: str
|
|
uuid: str | None = None
|
|
username: str
|
|
reason: str | None = None
|
|
punished_at: datetime
|
|
enable_at: datetime | None = None
|
|
ip_count: int = 0
|
|
limit: int = 0
|
|
enabled: bool = False
|
|
enabled_at: datetime | None = None
|
|
node_name: str | None = None
|
|
|
|
|
|
class BanPunishmentsListResponse(BaseModel):
|
|
"""List of active punishments."""
|
|
|
|
punishments: list[BanPunishmentItem] = []
|
|
total: int = 0
|
|
|
|
|
|
class BanHistoryResponse(BaseModel):
|
|
"""Punishment history."""
|
|
|
|
items: list[BanPunishmentItem] = []
|
|
total: int = 0
|
|
|
|
|
|
class BanUserRequest(BaseModel):
|
|
"""Request to ban a user."""
|
|
|
|
username: str = Field(..., min_length=1)
|
|
minutes: int = Field(default=30, ge=1)
|
|
reason: str | None = Field(None, max_length=500)
|
|
|
|
|
|
class UnbanResponse(BaseModel):
|
|
"""Unban response."""
|
|
|
|
success: bool
|
|
message: str
|
|
|
|
|
|
# === Nodes ===
|
|
|
|
|
|
class BanNodeItem(BaseModel):
|
|
"""Node information."""
|
|
|
|
name: str
|
|
address: str | None = None
|
|
is_connected: bool = False
|
|
last_seen: datetime | None = None
|
|
users_count: int = 0
|
|
agent_stats: dict[str, Any] | None = None
|
|
|
|
|
|
class BanNodesListResponse(BaseModel):
|
|
"""List of nodes."""
|
|
|
|
nodes: list[BanNodeItem] = []
|
|
total: int = 0
|
|
online: int = 0
|
|
|
|
|
|
# === Agents ===
|
|
|
|
|
|
class BanAgentItem(BaseModel):
|
|
"""Monitoring agent information."""
|
|
|
|
node_name: str
|
|
sent_total: int = 0
|
|
dropped_total: int = 0
|
|
batches_total: int = 0
|
|
reconnects: int = 0
|
|
failures: int = 0
|
|
queue_size: int = 0
|
|
queue_max: int = 0
|
|
dedup_checked: int = 0
|
|
dedup_skipped: int = 0
|
|
filter_checked: int = 0
|
|
filter_filtered: int = 0
|
|
health: str = 'unknown' # healthy, warning, critical
|
|
is_online: bool = False
|
|
last_report: datetime | None = None
|
|
|
|
|
|
class BanAgentsSummary(BaseModel):
|
|
"""Agents summary statistics."""
|
|
|
|
total_agents: int = 0
|
|
online_agents: int = 0
|
|
total_sent: int = 0
|
|
total_dropped: int = 0
|
|
avg_queue_size: float = 0.0
|
|
healthy_count: int = 0
|
|
warning_count: int = 0
|
|
critical_count: int = 0
|
|
|
|
|
|
class BanAgentsListResponse(BaseModel):
|
|
"""List of agents."""
|
|
|
|
agents: list[BanAgentItem] = []
|
|
summary: BanAgentsSummary | None = None
|
|
total: int = 0
|
|
online: int = 0
|
|
|
|
|
|
# === Traffic ===
|
|
|
|
|
|
class BanTrafficStats(BaseModel):
|
|
"""Traffic statistics."""
|
|
|
|
total_bytes: int = 0
|
|
upload_bytes: int = 0
|
|
download_bytes: int = 0
|
|
total_users: int = 0
|
|
violators_count: int = 0
|
|
|
|
|
|
class BanTrafficUserItem(BaseModel):
|
|
"""User traffic information."""
|
|
|
|
username: str
|
|
email: str | None = None
|
|
total_bytes: int = 0
|
|
upload_bytes: int = 0
|
|
download_bytes: int = 0
|
|
limit_bytes: int | None = None
|
|
is_over_limit: bool = False
|
|
|
|
|
|
class BanTrafficViolationItem(BaseModel):
|
|
"""Traffic limit violation entry."""
|
|
|
|
id: int | None = None
|
|
username: str
|
|
email: str | None = None
|
|
violation_type: str
|
|
description: str | None = None
|
|
bytes_used: int = 0
|
|
bytes_limit: int = 0
|
|
detected_at: datetime
|
|
resolved: bool = False
|
|
|
|
|
|
class BanTrafficViolationsResponse(BaseModel):
|
|
"""List of traffic violations."""
|
|
|
|
violations: list[BanTrafficViolationItem] = []
|
|
total: int = 0
|
|
|
|
|
|
class BanTrafficTopItem(BaseModel):
|
|
"""Top user by traffic."""
|
|
|
|
username: str
|
|
bytes_total: int = 0
|
|
bytes_limit: int | None = None
|
|
over_limit: bool = False
|
|
|
|
|
|
class BanTrafficResponse(BaseModel):
|
|
"""Full traffic statistics response."""
|
|
|
|
enabled: bool = False
|
|
stats: dict[str, Any] | None = None
|
|
top_users: list[BanTrafficTopItem] = []
|
|
recent_violations: list[BanTrafficViolationItem] = []
|
|
|
|
|
|
# === Settings ===
|
|
|
|
|
|
class BanSettingDefinition(BaseModel):
|
|
"""Setting definition with value."""
|
|
|
|
key: str
|
|
value: Any
|
|
type: str # bool, int, str, list
|
|
min_value: int | None = None
|
|
max_value: int | None = None
|
|
editable: bool = True
|
|
description: str | None = None
|
|
category: str | None = None
|
|
|
|
|
|
class BanSettingsResponse(BaseModel):
|
|
"""All settings response."""
|
|
|
|
settings: list[BanSettingDefinition] = []
|
|
|
|
|
|
class BanSettingUpdateRequest(BaseModel):
|
|
"""Request to update a setting."""
|
|
|
|
value: Any
|
|
|
|
|
|
class BanWhitelistRequest(BaseModel):
|
|
"""Request to add/remove from whitelist."""
|
|
|
|
username: str = Field(..., min_length=1)
|
|
|
|
|
|
# === Reports ===
|
|
|
|
|
|
class BanReportTopViolator(BaseModel):
|
|
"""Top violator in report."""
|
|
|
|
username: str
|
|
count: int = 0
|
|
|
|
|
|
class BanReportResponse(BaseModel):
|
|
"""Period report response."""
|
|
|
|
period_hours: int = 24
|
|
current_users: int = 0
|
|
current_ips: int = 0
|
|
punishment_stats: dict[str, Any] | None = None
|
|
top_violators: list[BanReportTopViolator] = []
|
|
|
|
|
|
# === Health ===
|
|
|
|
|
|
class BanHealthComponent(BaseModel):
|
|
"""Health component status."""
|
|
|
|
name: str
|
|
status: str # healthy, degraded, unhealthy
|
|
message: str | None = None
|
|
details: dict[str, Any] | None = None
|
|
|
|
|
|
class BanHealthResponse(BaseModel):
|
|
"""Health status response."""
|
|
|
|
status: str # healthy, degraded, unhealthy
|
|
uptime: int | None = None
|
|
components: list[BanHealthComponent] = []
|
|
|
|
|
|
class BanHealthDetailedResponse(BaseModel):
|
|
"""Detailed health response."""
|
|
|
|
status: str
|
|
uptime: int | None = None
|
|
components: dict[str, Any] = {}
|
|
|
|
|
|
# === Agent History ===
|
|
|
|
|
|
class BanAgentHistoryItem(BaseModel):
|
|
"""Agent history item."""
|
|
|
|
timestamp: datetime
|
|
sent_total: int = 0
|
|
dropped_total: int = 0
|
|
queue_size: int = 0
|
|
batches_total: int = 0
|
|
|
|
|
|
class BanAgentHistoryResponse(BaseModel):
|
|
"""Agent history response."""
|
|
|
|
node: str
|
|
hours: int = 24
|
|
records: int = 0
|
|
delta: dict[str, Any] | None = None
|
|
first: dict[str, Any] | None = None
|
|
last: dict[str, Any] | None = None
|
|
history: list[BanAgentHistoryItem] = []
|