Skip to content

Workflow On Exit

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
from hera.workflows import DAG, Workflow, WorkflowStatus, script


@script()
def echo(s: str):
    print(s)


with Workflow(generate_name="on-exit-", entrypoint="d") as w:
    with DAG(name="exit-procedure") as exit_procedure:
        (
            echo(name="t3", arguments={"s": 1}).on_workflow_status(WorkflowStatus.succeeded)
            >> echo(name="t4", arguments={"s": 2}).on_workflow_status(WorkflowStatus.succeeded)
        )

        (
            echo(name="t5", arguments={"s": "3"}).on_workflow_status(WorkflowStatus.error)
            >> echo(name="t6", arguments={"s": "4"}).on_workflow_status(WorkflowStatus.error)
        )

    with DAG(name="d"):
        echo(name="t1", arguments={"s": "a"}) >> echo(name="t2", arguments={"s": "b"})
    w.on_exit = exit_procedure

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
apiVersion: argoproj.io/v1alpha1
kind: Workflow
metadata:
  generateName: on-exit-
spec:
  entrypoint: d
  onExit: exit-procedure
  templates:
  - dag:
      tasks:
      - arguments:
          parameters:
          - name: s
            value: '1'
        name: t3
        template: echo
        when: '{{workflow.status}} == Succeeded'
      - arguments:
          parameters:
          - name: s
            value: '2'
        depends: t3
        name: t4
        template: echo
        when: '{{workflow.status}} == Succeeded'
      - arguments:
          parameters:
          - name: s
            value: '3'
        name: t5
        template: echo
        when: '{{workflow.status}} == Error'
      - arguments:
          parameters:
          - name: s
            value: '4'
        depends: t5
        name: t6
        template: echo
        when: '{{workflow.status}} == Error'
    name: exit-procedure
  - inputs:
      parameters:
      - name: s
    name: echo
    script:
      command:
      - python
      image: python:3.8
      source: 'import os

        import sys

        sys.path.append(os.getcwd())

        import json

        try: s = json.loads(r''''''{{inputs.parameters.s}}'''''')

        except: s = r''''''{{inputs.parameters.s}}''''''


        print(s)'
  - dag:
      tasks:
      - arguments:
          parameters:
          - name: s
            value: a
        name: t1
        template: echo
      - arguments:
          parameters:
          - name: s
            value: b
        depends: t1
        name: t2
        template: echo
    name: d

Comments