refactor: split up of schemas in separate files

Author: JanssenBrm

app/database/models/processing_job.py

@@ -5,7 +5,7 @@
 from app.database.db import Base
 from sqlalchemy.orm import Session, Mapped, mapped_column
 
-from app.schemas import ProcessTypeEnum, ProcessingStatusEnum
+from app.schemas.unit_job import ProcessTypeEnum, ProcessingStatusEnum
 
 
 logger = logging.getLogger(__name__)

app/platforms/base.py

@@ -1,6 +1,7 @@
 from abc import ABC, abstractmethod
 
-from app.schemas import ProcessingStatusEnum, ServiceDetails
+from app.schemas.enum import ProcessingStatusEnum
+from app.schemas.unit_job import ServiceDetails
 
 
 class BaseProcessingPlatform(ABC):

app/platforms/dispatcher.py

@@ -4,7 +4,7 @@
 import pkgutil
 from typing import Dict, Type
 from app.platforms.base import BaseProcessingPlatform
-from app.schemas import ProcessTypeEnum
+from app.schemas.enum import ProcessTypeEnum
 
 PROCESSING_PLATFORMS: Dict[ProcessTypeEnum, Type[BaseProcessingPlatform]] = {}
 

app/platforms/implementations/ogc_api_process.py

@@ -2,7 +2,8 @@
 
 from app.platforms.base import BaseProcessingPlatform
 from app.platforms.dispatcher import register_processing_platform
-from app.schemas import ProcessTypeEnum, ProcessingStatusEnum, ServiceDetails
+from app.schemas.enum import ProcessTypeEnum, ProcessingStatusEnum
+from app.schemas.unit_job import ServiceDetails
 
 logger = logging.getLogger(__name__)
 

app/platforms/implementations/openeo.py

@@ -9,11 +9,8 @@
 
 from app.platforms.base import BaseProcessingPlatform
 from app.platforms.dispatcher import register_processing_platform
-from app.schemas import (
-    ProcessTypeEnum,
-    ProcessingStatusEnum,
-    ServiceDetails,
-)
+from app.schemas.enum import ProcessTypeEnum, ProcessingStatusEnum
+from app.schemas.unit_job import ServiceDetails
 
 load_dotenv()
 logger = logging.getLogger(__name__)

app/routers/jobs_status.py

@@ -5,7 +5,7 @@
 from sqlalchemy.orm import Session
 
 from app.database.db import get_db
-from app.schemas import JobsStatusResponse
+from app.schemas.jobs_status import JobsStatusResponse
 from app.services.processing import get_processing_jobs_by_user_id
 
 router = APIRouter()

app/routers/unit_jobs.py

@@ -4,7 +4,7 @@
 from sqlalchemy.orm import Session
 
 from app.database.db import get_db
-from app.schemas import BaseJobRequest, ProcessingJobSummary, ProcessingJob
+from app.schemas.unit_job import BaseJobRequest, ProcessingJob, ProcessingJobSummary
 from app.services.processing import create_processing_job, get_processing_job_by_user_id
 
 # from app.auth import get_current_user

app/schemas.py

@@ -0,0 +1,16 @@
+from enum import Enum
+
+
+class ProcessTypeEnum(str, Enum):
+    OPENEO = "openeo"
+    OGC_API_PROCESS = "ogc_api_process"
+
+
+class ProcessingStatusEnum(str, Enum):
+    CREATED = "created"
+    QUEUED = "queued"
+    RUNNING = "running"
+    FINISHED = "finished"
+    CANCELED = "canceled"
+    FAILED = "failed"
+    UNKNOWN = "unknown"

app/schemas/enum.py

@@ -0,0 +1,10 @@
+from typing import List
+from pydantic import BaseModel
+
+from app.schemas.unit_job import ProcessingJobSummary
+from app.schemas.upscaling_task import UpscalingTaskSummary
+
+
+class JobsStatusResponse(BaseModel):
+    upscaling_tasks: List[UpscalingTaskSummary] = []
+    processing_jobs: List[ProcessingJobSummary] = []