46 lines
1.1 KiB
Python
46 lines
1.1 KiB
Python
from pydantic import BaseModel
|
|
from datetime import datetime
|
|
from typing import Optional, List
|
|
|
|
|
|
class ExportFilter(BaseModel):
|
|
min_images_per_species: int = 100
|
|
licenses: Optional[List[str]] = None # None means all
|
|
min_quality: Optional[float] = None
|
|
species_ids: Optional[List[int]] = None # None means all
|
|
|
|
|
|
class ExportCreate(BaseModel):
|
|
name: str
|
|
filter_criteria: ExportFilter
|
|
train_split: float = 0.8
|
|
|
|
|
|
class ExportResponse(BaseModel):
|
|
id: int
|
|
name: str
|
|
filter_criteria: Optional[str] = None
|
|
train_split: float
|
|
status: str
|
|
file_path: Optional[str] = None
|
|
file_size: Optional[int] = None
|
|
species_count: Optional[int] = None
|
|
image_count: Optional[int] = None
|
|
created_at: datetime
|
|
completed_at: Optional[datetime] = None
|
|
error_message: Optional[str] = None
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class ExportListResponse(BaseModel):
|
|
items: List[ExportResponse]
|
|
total: int
|
|
|
|
|
|
class ExportPreview(BaseModel):
|
|
species_count: int
|
|
image_count: int
|
|
estimated_size_mb: float
|