mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-04-26 01:06:23 +02:00
ff4e0f9b62
Some checks are pending
Build and Push Docker Images / tag_release (push) Waiting to run
Build and Push Docker Images / build (./surfsense_backend, ./surfsense_backend/Dockerfile, backend, surfsense-backend, ubuntu-24.04-arm, linux/arm64, arm64) (push) Blocked by required conditions
Build and Push Docker Images / build (./surfsense_backend, ./surfsense_backend/Dockerfile, backend, surfsense-backend, ubuntu-latest, linux/amd64, amd64) (push) Blocked by required conditions
Build and Push Docker Images / build (./surfsense_web, ./surfsense_web/Dockerfile, web, surfsense-web, ubuntu-24.04-arm, linux/arm64, arm64) (push) Blocked by required conditions
Build and Push Docker Images / build (./surfsense_web, ./surfsense_web/Dockerfile, web, surfsense-web, ubuntu-latest, linux/amd64, amd64) (push) Blocked by required conditions
Build and Push Docker Images / create_manifest (backend, surfsense-backend) (push) Blocked by required conditions
Build and Push Docker Images / create_manifest (web, surfsense-web) (push) Blocked by required conditions
101 lines
2.5 KiB
Python
101 lines
2.5 KiB
Python
"""Schemas for Stripe-backed page purchases."""
|
|
|
|
import uuid
|
|
from datetime import datetime
|
|
|
|
from pydantic import BaseModel, ConfigDict, Field
|
|
|
|
from app.db import PagePurchaseStatus
|
|
|
|
|
|
class CreateCheckoutSessionRequest(BaseModel):
|
|
"""Request body for creating a page-purchase checkout session."""
|
|
|
|
quantity: int = Field(ge=1, le=100)
|
|
search_space_id: int = Field(ge=1)
|
|
|
|
|
|
class CreateCheckoutSessionResponse(BaseModel):
|
|
"""Response containing the Stripe-hosted checkout URL."""
|
|
|
|
checkout_url: str
|
|
|
|
|
|
class StripeStatusResponse(BaseModel):
|
|
"""Response describing Stripe page-buying availability."""
|
|
|
|
page_buying_enabled: bool
|
|
|
|
|
|
class PagePurchaseRead(BaseModel):
|
|
"""Serialized page-purchase record for purchase history."""
|
|
|
|
id: uuid.UUID
|
|
stripe_checkout_session_id: str
|
|
stripe_payment_intent_id: str | None = None
|
|
quantity: int
|
|
pages_granted: int
|
|
amount_total: int | None = None
|
|
currency: str | None = None
|
|
status: PagePurchaseStatus
|
|
completed_at: datetime | None = None
|
|
created_at: datetime
|
|
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
|
|
class PagePurchaseHistoryResponse(BaseModel):
|
|
"""Response containing the authenticated user's page purchases."""
|
|
|
|
purchases: list[PagePurchaseRead]
|
|
|
|
|
|
class StripeWebhookResponse(BaseModel):
|
|
"""Generic acknowledgement for Stripe webhook delivery."""
|
|
|
|
received: bool = True
|
|
|
|
|
|
class CreateTokenCheckoutSessionRequest(BaseModel):
|
|
"""Request body for creating a premium token purchase checkout session."""
|
|
|
|
quantity: int = Field(ge=1, le=100)
|
|
search_space_id: int = Field(ge=1)
|
|
|
|
|
|
class CreateTokenCheckoutSessionResponse(BaseModel):
|
|
"""Response containing the Stripe-hosted checkout URL."""
|
|
|
|
checkout_url: str
|
|
|
|
|
|
class TokenPurchaseRead(BaseModel):
|
|
"""Serialized premium token purchase record."""
|
|
|
|
id: uuid.UUID
|
|
stripe_checkout_session_id: str
|
|
stripe_payment_intent_id: str | None = None
|
|
quantity: int
|
|
tokens_granted: int
|
|
amount_total: int | None = None
|
|
currency: str | None = None
|
|
status: str
|
|
completed_at: datetime | None = None
|
|
created_at: datetime
|
|
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
|
|
class TokenPurchaseHistoryResponse(BaseModel):
|
|
"""Response containing the user's premium token purchases."""
|
|
|
|
purchases: list[TokenPurchaseRead]
|
|
|
|
|
|
class TokenStripeStatusResponse(BaseModel):
|
|
"""Response describing token-buying availability and current quota."""
|
|
|
|
token_buying_enabled: bool
|
|
premium_tokens_used: int = 0
|
|
premium_tokens_limit: int = 0
|
|
premium_tokens_remaining: int = 0
|