瀏覽代碼

调试任务增加依赖

liweiquan 2 年之前
父節點
當前提交
76f94d3862
共有 1 個文件被更改,包括 6 次插入0 次删除
  1. 6 0
      app/services/dag.py

+ 6 - 0
app/services/dag.py

@@ -35,6 +35,9 @@ def dag_create_task(dag_uuid:str,dag_script: str,db: Session):
     envs = {}
     requirements_relation = crud.get_requirements_relation(db, dag_uuid)
     if requirements_relation:
+        requirements = crud.get_requirements_status(db, dag_uuid)
+        if requirements.status != 2:
+            raise Exception('依赖未安装成功,不可执行')
         envs.update({'requirement_package_path': f'jpt/requirements/dag_{dag_uuid.lower()})'})
     af_task = {
         "name": "调试作业",
@@ -79,6 +82,9 @@ def dag_put_task(dag_uuid:str,dag_script: str,db: Session,old_af_task):
     envs = {}
     requirements_relation = crud.get_requirements_relation(db, dag_uuid)
     if requirements_relation:
+        requirements = crud.get_requirements_status(db, dag_uuid)
+        if requirements.status != 2:
+            raise Exception('依赖未安装成功,不可执行')
         envs.update({'requirement_package_path': f'jpt/requirements/dag_{dag_uuid.lower()})'})
     af_task = {
         "name": "调试作业",