bjoernager/rust
bjoernager/rust
1
Fork 0
Code Activity

Make it possible to select PR vs auto jobs

Browse source
This commit is contained in:
Jakub Beránek 2024-12-31 10:18:02 +01:00 • committed by Jakub Beránek
parent e62d1e46bb
commit 0fc2f14c0c
1 changed files with 22 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

29
src/ci/github-actions/ci.py
View file

@ -196,15 +196,14 @@ def get_job_image(job) -> str:
return job.get("env", {}).get("IMAGE", job["name"]) return job.get("env", {}).get("IMAGE", job["name"])
def run_workflow_locally(job_data: Dict[str, Any], job_name: str): def run_workflow_locally(job_data: Dict[str, Any], job_name: str, pr_jobs: bool):
DOCKER_DIR = Path(__file__).absolute().parent.parent / "docker" DOCKER_DIR = Path(__file__).absolute().parent.parent / "docker"
jobs = list(job_data["auto"]) jobs = job_data["pr"] if pr_jobs else job_data["auto"]
jobs.extend(job_data["pr"])
jobs = [job for job in jobs if job.get("name") == job_name] jobs = [job for job in jobs if job.get("name") == job_name]
if len(jobs) == 0: if len(jobs) == 0:
raise Exception(f"Job `{job_name}` not found") raise Exception(f"Job `{job_name}` not found in {'pr' if pr_jobs else 'auto'} jobs")
assert len(jobs) == 1
job = jobs[0] job = jobs[0]
if "ubuntu" not in job["os"]: if "ubuntu" not in job["os"]:
raise Exception("Only Linux jobs can be executed locally") raise Exception("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 = os.environ.copy()
env.update(custom_env) env.update(custom_env)
subprocess.run(args, env=env)
process = subprocess.Popen(args, env=env)
try:
process.wait()
except KeyboardInterrupt:
process.kill()
if __name__ == "__main__": if __name__ == "__main__":
@ -239,7 +243,16 @@ if __name__ == "__main__":
subparsers = parser.add_subparsers(help="Command to execute", dest="command", required=True) subparsers = parser.add_subparsers(help="Command to execute", dest="command", required=True)
subparsers.add_parser("calculate-job-matrix") subparsers.add_parser("calculate-job-matrix")
run_parser = subparsers.add_parser("run-local") 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() args = parser.parse_args()
if args.command == "calculate-job-matrix": if args.command == "calculate-job-matrix":
@ -265,6 +278,6 @@ if __name__ == "__main__":
print(f"jobs={json.dumps(jobs)}") print(f"jobs={json.dumps(jobs)}")
print(f"run_type={run_type}") print(f"run_type={run_type}")
elif args.command == "run-local": elif args.command == "run-local":
run_workflow_locally(data, args.job_name) run_workflow_locally(data, args.job_name, args.pr)
else: else:
raise Exception(f"Unknown command {args.command}") raise Exception(f"Unknown command {args.command}")