fine tuning models

api/graphql.py

@@ -10,7 +10,7 @@ import os
 
 import graphene
 
-from api.schemas.graphql_types import (
+from api.schemas.graphql import (
     CreateJobInput,
     MediaAssetType,
     ScanResultType,

api/schemas/asset.py

@@ -1,21 +1,16 @@
 """MediaAsset Schemas - GENERATED FILE"""
 
 from datetime import datetime
-from enum import Enum
 from typing import Any, Dict, List, Optional
 from uuid import UUID
 
 from .base import BaseSchema
-
+from .models import AssetStatus
-
-class AssetStatus(str, Enum):
-    PENDING = "pending"
-    READY = "ready"
-    ERROR = "error"
 
 
 class AssetCreate(BaseSchema):
     """AssetCreate schema."""
+
     filename: str
     file_path: str
     file_size: Optional[int] = None
@@ -30,8 +25,10 @@ class AssetCreate(BaseSchema):
     comments: str = ""
     tags: List[str]
 
+
 class AssetUpdate(BaseSchema):
     """AssetUpdate schema."""
+
     filename: Optional[str] = None
     file_path: Optional[str] = None
     status: Optional[AssetStatus] = None
@@ -48,8 +45,10 @@ class AssetUpdate(BaseSchema):
     comments: Optional[str] = None
     tags: Optional[List[str]] = None
 
+
 class AssetResponse(BaseSchema):
     """AssetResponse schema."""
+
     id: UUID
     filename: str
     file_path: str

api/schemas/job.py

@@ -1,19 +1,11 @@
 """TranscodeJob Schemas - GENERATED FILE"""
 
 from datetime import datetime
-from enum import Enum
 from typing import Any, Dict, List, Optional
 from uuid import UUID
 
 from .base import BaseSchema
-
+from .models import JobStatus
-
-class JobStatus(str, Enum):
-    PENDING = "pending"
-    PROCESSING = "processing"
-    COMPLETED = "completed"
-    FAILED = "failed"
-    CANCELLED = "cancelled"
 
 
 class JobCreate(BaseSchema):

ctrl/generate.sh

@@ -32,7 +32,7 @@ python -m modelgen from-schema \
 # Graphene types for GraphQL: domain models + enums + API types
 python -m modelgen from-schema \
     --schema schema/models \
-    --output api/schemas/graphql_types.py \
+    --output api/schemas/graphql.py \
     --targets graphene \
     --include dataclasses,enums,api