Files
docker-stacks/docs/using/recipe_code/generate_matrix.py
2024-08-31 22:32:23 +01:00

27 lines
755 B
Python
Executable File

#!/usr/bin/env python3
# Copyright (c) Jupyter Development Team.
# Distributed under the terms of the Modified BSD License.
import json
import os
from pathlib import Path
from typing import Any
THIS_DIR = Path(__file__).parent.resolve()
REPOSITORY_OWNER = os.environ["REPOSITORY_OWNER"]
def generate_matrix() -> dict[str, Any]:
dockerfiles = sorted(file.name for file in THIS_DIR.glob("*.dockerfile"))
runs_on = ["ubuntu-latest"]
if REPOSITORY_OWNER == "jupyter":
runs_on.append("ARM64")
return {
"dockerfile": dockerfiles,
"runs-on": runs_on,
"exclude": [{"dockerfile": "oracledb.dockerfile", "runs-on": "ARM64"}],
}
if __name__ == "__main__":
print("matrix=" + json.dumps(generate_matrix()))