import sys
import os
from pathlib import Path
import subprocess
from collections import namedtuple
from pkg_resources import resource_stream
import logging
import docker
import requests
from ruamel.yaml import YAML
logging.basicConfig(level=logging.INFO)
yaml = YAML()
GALLERY_PATH = 'gallery.yaml' # Relative to module root
Example = namedtuple(
'Example',
[
'image',
'repo_url',
'ref'
]
)
def build_image(example):
subprocess.run([
sys.executable,
'-m',
'repo2docker',
'--ref',
example.ref,
'--user-name',
'jovyan',
'--user-id',
'1000',
'--no-run',
'--image-name',
example.image,
example.repo_url
])
def parse_gallery_config(fp):
config = yaml.load(fp)
examples = []
for example_name, example_yaml in config['examples'].items():
example = Example(
image=f'{example_name}:latest',
repo_url=example_yaml['repo_url'],
ref=example_yaml.get('ref', 'master'),
)
examples.append(example)
return examples
def main():
# prune stopped containers and dangling images
client = docker.from_env()
client.containers.prune()
client.images.prune()
# build the new images
with resource_stream(__name__, GALLERY_PATH) as fp:
examples = parse_gallery_config(fp)
for example in examples:
logging.info(f'Building image {example.image}')
build_image(example)
if __name__ == '__main__':
main()
