Merge pull request ClickHouse#80180 from ClickHouse/ci_praktika_support_for_wf_choice_input

CI: praktika to support choice type in workflow input

Author: maxknv

ci/praktika/info.py

@@ -113,6 +113,10 @@ def is_merge_queue_event(self):
     def is_push_event(self):
         return self.env.EVENT_TYPE == "push"
 
+    @property
+    def is_dispatch_event(self):
+        return self.env.EVENT_TYPE == "dispatch"
+
     @property
     def instance_lifecycle(self):
         return self.env.INSTANCE_LIFE_CYCLE

ci/praktika/job.py

@@ -92,7 +92,7 @@ def parametrize(
                 == len(timeout)
                 == len(provides)
                 == len(requires)
-            ), f"Parametrization lists must be of the same size [{len(parameter)}, {len(runs_on)}, {len(timeout)}, {len(provides)}, {len(requires)}]"
+            ), f"Parametrization lists for job [{self.name}] must be of the same size [{len(parameter)}, {len(runs_on)}, {len(timeout)}, {len(provides)}, {len(requires)}]"
 
             res = []
             for parameter_, runs_on_, timeout_, provides_, requires_ in zip(

ci/praktika/runner.py

@@ -36,7 +36,7 @@ def generate_local_run_environment(workflow, job, pr=None, sha=None):
             BRANCH="branch_name",
             SHA=sha or Shell.get_output("git rev-parse HEAD"),
             PR_NUMBER=pr or -1,
-            EVENT_TYPE="",
+            EVENT_TYPE=workflow.event,
             JOB_OUTPUT_STREAM="",
             EVENT_FILE_PATH="",
             CHANGE_URL="",

ci/praktika/validator.py

@@ -56,7 +56,7 @@ def validate(cls):
             for job in workflow.jobs:
                 cls.evaluate_check(
                     isinstance(job, Job.Config),
-                    f"Invalid job type [{job}]",
+                    f"Invalid job type [{job}]: type [{type(job)}]",
                     workflow.name,
                 )
 
@@ -158,7 +158,7 @@ def validate(cls):
                         artifact.is_s3_artifact()
                     ), f"All artifacts must be of S3 type if enable_cache|enable_html=True, artifact [{artifact.name}], type [{artifact.type}], workflow [{workflow.name}]"
 
-            if workflow.dockers:
+            if workflow.dockers and not workflow.disable_dockers_build:
                 assert (
                     Settings.DOCKERHUB_USERNAME
                 ), f"Settings.DOCKERHUB_USERNAME must be provided if workflow has dockers, workflow [{workflow.name}]"

ci/praktika/workflow.py

@@ -107,3 +107,4 @@ class InputConfig:
             description: str
             is_required: bool
             default_value: str
+            options: Optional[List] = None

ci/praktika/yaml_generator.py

@@ -112,6 +112,14 @@ class Templates:
         default: {DEFAULT_VALUE}\
 """
 
+        TEMPLATE_OPTIONS_INPUT = """
+      {NAME}:
+        description: {DESCRIPTION}
+        type: choice
+        options: {OPTIONS}
+        default: {DEFAULT_VALUE}\
+"""
+
         TEMPLATE_SECRET_CONFIG = """\
       {SECRET_NAME}:
         required: true
@@ -378,12 +386,22 @@ def generate(self):
         # for dispatch workflows only
         dispatch_inputs = ""
         for input_item in self.workflow_config.dispatch_inputs:
-            dispatch_inputs += YamlGenerator.Templates.TEMPLATE_INPUT.format(
-                NAME=input_item.name,
-                DESCRIPTION=input_item.description,
-                IS_REQUIRED="true" if input_item.is_required else "false",
-                DEFAULT_VALUE=input_item.default_value or "''",
-            )
+            if not input_item.options:
+                dispatch_inputs += YamlGenerator.Templates.TEMPLATE_INPUT.format(
+                    NAME=input_item.name,
+                    DESCRIPTION=input_item.description,
+                    IS_REQUIRED="true" if input_item.is_required else "false",
+                    DEFAULT_VALUE=input_item.default_value or "''",
+                )
+            else:
+                dispatch_inputs += (
+                    YamlGenerator.Templates.TEMPLATE_OPTIONS_INPUT.format(
+                        NAME=input_item.name,
+                        DESCRIPTION=input_item.description,
+                        OPTIONS=input_item.options,
+                        DEFAULT_VALUE=input_item.default_value or "''",
+                    )
+                )
 
         if self.workflow_config.event in (
             Workflow.Event.PULL_REQUEST,