349 lines
8.3 KiB
Python
349 lines
8.3 KiB
Python
"""Pydantic schemas for admin panel endpoints."""
|
|
from datetime import datetime
|
|
from typing import Literal, Optional
|
|
from uuid import UUID
|
|
from pydantic import BaseModel, EmailStr, Field
|
|
|
|
|
|
# --- Dashboard ---
|
|
|
|
class DashboardMetrics(BaseModel):
|
|
total_users: int
|
|
active_subscriptions: int
|
|
paid_accounts: int
|
|
total_trees: int
|
|
|
|
|
|
class ActivityEntry(BaseModel):
|
|
id: UUID
|
|
user_email: Optional[str] = None
|
|
action: str
|
|
resource_type: str
|
|
resource_id: Optional[UUID] = None
|
|
details: Optional[dict] = None
|
|
ip_address: Optional[str] = None
|
|
created_at: datetime
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
# --- Admin Accounts & People Search ---
|
|
|
|
class AdminUserListItem(BaseModel):
|
|
id: UUID
|
|
email: EmailStr
|
|
name: str
|
|
role: str
|
|
is_super_admin: bool = False
|
|
is_active: bool = True
|
|
account_id: Optional[UUID] = None
|
|
account_role: Optional[str] = None
|
|
account_name: Optional[str] = None
|
|
account_display_code: Optional[str] = None
|
|
created_at: datetime
|
|
last_login: Optional[datetime] = None
|
|
deleted_at: Optional[datetime] = None
|
|
|
|
|
|
class AdminUserListResponse(BaseModel):
|
|
items: list[AdminUserListItem]
|
|
total: int
|
|
page: int
|
|
per_page: int
|
|
|
|
|
|
class AdminAccountMember(BaseModel):
|
|
id: UUID
|
|
email: EmailStr
|
|
name: str
|
|
role: str
|
|
is_super_admin: bool = False
|
|
is_active: bool = True
|
|
account_role: Optional[str] = None
|
|
created_at: datetime
|
|
last_login: Optional[datetime] = None
|
|
deleted_at: Optional[datetime] = None
|
|
|
|
|
|
class AdminAccountOwnerSummary(BaseModel):
|
|
id: UUID
|
|
name: str
|
|
email: EmailStr
|
|
|
|
|
|
class AdminAccountSubscriptionSummary(BaseModel):
|
|
id: UUID
|
|
plan: str
|
|
status: str
|
|
billing_interval: Optional[str] = None
|
|
current_period_end: Optional[datetime] = None
|
|
cancel_at_period_end: bool = False
|
|
|
|
|
|
class AdminAccountUsageSummary(BaseModel):
|
|
tree_count: int = 0
|
|
session_count_this_month: int = 0
|
|
|
|
|
|
class AdminAccountInviteSummary(BaseModel):
|
|
id: UUID
|
|
email: EmailStr
|
|
role: str
|
|
expires_at: Optional[datetime] = None
|
|
created_at: datetime
|
|
used_at: Optional[datetime] = None
|
|
|
|
|
|
class AdminAccountListItem(BaseModel):
|
|
id: UUID
|
|
name: str
|
|
display_code: str
|
|
created_at: datetime
|
|
owner_id: Optional[UUID] = None
|
|
owner: Optional[AdminAccountOwnerSummary] = None
|
|
subscription: Optional[AdminAccountSubscriptionSummary] = None
|
|
usage: AdminAccountUsageSummary = Field(default_factory=AdminAccountUsageSummary)
|
|
member_count: int = 0
|
|
active_member_count: int = 0
|
|
pending_invite_count: int = 0
|
|
sso_enabled: bool = False
|
|
branding_company_name: Optional[str] = None
|
|
members: list[AdminAccountMember] = Field(default_factory=list)
|
|
|
|
|
|
class AdminAccountListResponse(BaseModel):
|
|
items: list[AdminAccountListItem]
|
|
total: int
|
|
page: int
|
|
per_page: int
|
|
|
|
|
|
class AdminAccountDetailResponse(AdminAccountListItem):
|
|
invites: list[AdminAccountInviteSummary] = Field(default_factory=list)
|
|
|
|
|
|
class AdminAccountCreate(BaseModel):
|
|
name: str = Field(..., min_length=1, max_length=255)
|
|
plan: Literal["free", "pro", "team"] = "free"
|
|
|
|
|
|
class AdminAccountUpdate(BaseModel):
|
|
name: str = Field(..., min_length=1, max_length=255)
|
|
|
|
|
|
# --- Audit Logs ---
|
|
|
|
class AuditLogEntry(BaseModel):
|
|
id: UUID
|
|
user_id: UUID
|
|
user_email: Optional[str] = None
|
|
action: str
|
|
resource_type: str
|
|
resource_id: Optional[UUID] = None
|
|
details: Optional[dict] = None
|
|
ip_address: Optional[str] = None
|
|
created_at: datetime
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class AuditLogListResponse(BaseModel):
|
|
items: list[AuditLogEntry]
|
|
total: int
|
|
page: int
|
|
per_page: int
|
|
|
|
|
|
# --- Plan Limits ---
|
|
|
|
class PlanLimitResponse(BaseModel):
|
|
plan: str
|
|
max_trees: Optional[int] = None
|
|
max_sessions_per_month: Optional[int] = None
|
|
max_users: Optional[int] = None
|
|
custom_branding: bool = False
|
|
priority_support: bool = False
|
|
export_formats: list = []
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class PlanLimitUpdate(BaseModel):
|
|
plan: str
|
|
max_trees: Optional[int] = None
|
|
max_sessions_per_month: Optional[int] = None
|
|
max_users: Optional[int] = None
|
|
custom_branding: bool = False
|
|
priority_support: bool = False
|
|
export_formats: list = Field(default_factory=lambda: ["markdown", "text"])
|
|
|
|
|
|
class AccountOverrideCreate(BaseModel):
|
|
account_display_code: str = Field(..., description="Account display code to look up")
|
|
override_max_trees: Optional[int] = None
|
|
override_max_sessions_per_month: Optional[int] = None
|
|
override_max_users: Optional[int] = None
|
|
note: Optional[str] = None
|
|
|
|
|
|
class AccountOverrideUpdate(BaseModel):
|
|
override_max_trees: Optional[int] = None
|
|
override_max_sessions_per_month: Optional[int] = None
|
|
override_max_users: Optional[int] = None
|
|
note: Optional[str] = None
|
|
|
|
|
|
class AccountOverrideResponse(BaseModel):
|
|
id: UUID
|
|
account_id: UUID
|
|
account_name: Optional[str] = None
|
|
account_display_code: Optional[str] = None
|
|
override_max_trees: Optional[int] = None
|
|
override_max_sessions_per_month: Optional[int] = None
|
|
override_max_users: Optional[int] = None
|
|
note: Optional[str] = None
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
# --- Feature Flags ---
|
|
|
|
class FeatureFlagCreate(BaseModel):
|
|
flag_key: str = Field(..., max_length=100)
|
|
display_name: str = Field(..., max_length=255)
|
|
description: Optional[str] = None
|
|
|
|
|
|
class FeatureFlagUpdate(BaseModel):
|
|
display_name: Optional[str] = None
|
|
description: Optional[str] = None
|
|
|
|
|
|
class PlanDefaultEntry(BaseModel):
|
|
plan: str
|
|
enabled: bool
|
|
|
|
|
|
class FeatureFlagResponse(BaseModel):
|
|
id: UUID
|
|
flag_key: str
|
|
display_name: str
|
|
description: Optional[str] = None
|
|
plan_defaults: list[PlanDefaultEntry] = []
|
|
created_at: datetime
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class PlanDefaultUpdate(BaseModel):
|
|
plan: str
|
|
flag_id: UUID
|
|
enabled: bool
|
|
|
|
|
|
class AccountFeatureOverrideCreate(BaseModel):
|
|
account_display_code: str
|
|
flag_id: UUID
|
|
enabled: bool
|
|
note: Optional[str] = None
|
|
|
|
|
|
class AccountFeatureOverrideResponse(BaseModel):
|
|
id: UUID
|
|
account_id: UUID
|
|
account_display_code: Optional[str] = None
|
|
flag_id: UUID
|
|
flag_key: Optional[str] = None
|
|
flag_display_name: Optional[str] = None
|
|
enabled: bool
|
|
note: Optional[str] = None
|
|
created_at: datetime
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
# --- Platform Settings ---
|
|
|
|
class SettingsResponse(BaseModel):
|
|
settings: dict
|
|
|
|
|
|
class SettingsUpdate(BaseModel):
|
|
settings: dict = Field(..., description="Key-value pairs to update")
|
|
|
|
|
|
# --- Global Categories ---
|
|
|
|
class GlobalCategoryCreate(BaseModel):
|
|
name: str = Field(..., min_length=1, max_length=100)
|
|
slug: str = Field(..., min_length=1, max_length=100)
|
|
description: Optional[str] = None
|
|
|
|
|
|
class GlobalCategoryUpdate(BaseModel):
|
|
name: Optional[str] = Field(None, max_length=100)
|
|
slug: Optional[str] = Field(None, max_length=100)
|
|
description: Optional[str] = None
|
|
|
|
|
|
class GlobalCategoryResponse(BaseModel):
|
|
id: UUID
|
|
name: str
|
|
slug: str
|
|
description: Optional[str] = None
|
|
account_id: Optional[UUID] = None
|
|
tree_count: int = 0
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
# --- Move User ---
|
|
|
|
class MoveUserAccount(BaseModel):
|
|
display_code: str = Field(..., description="Target account display code")
|
|
|
|
|
|
# --- Admin User Creation ---
|
|
|
|
class AdminUserCreate(BaseModel):
|
|
email: EmailStr
|
|
name: str = Field(..., min_length=1, max_length=255)
|
|
account_mode: Literal["existing", "personal"]
|
|
account_display_code: Optional[str] = Field(None, description="Required when account_mode='existing'")
|
|
account_role: Optional[Literal["engineer", "viewer"]] = Field(None, description="Required when account_mode='existing'")
|
|
send_email: bool = True
|
|
|
|
|
|
class AdminUserCreateResponse(BaseModel):
|
|
user: dict # UserResponse fields
|
|
temporary_password: str
|
|
email_sent: bool
|
|
|
|
|
|
# --- Admin Password Reset ---
|
|
|
|
class AdminPasswordReset(BaseModel):
|
|
mode: Literal["email_link", "temp_password"]
|
|
|
|
|
|
class AdminPasswordResetResponse(BaseModel):
|
|
message: str
|
|
temporary_password: Optional[str] = None
|
|
email_sent: bool = False
|
|
|
|
|
|
# --- Hard Delete Precheck ---
|
|
|
|
class HardDeleteCheckResponse(BaseModel):
|
|
can_delete: bool
|
|
blockers: dict
|