Michael Chihlas
d05b475a41
Task 30 of self-serve signup Phase 2. Super-admins can now manage Stripe IDs, display names, prices, and public/archived flags via the existing admin plan-limits endpoints. - GET /admin/plan-limits now outer-joins plan_billing and returns merged PlanLimitWithBillingResponse rows. Plans without a plan_billing row return None for the billing fields. - PUT /admin/plan-limits accepts the new optional billing fields and upserts plan_billing in the same transaction. If no plan_billing row exists for the plan and the body includes any billing field, a row is created (display_name defaults to plan.capitalize() when omitted; display_name is never NULLed out on an existing row). - After commit, the handler queries account_ids on the affected plan and calls BillingService.invalidate_billing_cache(account_ids). This is a no-op stub today (logs only) — there's no in-process billing cache yet. TODO comment marks the wire-up point. - 3 new integration tests cover GET-with-billing-present, PUT creating a plan_billing row, and the invalidation hook being awaited with a list of account_ids. Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
378 lines
9.7 KiB
Python
378 lines
9.7 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"
|
|
owner_email: Optional[EmailStr] = Field(None, description="Email of an existing user to set as owner")
|
|
|
|
|
|
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 PlanLimitWithBillingResponse(PlanLimitResponse):
|
|
"""PlanLimits + plan_billing fields merged. Billing fields are None when no
|
|
plan_billing row exists for the plan yet."""
|
|
display_name: Optional[str] = None
|
|
description: Optional[str] = None
|
|
monthly_price_cents: Optional[int] = None
|
|
annual_price_cents: Optional[int] = None
|
|
stripe_product_id: Optional[str] = None
|
|
stripe_monthly_price_id: Optional[str] = None
|
|
stripe_annual_price_id: Optional[str] = None
|
|
is_public: Optional[bool] = None
|
|
is_archived: Optional[bool] = None
|
|
sort_order: Optional[int] = None
|
|
|
|
|
|
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"])
|
|
# plan_billing fields — all optional, partial-update semantics. If any are
|
|
# set in the body, the admin endpoint upserts the plan_billing row in the
|
|
# same transaction.
|
|
display_name: Optional[str] = None
|
|
description: Optional[str] = None
|
|
monthly_price_cents: Optional[int] = None
|
|
annual_price_cents: Optional[int] = None
|
|
stripe_product_id: Optional[str] = None
|
|
stripe_monthly_price_id: Optional[str] = None
|
|
stripe_annual_price_id: Optional[str] = None
|
|
is_public: Optional[bool] = None
|
|
is_archived: Optional[bool] = None
|
|
sort_order: Optional[int] = None
|
|
|
|
|
|
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["owner", "admin", "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
|