diff options
Diffstat (limited to 'build/download/docker_downloader.py')
| -rwxr-xr-x | build/download/docker_downloader.py | 30 |
1 files changed, 27 insertions, 3 deletions
diff --git a/build/download/docker_downloader.py b/build/download/docker_downloader.py index c5f53cab..af9d513d 100755 --- a/build/download/docker_downloader.py +++ b/build/download/docker_downloader.py @@ -36,8 +36,12 @@ log = logging.getLogger(__name__) class DockerDownloader(ConcurrentDownloader): - def __init__(self, save, *list_args, workers=3): + def __init__(self, save, *list_args, mirror=None, workers=3): + """ + :param mirror: private repository mirror address (ip:port) + """ self._save = save + self._mirror = mirror try: # big timeout in case of massive images like pnda-mirror-container:5.0.0 (11.4GB) self._docker_client = docker.from_env(timeout=300) @@ -153,7 +157,24 @@ class DockerDownloader(ConcurrentDownloader): if ':' not in image_name.rsplit('/')[-1]: image_name = '{}:latest'.format(image_name) try: - image = self._docker_client.images.pull(image_name) + if self._mirror: + # if docker mirroring repository is set + image_name_split = image_name.split('/') + if (len(image_name_split) > 1) \ + and (image_name_split[0].find(".")) \ + and not (image_name.startswith('docker.io/')): + # if image originates from private registry and its name does not start with 'docker.io' + # download image from docker mirror and retag it to its original name + mirrored_image_name = self._mirror + "/" + '/'.join(image_name_split[1:]) + img = self._docker_client.images.pull(mirrored_image_name) + self._docker_client.images.model.tag(img, image_name) + # untag the image pulled from mirror + self._docker_client.images.remove(mirrored_image_name) + image = self._docker_client.images.get(image_name) + else: + image = self._docker_client.images.pull(image_name) + else: + image = self._docker_client.images.pull(image_name) log.info('Image {} pulled'.format(image_name)) return image except docker.errors.APIError as err: @@ -205,6 +226,9 @@ def run_cli(): help='Save images (without it only pull is executed)') parser.add_argument('--output-dir', '-o', default=os.getcwd(), help='Download destination') + parser.add_argument('--private-registry-mirror', default=None, metavar='IP:PORT', + help='Address of docker mirroring repository that caches images' + ' from private registries to get those images from') parser.add_argument('--check', '-c', action='store_true', default=False, help='Check what is missing. No download.' 'Use with combination with -s to check saved images as well.') @@ -220,7 +244,7 @@ def run_cli(): else: logging.basicConfig(stream=sys.stdout, level=logging.INFO, format='%(message)s') - downloader = DockerDownloader(args.save, [args.image_list, args.output_dir], workers=args.workers) + downloader = DockerDownloader(args.save, [args.image_list, args.output_dir], mirror=args.private_registry_mirror, workers=args.workers) if args.check: log.info('Check mode. No download will be executed.') |