cuebook/cuelake

import json
import datetime as dt
import dateutil.parser as dp
import requests
import polling
from celery import shared_task
from django.conf import settings

from system.services import NotificationServices
from workflows.taskUtils import TaskUtils
from workflows.models import Workflow, WorkflowRunLogs, STATUS_QUEUED, STATUS_ERROR, STATUS_ALWAYS

import logging

# Get an instance of a logger
logger = logging.getLogger(__name__)

@shared_task
def runWorkflowJob(workflowId: int, workflowRunLogsId: int = None):
    """
    Celery task to run a Workflow
    :param workflowId: ID of Workflows.workflow model
    """
    try:
        workflowRunStatus = TaskUtils.runWorkflow(workflowId=workflowId, workflowRunLogsId=workflowRunLogsId, taskId=runWorkflowJob.request.id if runWorkflowJob.request.id else "")
        dependentWorkflowIds = list(
            Workflow.objects.filter(
                triggerWorkflow_id=workflowId,
                triggerWorkflowStatus__in=[STATUS_ALWAYS, workflowRunStatus],
            ).values_list("id", flat=True)
        )
        for workflowId in dependentWorkflowIds:
            workflowRun = WorkflowRunLogs.objects.create(workflow_id=workflowId, status=STATUS_QUEUED)
            runWorkflowJob.delay(workflowId=workflowId, workflowRunLogsId=workflowRun.id)
    except Exception as ex:
        WorkflowRunLogs.objects.filter(id=workflowRunLogsId).update(status=STATUS_ERROR, endTimestamp=dt.datetime.now())
        print(str(ex))