41 lines
889 B
Python
41 lines
889 B
Python
from datetime import datetime, date
|
|
from typing import Optional
|
|
from pydantic import BaseModel
|
|
|
|
from app.models.entry import EntryType
|
|
|
|
|
|
class EntryCreate(BaseModel):
|
|
entry_type: EntryType
|
|
date: date
|
|
bourbon_name: Optional[str] = None
|
|
proof: Optional[float] = None
|
|
amount_shots: float = 1.0
|
|
notes: Optional[str] = None
|
|
|
|
|
|
class EntryResponse(BaseModel):
|
|
id: int
|
|
entry_type: EntryType
|
|
date: date
|
|
bourbon_name: Optional[str]
|
|
proof: Optional[float]
|
|
amount_shots: float
|
|
notes: Optional[str]
|
|
created_at: datetime
|
|
|
|
model_config = {"from_attributes": True}
|
|
|
|
|
|
class BottleStats(BaseModel):
|
|
total_add_entries: int
|
|
current_total_shots: float
|
|
estimated_proof: Optional[float]
|
|
|
|
|
|
class PublicUserStats(BaseModel):
|
|
display_name: str
|
|
total_add_entries: int
|
|
current_total_shots: float
|
|
estimated_proof: Optional[float]
|