def create_archive():
if not request.json or 'images' not in request.json:
abort(400)
if not isinstance(request.json['images'], list):
abort(400)
images = [DockerImage(i) for i in request.json['images']]
with Connection(conn):
q = Queue('default')
job = q.enqueue(save_images,
Path(app.config['OUTPUT_DIR']),
images,
job_timeout='1h',
ttl='24h',
result_ttl='7d')
response = {
'job_id': job.get_id(),
'images': [i.name for i in images],
}
headers = {'Location': '/api/v1/job/%s' % job.get_id()}
return jsonify(response), 202, headers
def test_image_registry_no_port(self):
name = 'registry.example.com/username/image:latest'
image = DockerImage(name)
assert image.registry == 'registry.example.com'
def test_image_registry(self):
name = 'registry.example.com:5000/username/image:latest'
image = DockerImage(name)
assert image.registry == 'registry.example.com:5000'
def test_image_registry_notag(self):
name = 'image'
image = DockerImage(name)
assert image.registry == None
def test_image_registry_default(self):
name = 'image:latest'
image = DockerImage(name)
assert image.registry == None
def test_image_registry_group(self):
name = 'username/image:latest'
image = DockerImage(name)
assert image.registry == None
def main():
# global arguments
parser = argparse.ArgumentParser()
parser.add_argument(
'--log-level',
choices=['critical', 'error', 'warning', 'info', 'debug'],
default='info',
help='Set log level for console output')
subparsers = parser.add_subparsers(title='subcommands', dest='subcommand')
# save subcommand
save_parser = subparsers.add_parser('save',
help='save docker image to file')
save_parser.add_argument('--output-dir',
metavar='DIRECTORY',
default='/var/spool/dockermirror',
help='directory to put archive file in')
save_parser.add_argument('--remove',
action='store_true',
default=False,
help='remove docker images after save')
save_parser.add_argument('images',
metavar='IMAGE',
nargs='+',
help='docker images to save')
# load subcommands
load_parser = subparsers.add_parser('load',
help='load docker image from file')
load_parser.add_argument('--registry',
metavar='REGISTRY',
default=None,
help='push archive images to registry')
load_parser.add_argument('--remove',
action='store_true',
default=False,
help='remove archive after load')
load_parser.add_argument('archive',
metavar='NAME',
help='archive filename')
# remove subcommands
remove_parser = subparsers.add_parser('remove',
help='remove saved docker images')
remove_parser.add_argument('archive',
metavar='ARCHIVE',
help='archive filename')
# monitor subcommands
monitor_parser = subparsers.add_parser(
'monitor', help='monitor folder for new archives and auto-load')
monitor_parser.add_argument('--registry',
metavar='REGISTRY',
default=None,
help='push archive images to registry')
monitor_parser.add_argument('--interval',
metavar='SECONDS',
default=60,
type=int,
help='monitor directory scan interval')
monitor_parser.add_argument('--directory',
metavar='DIRECTORY',
default='/var/spool/dockermirror',
help='directory to monitor')
# show subcommands
show_parser = subparsers.add_parser('show',
help='show archive information')
show_parser.add_argument('archive',
metavar='ARCHIVE',
help='archive filename')
args = parser.parse_args()
# Set up logging
LOGGER.setLevel(args.log_level.upper())
console = logging.StreamHandler()
LOGGER.addHandler(console)
if args.subcommand == 'save':
images = [DockerImage(i) for i in args.images]
dm = DockerMirror()
dm.save(Path(args.output_dir), images, args.remove)
elif args.subcommand == 'load':
archive = DockerArchive(Path(args.archive))
archive.load(args.registry, args.remove)
elif args.subcommand == 'remove':
archive = DockerArchive(Path(args.archive))
for image in archive.images:
image.remove()
archive.remove()
elif args.subcommand == 'monitor':
while True:
start_time = time.time()
LOGGER.debug('Scanning %s for new docker archives', args.directory)
for archive in get_archives(args.directory):
LOGGER.info('Loading %s', archive.filepath)
archive.load(args.registry, remove=True)
run_time = time.time() - start_time
if run_time < args.interval:
sleep_time = args.interval - run_time
LOGGER.debug('Sleeping %s seconds', sleep_time)
time.sleep(args.interval - run_time)
elif args.subcommand == 'show':
archive = DockerArchive(Path(args.archive))
print(json.dumps(archive.manifest, sort_keys=True, indent=4))