131 lines
2.8 KiB
Python
131 lines
2.8 KiB
Python
"""
|
|
MPR Schema Models
|
|
|
|
This module exports all dataclasses, enums, and constants that the generator
|
|
should process. Add new models here to have them included in generation.
|
|
"""
|
|
|
|
from .api import (
|
|
CreateJobRequest,
|
|
DeleteResult,
|
|
ScanResult,
|
|
SystemStatus,
|
|
UpdateAssetRequest,
|
|
)
|
|
from .grpc import (
|
|
GRPC_SERVICE,
|
|
CancelRequest,
|
|
CancelResponse,
|
|
ChunkPipelineEvent,
|
|
ChunkStreamRequest,
|
|
Empty,
|
|
JobRequest,
|
|
JobResponse,
|
|
ProgressRequest,
|
|
ProgressUpdate,
|
|
WorkerStatus,
|
|
)
|
|
from .job import (
|
|
Job, JobStatus, RunType,
|
|
Timeline, Checkpoint,
|
|
BrandSource, Brand,
|
|
)
|
|
from .media import AssetStatus, MediaAsset
|
|
from .presets import BUILTIN_PRESETS, TranscodePreset
|
|
from .detect import DETECT_VIEWS # noqa: F401 — discovered by modelgen generic loader
|
|
from .inference import INFERENCE_VIEWS # noqa: F401 — GPU inference server API types
|
|
from .ui_state import UI_STATE_VIEWS # noqa: F401 — UI store state types
|
|
from .stages import StageConfigField, StageIO, StageDefinition, STAGE_VIEWS # noqa: F401
|
|
from .views import ChunkEvent, ChunkOutputFile, PipelineStats, WorkerEvent
|
|
from .sources import ChunkInfo, SourceJob, SourceType
|
|
|
|
# Core domain models - generates SQLModel, TypeScript
|
|
DATACLASSES = [MediaAsset, TranscodePreset,
|
|
Job, Timeline, Checkpoint, Brand]
|
|
|
|
# API request/response models
|
|
API_MODELS = [
|
|
CreateJobRequest,
|
|
UpdateAssetRequest,
|
|
SystemStatus,
|
|
ScanResult,
|
|
DeleteResult,
|
|
WorkerStatus,
|
|
SourceJob,
|
|
ChunkInfo,
|
|
]
|
|
|
|
# Status enums
|
|
ENUMS = [AssetStatus, JobStatus, RunType, BrandSource, SourceType]
|
|
|
|
# View/event models
|
|
VIEWS = [ChunkEvent, WorkerEvent, PipelineStats, ChunkOutputFile]
|
|
|
|
# gRPC messages
|
|
GRPC_MESSAGES = [
|
|
JobRequest,
|
|
JobResponse,
|
|
ProgressRequest,
|
|
ProgressUpdate,
|
|
CancelRequest,
|
|
CancelResponse,
|
|
WorkerStatus,
|
|
Empty,
|
|
ChunkStreamRequest,
|
|
ChunkPipelineEvent,
|
|
]
|
|
|
|
__all__ = [
|
|
# Models
|
|
"MediaAsset",
|
|
"TranscodePreset",
|
|
"Job",
|
|
"Timeline",
|
|
"Checkpoint",
|
|
"KnownBrand",
|
|
"SourceBrandSighting",
|
|
# Enums
|
|
"AssetStatus",
|
|
"JobStatus",
|
|
"RunType",
|
|
"BrandSource",
|
|
"SourceType",
|
|
# Stages
|
|
"StageConfigField",
|
|
"StageIO",
|
|
"StageDefinition",
|
|
# API
|
|
"CreateJobRequest",
|
|
"UpdateAssetRequest",
|
|
"DeleteResult",
|
|
"ScanResult",
|
|
"SystemStatus",
|
|
# gRPC
|
|
"GRPC_SERVICE",
|
|
"JobRequest",
|
|
"JobResponse",
|
|
"ProgressRequest",
|
|
"ProgressUpdate",
|
|
"CancelRequest",
|
|
"CancelResponse",
|
|
"WorkerStatus",
|
|
"Empty",
|
|
"ChunkStreamRequest",
|
|
"ChunkPipelineEvent",
|
|
# Views
|
|
"ChunkEvent",
|
|
"WorkerEvent",
|
|
"PipelineStats",
|
|
"ChunkOutputFile",
|
|
# Sources
|
|
"SourceJob",
|
|
"ChunkInfo",
|
|
# For generator
|
|
"DATACLASSES",
|
|
"API_MODELS",
|
|
"ENUMS",
|
|
"VIEWS",
|
|
"GRPC_MESSAGES",
|
|
"BUILTIN_PRESETS",
|
|
]
|