33 lines
1019 B
Python
33 lines
1019 B
Python
|
from services.workflow.entities import WorkflowScheduleCFSPlanEntity
|
||
|
|
from services.workflow.scheduler import CFSPlanScheduler, SchedulerCommand
|
||
|
|
from tasks.workflow_cfs_scheduler.entities import AsyncWorkflowQueue
|
||
|
|
|
||
|
|
|
||
|
|
class AsyncWorkflowCFSPlanEntity(WorkflowScheduleCFSPlanEntity):
|
||
|
|
"""
|
||
|
|
Trigger workflow CFS plan entity.
|
||
|
|
"""
|
||
|
|
|
||
|
|
queue: AsyncWorkflowQueue
|
||
|
|
|
||
|
|
|
||
|
|
class AsyncWorkflowCFSPlanScheduler(CFSPlanScheduler):
|
||
|
|
"""
|
||
|
|
Trigger workflow CFS plan scheduler.
|
||
|
|
"""
|
||
|
|
|
||
|
|
plan: AsyncWorkflowCFSPlanEntity
|
||
|
|
|
||
|
|
def can_schedule(self) -> SchedulerCommand:
|
||
|
|
"""
|
||
|
|
Check if the workflow can be scheduled.
|
||
|
|
"""
|
||
|
|
if self.plan.queue in [AsyncWorkflowQueue.PROFESSIONAL_QUEUE, AsyncWorkflowQueue.TEAM_QUEUE]:
|
||
|
|
"""
|
||
|
|
permitted all paid users to schedule the workflow any time
|
||
|
|
"""
|
||
|
|
return SchedulerCommand.NONE
|
||
|
|
|
||
|
|
# FIXME: avoid the sandbox user's workflow at a running state for ever
|
||
|
|
return SchedulerCommand.RESOURCE_LIMIT_REACHED
|