88 lines
1.9 KiB
Python
88 lines
1.9 KiB
Python
from datetime import datetime
|
|
from pydantic import BaseModel
|
|
from app.models.deck import DeckMode, CardSlot
|
|
|
|
|
|
class DeckConstraints(BaseModel):
|
|
prefer_owned: bool = False
|
|
budget_enabled: bool = False
|
|
budget_amount: float | None = None
|
|
budget_scope: str = "purchase"
|
|
|
|
|
|
class ExistingCard(BaseModel):
|
|
card_name: str
|
|
slot: str | None = None
|
|
quantity: int = 1
|
|
|
|
|
|
class GenerateRequest(BaseModel):
|
|
commander: str
|
|
playstyle: str | None = None
|
|
name: str | None = None
|
|
description: str | None = None
|
|
constraints: DeckConstraints = DeckConstraints()
|
|
|
|
|
|
class CompleteRequest(BaseModel):
|
|
commander: str
|
|
playstyle: str | None = None
|
|
name: str | None = None
|
|
existing_cards: list[ExistingCard]
|
|
constraints: DeckConstraints = DeckConstraints()
|
|
|
|
|
|
class CullRequest(BaseModel):
|
|
commander: str
|
|
name: str | None = None
|
|
existing_cards: list[ExistingCard]
|
|
target_count: int = 99
|
|
constraints: DeckConstraints = DeckConstraints()
|
|
|
|
|
|
class DeckCardOut(BaseModel):
|
|
id: int
|
|
deck_id: int
|
|
card_name: str
|
|
slot: CardSlot
|
|
quantity: int
|
|
is_owned: bool
|
|
is_commander: bool
|
|
ai_reasoning: str | None
|
|
scryfall_id: str
|
|
scryfall_data: dict | None
|
|
|
|
model_config = {"from_attributes": True}
|
|
|
|
|
|
class DeckOut(BaseModel):
|
|
id: int
|
|
owner_id: int
|
|
name: str
|
|
commander: str
|
|
description: str | None
|
|
mode: DeckMode
|
|
playstyle: str | None
|
|
prefer_owned: bool
|
|
budget_enabled: bool
|
|
budget_amount: float | None
|
|
budget_scope: str
|
|
ai_reasoning: dict | None
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
cards: list[DeckCardOut] = []
|
|
|
|
model_config = {"from_attributes": True}
|
|
|
|
|
|
class DeckSummaryOut(BaseModel):
|
|
id: int
|
|
name: str
|
|
commander: str
|
|
mode: DeckMode
|
|
playstyle: str | None
|
|
created_at: datetime
|
|
card_count: int | None = None
|
|
|
|
model_config = {"from_attributes": True}
|