dag.py 1.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940
  1. from app import crud, models
  2. from app.utils.send_util import *
  3. from app.utils.utils import get_cmd_parameter
  4. from sqlalchemy.orm import Session
  5. def dag_create_job(dag_script: str):
  6. af_task = dag_create_task(dag_script)
  7. af_job = {
  8. "tasks": [af_task],
  9. "name": "123",
  10. "dependence": [],
  11. "cron": "once",
  12. "desc": "123",
  13. "route_strategy": "",
  14. "block_strategy": "",
  15. "executor_timeout": "",
  16. "executor_fail_retry_count": "",
  17. "trigger_status": 1,
  18. "job_type": 0,
  19. "user_id": 0,
  20. }
  21. res = send_post('/jpt/af_job', af_job)
  22. af_job = res['data']
  23. send_submit(af_job['id'])
  24. def dag_create_task(dag_script: str):
  25. af_task = {
  26. "name": "123",
  27. "file_urls": [],
  28. "script": dag_script,
  29. "cmd": "",
  30. "cmd_parameters": "",
  31. "envs": {},
  32. "run_image": "",
  33. "task_type": "sparks",
  34. "user_id": 0,
  35. }
  36. res = send_post('/jpt/af_task', af_task)
  37. af_task = res['data']
  38. return af_task