Include username in metadata (#81)

dask · Jul 20, 2018 · f746159 · f746159
1 parent 238c31a
commit f746159
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/dask_kubernetes/core.py b/dask_kubernetes/core.py
@@ -193,6 +193,7 @@ def __init__(
         self.pod_template = clean_pod_template(pod_template)
         # Default labels that can't be overwritten
         self.pod_template.metadata.labels['dask.pydata.org/cluster-name'] = name
+        self.pod_template.metadata.labels['user'] = escape(getpass.getuser())
         self.pod_template.metadata.labels['app'] = 'dask'
         self.pod_template.metadata.labels['component'] = 'dask-worker'
         self.pod_template.metadata.namespace = namespace

diff --git a/dask_kubernetes/tests/test_core.py b/dask_kubernetes/tests/test_core.py
@@ -454,6 +454,7 @@ def test_escape_username(pod_spec, loop, ns):
         with KubeCluster(pod_spec, loop=loop, namespace=ns) as cluster:
             assert 'foo' in cluster.name
             assert '!' not in cluster.name
+            assert 'foo' in cluster.pod_template.metadata.labels['user']
     finally:
         os.environ['LOGNAME'] = old_logname