Make it possible to select PR vs auto jobs

Author: Kobzol

src/ci/github-actions/ci.py

@@ -30,7 +30,7 @@
 defadd_job_properties(jobs: List[Dict], prefix: str) ->List[Job]:
 """
  Modify the `name` attribute of each job, based on its base name and the given `prefix`.
- Add an `image` attribute to each job, base don its image.
+ Add an `image` attribute to each job, based on its image.
  """
 modified_jobs= []
 forjobinjobs:
@@ -196,15 +196,14 @@ def get_job_image(job) -> str:
 returnjob.get("env", {}).get("IMAGE", job["name"])
 
 
-defrun_workflow_locally(job_data: Dict[str, Any], job_name: str):
+defrun_workflow_locally(job_data: Dict[str, Any], job_name: str, pr_jobs: bool):
 DOCKER_DIR=Path(__file__).absolute().parent.parent/"docker"
 
-jobs=list(job_data["auto"])
-jobs.extend(job_data["pr"])
-
+jobs=job_data["pr"] ifpr_jobselsejob_data["auto"]
 jobs= [jobforjobinjobsifjob.get("name") ==job_name]
 iflen(jobs) ==0:
-raiseException(f"Job `{job_name}` not found")
+raiseException(f"Job `{job_name}` not found in {'pr'ifpr_jobselse'auto'} jobs")
+assertlen(jobs) ==1
 job=jobs[0]
 if"ubuntu"notinjob["os"]:
 raiseException("Only Linux jobs can be executed locally")
@@ -222,7 +221,12 @@ def run_workflow_locally(job_data: Dict[str, Any], job_name: str):
 
 env=os.environ.copy()
 env.update(custom_env)
-subprocess.run(args, env=env)
+
+process=subprocess.Popen(args, env=env)
+try:
+process.wait()
+exceptKeyboardInterrupt:
+process.kill()
 
 
 if__name__=="__main__":
@@ -239,7 +243,16 @@ def run_workflow_locally(job_data: Dict[str, Any], job_name: str):
 subparsers=parser.add_subparsers(help="Command to execute", dest="command", required=True)
 subparsers.add_parser("calculate-job-matrix")
 run_parser=subparsers.add_parser("run-local")
-run_parser.add_argument("job_name", help="CI job that should be executed")
+run_parser.add_argument(
+"job_name",
+help="CI job that should be executed. By default, a merge (auto) "
+"job with the given name will be executed"
+ )
+run_parser.add_argument(
+"--pr",
+action="store_true",
+help="Run a PR job instead of an auto job"
+ )
 args=parser.parse_args()
 
 ifargs.command=="calculate-job-matrix":
@@ -265,6 +278,6 @@ def run_workflow_locally(job_data: Dict[str, Any], job_name: str):
 print(f"jobs={json.dumps(jobs)}")
 print(f"run_type={run_type}")
 elifargs.command=="run-local":
-run_workflow_locally(data, args.job_name)
+run_workflow_locally(data, args.job_name, args.pr)
 else:
 raiseException(f"Unknown command {args.command}")