75 lines
1.6 KiB
Python
75 lines
1.6 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, ScanResult, SystemStatus
|
|
from .grpc import (
|
|
GRPC_SERVICE,
|
|
CancelRequest,
|
|
CancelResponse,
|
|
Empty,
|
|
JobRequest,
|
|
JobResponse,
|
|
ProgressRequest,
|
|
ProgressUpdate,
|
|
WorkerStatus,
|
|
)
|
|
from .jobs import JobStatus, TranscodeJob
|
|
from .media import AssetStatus, MediaAsset
|
|
from .presets import BUILTIN_PRESETS, TranscodePreset
|
|
|
|
# Core domain models - generates Django, Pydantic, TypeScript
|
|
DATACLASSES = [MediaAsset, TranscodePreset, TranscodeJob]
|
|
|
|
# API request/response models - generates TypeScript only (no Django)
|
|
# WorkerStatus from grpc.py is reused here
|
|
API_MODELS = [CreateJobRequest, SystemStatus, ScanResult, WorkerStatus]
|
|
|
|
# Status enums - included in generated code
|
|
ENUMS = [AssetStatus, JobStatus]
|
|
|
|
# gRPC messages - generates Proto
|
|
GRPC_MESSAGES = [
|
|
JobRequest,
|
|
JobResponse,
|
|
ProgressRequest,
|
|
ProgressUpdate,
|
|
CancelRequest,
|
|
CancelResponse,
|
|
WorkerStatus,
|
|
Empty,
|
|
]
|
|
|
|
__all__ = [
|
|
# Models
|
|
"MediaAsset",
|
|
"TranscodePreset",
|
|
"TranscodeJob",
|
|
# API Models
|
|
"CreateJobRequest",
|
|
"ScanResult",
|
|
"SystemStatus",
|
|
# Enums
|
|
"AssetStatus",
|
|
"JobStatus",
|
|
# gRPC
|
|
"GRPC_SERVICE",
|
|
"JobRequest",
|
|
"JobResponse",
|
|
"ProgressRequest",
|
|
"ProgressUpdate",
|
|
"CancelRequest",
|
|
"CancelResponse",
|
|
"WorkerStatus",
|
|
"Empty",
|
|
# For generator
|
|
"DATACLASSES",
|
|
"API_MODELS",
|
|
"ENUMS",
|
|
"GRPC_MESSAGES",
|
|
"BUILTIN_PRESETS",
|
|
]
|