Merge pull request #3 from flyteorg/hash-execution-name

hash the execution name

flyte_provider/operators/flyte.py

@@ -1,5 +1,5 @@
+import hashlib
 import inspect
-import re
 from dataclasses import fields
 from typing import TYPE_CHECKING, Any, Dict, List, Optional, Sequence, Union
 
@@ -210,16 +210,15 @@ def execute(self, context: "Context") -> str:
         """Trigger an execution."""
 
         # create a deterministic execution name
-        task_id = re.sub(r"[\W_]+", "", context["task"].task_id)[:4] + str(
-            context["task_instance"].try_number
+        unhashed_execution_name = (
+            context["task"].dag_id
+            + context["task"].task_id
+            + context["dag_run"].run_id
+            + str(context["task_instance"].try_number)
         )
+
         self.execution_name = (
-            task_id
-            + re.sub(
-                r"[\W_t]+",
-                "",
-                context["dag_run"].run_id.split("__")[-1].lower(),
-            )[: (20 - len(task_id))]
+            "a" + hashlib.md5(unhashed_execution_name.encode()).hexdigest()[:19]
         )
 
         hook = FlyteHook(

tests/hooks/test_flyte.py

@@ -14,7 +14,7 @@
 class TestFlyteHook(unittest.TestCase):
 
     flyte_conn_id = "flyte_default"
-    execution_name = "flyt1202203301338565"
+    execution_name = "a038d0f1339c4e7700c0"
     conn_type = "flyte"
     host = "localhost"
     port = "30081"

tests/operators/test_flyte.py

@@ -25,7 +25,7 @@ class TestFlyteOperator(unittest.TestCase):
     labels = {"key1": "value1"}
     version = "v1"
     inputs = {"name": "hello world"}
-    execution_name = "test1202203301355087"
+    execution_name = "a038d0f1339c4e7700c0"
     oauth2_client = {"client_id": "123", "client_secret": "456"}
     secrets = [{"group": "secrets", "key": "123"}]
     notifications = [{"phases": [1], "email": {"recipients_email": ["[email protected]"]}}]

tests/sensors/test_flyte.py

@@ -21,7 +21,7 @@ class TestFlyteSensor(unittest.TestCase):
     port = "30081"
     project = "flytesnacks"
     domain = "development"
-    execution_name = "test1202203301355081"
+    execution_name = "a038d0f1339c4e7700c0"
 
     @classmethod
     def get_connection(cls):