Add user email to the runtime context

Author: polyaxon-ci

cli/polyaxon/_compiler/contexts/contexts.py

@@ -62,6 +62,7 @@ def resolve_globals_contexts(
     original_uuid: Optional[str] = None,
     is_independent: bool = True,
     username: Optional[str] = None,
+    user_email: Optional[str] = None,
 ) -> Dict:
     resolved_contexts = {
         ctx_sections.GLOBALS: {
@@ -72,6 +73,7 @@ def resolve_globals_contexts(
             ),
             ctx_keys.PROJECT_UUID: project_uuid,
             ctx_keys.USERNAME: username,
+            ctx_keys.USER_EMAIL: user_email,
             ctx_keys.RUN_INFO: get_run_instance(owner_name, project_name, run_uuid),
             ctx_keys.NAME: run_name,
             ctx_keys.UUID: run_uuid,
@@ -141,6 +143,7 @@ def resolve_contexts(
     original_uuid: Optional[str] = None,
     is_independent: bool = True,
     username: Optional[str] = None,
+    user_email: Optional[str] = None,
 ) -> Dict:
     run_kind = compiled_operation.get_run_kind()
     if run_kind not in CONTEXTS_MANAGERS:
@@ -172,6 +175,7 @@ def resolve_contexts(
         original_uuid=original_uuid,
         is_independent=is_independent,
         username=username,
+        user_email=user_email,
     )
 
     return CONTEXTS_MANAGERS[run_kind].resolve(

cli/polyaxon/_compiler/resolver/runtime.py

@@ -47,13 +47,15 @@ def __init__(
         original_uuid: Optional[str] = None,
         is_independent: bool = True,
         username: Optional[str] = None,
+        user_email: Optional[str] = None,
     ):
         if not compiled_operation:
             raise PolyaxonCompilerError("A run spec is required for resolution.")
         self.run = run
         self.compiled_operation = compiled_operation
         self.owner_name = owner_name
         self.username = username
+        self.user_email = user_email
         self.project_name = project_name
         self.project_uuid = project_uuid
         self.project_uuid = project_uuid or project_name
@@ -108,6 +110,7 @@ def resolve_globals_contexts(self):
             project_name=self.project_name,
             project_uuid=self.project_uuid,
             username=self.username,
+            user_email=self.user_email,
             run_uuid=self.run_uuid,
             run_name=self.run_name,
             run_path=self.run_path,
@@ -236,6 +239,7 @@ def _resolve_contexts(self):
             original_uuid=self.original_uuid,
             is_independent=self.is_independent,
             username=self.username,
+            user_email=self.user_email,
         )
 
     def _apply_runtime_contexts(self):

cli/polyaxon/_contexts/keys.py

@@ -3,6 +3,7 @@
 PROJECT_UUID = "project_uuid"
 PROJECT_UNIQUE_NAME = "project_unique_name"
 USERNAME = "username"
+USER_EMAIL = "user_email"
 CONTEXT_PATH = "context_path"
 ARTIFACTS_PATH = "artifacts_path"
 ITERATION = "iteration"

cli/tests/test_compiler/test_contexts.py

@@ -47,6 +47,7 @@ def test_resolver_default_contexts(self):
             created_at=None,
             compiled_at=None,
             username="user",
+            user_email="user@local.com",
         )
         assert spec == {
             "globals": {
@@ -72,6 +73,7 @@ def test_resolver_default_contexts(self):
                 "is_independent": True,
                 "store_path": "",
                 "username": "user",
+                "user_email": "user@local.com",
             },
             "init": {},
             "connections": {},
@@ -149,6 +151,7 @@ def test_resolver_init_and_connections_contexts(self):
                 "duration": None,
                 "is_independent": False,
                 "username": None,
+                "user_email": None,
                 "cloning_kind": V1CloningKind.COPY,
                 "original_uuid": "uuid-copy",
                 "store_path": "/claim/path",
@@ -227,6 +230,7 @@ def test_resolver_outputs_collections(self):
                 "original_uuid": None,
                 "is_independent": True,
                 "username": None,
+                "user_email": None,
                 "store_path": "",
             },
             "init": {"test_claim": store.schema_.to_dict()},
@@ -302,6 +306,7 @@ def test_resolver_mount_artifacts_store(self):
                 "original_uuid": None,
                 "is_independent": True,
                 "username": None,
+                "user_email": None,
                 "store_path": "/claim/path",
             },
             "init": {"test_claim": store.schema_.to_dict()},
@@ -374,6 +379,7 @@ def test_resolver_default_service_ports(self):
                 "original_uuid": None,
                 "is_independent": True,
                 "username": None,
+                "user_email": None,
                 "store_path": "",
             },
             "init": {},