mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-09-20 08:40:50 +00:00
87 lines
2.6 KiB
Python
87 lines
2.6 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import os
|
|
import json
|
|
import time
|
|
import subprocess
|
|
import logging
|
|
|
|
from typing import List, Optional
|
|
|
|
|
|
class DockerImage:
|
|
def __init__(self, name: str, version: Optional[str] = None):
|
|
self.name = name
|
|
if version is None:
|
|
self.version = "latest"
|
|
else:
|
|
self.version = version
|
|
|
|
def __str__(self):
|
|
return f"{self.name}:{self.version}"
|
|
|
|
|
|
def get_images_with_versions(
|
|
reports_path: str,
|
|
required_images: List[str],
|
|
pull: bool = True,
|
|
version: Optional[str] = None,
|
|
) -> List[DockerImage]:
|
|
images_path = None
|
|
for root, _, files in os.walk(reports_path):
|
|
for f in files:
|
|
if f == "changed_images.json":
|
|
images_path = os.path.join(root, "changed_images.json")
|
|
break
|
|
|
|
if not images_path:
|
|
logging.info("Images file not found")
|
|
else:
|
|
logging.info("Images file path %s", images_path)
|
|
|
|
if images_path is not None and os.path.exists(images_path):
|
|
logging.info("Images file exists")
|
|
with open(images_path, "r", encoding="utf-8") as images_fd:
|
|
images = json.load(images_fd)
|
|
logging.info("Got images %s", images)
|
|
else:
|
|
images = {}
|
|
|
|
docker_images = []
|
|
for image_name in required_images:
|
|
docker_image = DockerImage(image_name, version)
|
|
if image_name in images:
|
|
docker_image.version = images[image_name]
|
|
docker_images.append(docker_image)
|
|
|
|
latest_error = Exception("predefined to avoid access before created")
|
|
if pull:
|
|
for docker_image in docker_images:
|
|
for i in range(10):
|
|
try:
|
|
logging.info("Pulling image %s", docker_image)
|
|
subprocess.check_output(
|
|
f"docker pull {docker_image}",
|
|
stderr=subprocess.STDOUT,
|
|
shell=True,
|
|
)
|
|
break
|
|
except Exception as ex:
|
|
latest_error = ex
|
|
time.sleep(i * 3)
|
|
logging.info("Got execption pulling docker %s", ex)
|
|
else:
|
|
raise Exception(
|
|
"Cannot pull dockerhub for image docker pull "
|
|
f"{docker_image} because of {latest_error}"
|
|
)
|
|
|
|
return docker_images
|
|
|
|
|
|
def get_image_with_version(
|
|
reports_path: str, image: str, pull: bool = True, version: Optional[str] = None
|
|
) -> DockerImage:
|
|
logging.info("Looking for images file in %s", reports_path)
|
|
return get_images_with_versions(reports_path, [image], pull, version=version)[0]
|