127 lines
3.1 KiB
Python
127 lines
3.1 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 .jobs import ChunkJob, ChunkJobStatus, JobStatus, TranscodeJob
|
|
from .detect_jobs import (
|
|
DetectJob, DetectJobStatus, RunType,
|
|
Timeline, Checkpoint,
|
|
BrandSource, KnownBrand, SourceBrandSighting,
|
|
)
|
|
from .stages import StageConfigField, StageIO, StageDefinition, STAGE_VIEWS
|
|
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 .views import ChunkEvent, ChunkOutputFile, PipelineStats, WorkerEvent
|
|
from .sources import ChunkInfo, SourceJob, SourceType
|
|
|
|
# Core domain models - generates Django, SQLModel, TypeScript
|
|
DATACLASSES = [MediaAsset, TranscodePreset, TranscodeJob, ChunkJob,
|
|
DetectJob, Timeline, Checkpoint,
|
|
KnownBrand, SourceBrandSighting]
|
|
|
|
# API request/response models - generates TypeScript only (no Django)
|
|
# WorkerStatus from grpc.py is reused here
|
|
API_MODELS = [
|
|
CreateJobRequest,
|
|
UpdateAssetRequest,
|
|
SystemStatus,
|
|
ScanResult,
|
|
DeleteResult,
|
|
WorkerStatus,
|
|
SourceJob,
|
|
ChunkInfo,
|
|
]
|
|
|
|
# Status enums - included in generated code
|
|
ENUMS = [AssetStatus, JobStatus, ChunkJobStatus, DetectJobStatus, RunType, BrandSource, SourceType]
|
|
|
|
# View/event models - generates TypeScript for UI consumption
|
|
VIEWS = [ChunkEvent, WorkerEvent, PipelineStats, ChunkOutputFile]
|
|
|
|
|
|
# gRPC messages - generates Proto
|
|
GRPC_MESSAGES = [
|
|
JobRequest,
|
|
JobResponse,
|
|
ProgressRequest,
|
|
ProgressUpdate,
|
|
CancelRequest,
|
|
CancelResponse,
|
|
WorkerStatus,
|
|
Empty,
|
|
ChunkStreamRequest,
|
|
ChunkPipelineEvent,
|
|
]
|
|
|
|
__all__ = [
|
|
# Models
|
|
"MediaAsset",
|
|
"TranscodePreset",
|
|
"TranscodeJob",
|
|
"ChunkJob",
|
|
# API Models
|
|
"CreateJobRequest",
|
|
"UpdateAssetRequest",
|
|
"DeleteResult",
|
|
"ScanResult",
|
|
"SystemStatus",
|
|
# Enums
|
|
"AssetStatus",
|
|
"JobStatus",
|
|
"ChunkJobStatus",
|
|
# gRPC
|
|
"GRPC_SERVICE",
|
|
"JobRequest",
|
|
"JobResponse",
|
|
"ProgressRequest",
|
|
"ProgressUpdate",
|
|
"CancelRequest",
|
|
"CancelResponse",
|
|
"WorkerStatus",
|
|
"Empty",
|
|
"ChunkStreamRequest",
|
|
"ChunkPipelineEvent",
|
|
# Views
|
|
"ChunkEvent",
|
|
"WorkerEvent",
|
|
"PipelineStats",
|
|
"ChunkOutputFile",
|
|
# Sources
|
|
"SourceType",
|
|
"SourceJob",
|
|
"ChunkInfo",
|
|
# For generator
|
|
"DATACLASSES",
|
|
"API_MODELS",
|
|
"ENUMS",
|
|
"VIEWS",
|
|
"GRPC_MESSAGES",
|
|
"BUILTIN_PRESETS",
|
|
]
|