Skip to content

Volume Mounts Wt

 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
from hera.workflows import (
    DAG,
    Parameter,
    Volume,
    WorkflowTemplate,
    models as m,
    script,
)


@script(
    inputs=Parameter(name="vol"),
    volume_mounts=[m.VolumeMount(name="{{inputs.parameters.vol}}", mount_path="/mnt/vol")],
)
def foo():
    import os
    import subprocess

    print(os.listdir("/mnt"))
    print(subprocess.run("cd /mnt && df -h", shell=True, capture_output=True).stdout.decode())


with WorkflowTemplate(
    generate_name="volumes-",
    entrypoint="d",
    volumes=[
        Volume(name="v1", size="1Gi"),
        Volume(name="v2", size="3Gi"),
        Volume(name="v3", size="5Gi"),
    ],
) as w:
    with DAG(name="d"):
        foo(name="v1", arguments=Parameter(name="vol", value="v1"))
        foo(name="v2", arguments=Parameter(name="vol", value="v2"))
        foo(name="v3", arguments=Parameter(name="vol", value="v3"))

 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
apiVersion: argoproj.io/v1alpha1
kind: WorkflowTemplate
metadata:
  generateName: volumes-
spec:
  entrypoint: d
  templates:
  - dag:
      tasks:
      - arguments:
          parameters:
          - name: vol
            value: v1
        name: v1
        template: foo
      - arguments:
          parameters:
          - name: vol
            value: v2
        name: v2
        template: foo
      - arguments:
          parameters:
          - name: vol
            value: v3
        name: v3
        template: foo
    name: d
  - inputs:
      parameters:
      - name: vol
    name: foo
    script:
      command:
      - python
      image: python:3.8
      source: |-
        import os
        import sys
        sys.path.append(os.getcwd())
        import os
        import subprocess
        print(os.listdir('/mnt'))
        print(subprocess.run('cd /mnt && df -h', shell=True, capture_output=True).stdout.decode())
      volumeMounts:
      - mountPath: /mnt/vol
        name: '{{inputs.parameters.vol}}'
  volumeClaimTemplates:
  - metadata:
      name: v1
    spec:
      accessModes:
      - ReadWriteOnce
      resources:
        requests:
          storage: 1Gi
  - metadata:
      name: v2
    spec:
      accessModes:
      - ReadWriteOnce
      resources:
        requests:
          storage: 3Gi
  - metadata:
      name: v3
    spec:
      accessModes:
      - ReadWriteOnce
      resources:
        requests:
          storage: 5Gi

Comments