78 lines
1.9 KiB
Python
78 lines
1.9 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
|
|
from .timeline import Timeline
|
|
from .checkpoint import Checkpoint
|
|
from .stage_output import StageOutput
|
|
from .brand import BrandSource, Brand
|
|
from .media import AssetStatus, MediaAsset
|
|
from .profile import Profile
|
|
from .preset import BUILTIN_PRESETS, TranscodePreset
|
|
from .event import DETECT_VIEWS # noqa: F401
|
|
from .inference import INFERENCE_VIEWS # noqa: F401
|
|
from .ui_state import UI_STATE_VIEWS # noqa: F401
|
|
from .stage import STAGE_VIEWS # noqa: F401
|
|
from .view import ChunkEvent, ChunkOutputFile, PipelineStats, WorkerEvent
|
|
from .source import ChunkInfo, SourceJob, SourceType
|
|
|
|
# Core domain models - generates SQLModel, TypeScript
|
|
DATACLASSES = [MediaAsset, TranscodePreset,
|
|
Job, Timeline, Checkpoint, StageOutput, Brand, Profile]
|
|
|
|
# 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,
|
|
]
|