ADMP-LS / servers /Review /build_image.py
jackkuo's picture
reinit repo
82bf89e
import subprocess
from datetime import datetime
from pathlib import Path
from typing import Annotated
import typer
def main(
tag: Annotated[str | None, typer.Option(help="Image tag")] = None,
dockerfile: Annotated[Path | None, typer.Option(help="Dockerfile path")] = None,
docker_cmd: Annotated[str, typer.Option(help="docker command")] = "docker",
):
git_hash = subprocess.check_output(["git", "rev-parse", "HEAD"]).decode("UTF-8").strip()[:8]
timestamp = datetime.now().strftime("%Y%m%d")
if tag is None:
tag_val = f"{timestamp}-{git_hash}"
else:
tag_val = tag
if dockerfile is None:
# Default to Dockerfile next to this script
dockerfile = Path(__file__).parent / "Dockerfile"
image_name = f"review_mcp:{tag_val}"
print(f"[bold green]Building image {image_name}[/bold green]")
subprocess.run(
[
docker_cmd,
"build",
"-t",
image_name,
"-f",
str(dockerfile),
str(Path(__file__).parent),
],
check=True,
)
print(f"[bold blue]Built image {image_name}[/bold blue]")
if __name__ == "__main__":
typer.run(main)