36 lines
916 B
Python
36 lines
916 B
Python
from pydantic import BaseModel
|
|
from datetime import datetime
|
|
from typing import Optional, List
|
|
|
|
|
|
class JobCreate(BaseModel):
|
|
name: str
|
|
source: str
|
|
species_ids: Optional[List[int]] = None # None means all species
|
|
only_without_images: bool = False # If True, only scrape species with 0 images
|
|
max_images: Optional[int] = None # If set, only scrape species with fewer than N images
|
|
|
|
|
|
class JobResponse(BaseModel):
|
|
id: int
|
|
name: str
|
|
source: str
|
|
species_filter: Optional[str] = None
|
|
status: str
|
|
progress_current: int
|
|
progress_total: int
|
|
images_downloaded: int
|
|
images_rejected: int
|
|
started_at: Optional[datetime] = None
|
|
completed_at: Optional[datetime] = None
|
|
error_message: Optional[str] = None
|
|
created_at: datetime
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class JobListResponse(BaseModel):
|
|
items: List[JobResponse]
|
|
total: int
|