Harvey
62 lines
1.4 KiB
Python
62 lines
1.4 KiB
Python
from typing import Annotated
|
|
import msgspec
|
|
from models.request import QueueItem as QueueItem
|
|
from infrastructure.msgspec_fastapi import AppStruct
|
|
|
|
|
|
class AlbumRequest(AppStruct):
|
|
musicbrainz_id: str
|
|
artist: str | None = None
|
|
album: str | None = None
|
|
year: int | None = None
|
|
artist_mbid: str | None = None
|
|
monitor_artist: bool = False
|
|
auto_download_artist: bool = False
|
|
|
|
|
|
class RequestAcceptedResponse(AppStruct):
|
|
success: bool
|
|
message: str
|
|
musicbrainz_id: str
|
|
status: str = "pending"
|
|
|
|
|
|
class QueueStatusResponse(AppStruct):
|
|
queue_size: int
|
|
processing: bool
|
|
active_workers: int = 0
|
|
max_workers: int = 1
|
|
|
|
|
|
class BatchAlbumItem(AppStruct):
|
|
musicbrainz_id: str
|
|
artist_name: str = "Unknown"
|
|
album_title: str = "Unknown"
|
|
year: int | None = None
|
|
artist_mbid: str | None = None
|
|
|
|
|
|
class BatchAlbumRequest(AppStruct):
|
|
items: Annotated[list[BatchAlbumItem], msgspec.Meta(max_length=500)]
|
|
monitor_artist: bool = False
|
|
auto_download_artist: bool = False
|
|
|
|
|
|
class BatchRequestResponse(AppStruct):
|
|
success: bool
|
|
message: str
|
|
requested: int = 0
|
|
skipped: int = 0
|
|
overflow: int = 0
|
|
|
|
|
|
class BatchCancelRequest(AppStruct):
|
|
musicbrainz_ids: Annotated[list[str], msgspec.Meta(max_length=500)]
|
|
|
|
|
|
class BatchCancelResponse(AppStruct):
|
|
success: bool
|
|
cancelled: int = 0
|
|
failed: int = 0
|
|
message: str = ""
|