mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-08-30 04:27:40 +08:00
69 lines
2.7 KiB
Python
69 lines
2.7 KiB
Python
"""工作流领域的请求级 command/query 依赖。"""
|
|
|
|
from typing import cast
|
|
|
|
from fastapi import Depends
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
from sqlalchemy.orm import Session
|
|
|
|
from app.adapters.external.server import MoviePilotServerHelper
|
|
from app.api.context import get_async_session, get_host_runtime, get_sync_session
|
|
from app.application.scheduling import Scheduler
|
|
from app.application.workflow import (
|
|
WorkflowCachePort,
|
|
WorkflowDefinitionCommand,
|
|
WorkflowMutationCommand,
|
|
WorkflowQueryService,
|
|
get_workflow_manager,
|
|
)
|
|
from app.runtime.config import global_vars
|
|
from app.startup.composition.context import HostRuntime
|
|
|
|
|
|
def get_workflow_mutation_command(
|
|
db: Session = Depends(get_sync_session),
|
|
runtime: HostRuntime = Depends(get_host_runtime),
|
|
) -> WorkflowMutationCommand:
|
|
"""组装请求级工作流写用例和提交后的调度副作用。"""
|
|
scheduler = Scheduler()
|
|
workflow_manager = get_workflow_manager()
|
|
system_config = cast(WorkflowCachePort, runtime.workflow.system_config())
|
|
return WorkflowMutationCommand(
|
|
repository=runtime.workflow.repository(db),
|
|
unit_of_work=runtime.persistence.sync_transaction(db),
|
|
add_timer=scheduler.update_workflow_job,
|
|
remove_timer=scheduler.remove_workflow_job,
|
|
load_event=workflow_manager.load_workflow_events,
|
|
remove_event=workflow_manager.remove_workflow_event,
|
|
refresh_event=workflow_manager.update_workflow_event,
|
|
stop_running=global_vars.stop_workflow,
|
|
delete_cache=lambda workflow_id: system_config.delete(
|
|
f"WorkflowCache-{workflow_id}"
|
|
),
|
|
)
|
|
|
|
|
|
def get_workflow_definition_command(
|
|
db: AsyncSession = Depends(get_async_session),
|
|
runtime: HostRuntime = Depends(get_host_runtime),
|
|
) -> WorkflowDefinitionCommand:
|
|
"""组装工作流创建、复用和重置的异步写用例。"""
|
|
system_config = cast(WorkflowCachePort, runtime.workflow.system_config())
|
|
return WorkflowDefinitionCommand(
|
|
repository=runtime.workflow.repository(db),
|
|
unit_of_work=runtime.persistence.async_transaction(db),
|
|
stop_running=global_vars.stop_workflow,
|
|
async_delete_cache=lambda workflow_id: system_config.async_delete(
|
|
f"WorkflowCache-{workflow_id}"
|
|
),
|
|
report_fork=MoviePilotServerHelper.async_workflow_fork_by_id,
|
|
)
|
|
|
|
|
|
def get_workflow_query_service(
|
|
db: AsyncSession = Depends(get_async_session),
|
|
runtime: HostRuntime = Depends(get_host_runtime),
|
|
) -> WorkflowQueryService:
|
|
"""组装工作流只读查询用例，避免端点直接持有数据库操作器。"""
|
|
return WorkflowQueryService(repository=runtime.workflow.repository(db))
|