Sfoglia il codice sorgente

调试任务增加依赖

liweiquan 2 anni fa
parent
commit
76f94d3862
1 ha cambiato i file con 6 aggiunte e 0 eliminazioni
  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": "调试作业",