bocker

#!/usr/bin/python3

import re
import click
import sys

from commands.pull_command import run_pull
from commands.run_command import run_run


IMAGE_TAG_PATTERN = r'(?P<image>[^/:]+)(:)?(?P<tag>[^/:]*)'

@click.group()
def cli():
    pass

@click.command()
@click.argument('image_name')
def pull(image_name):
    match = re.match(IMAGE_TAG_PATTERN, image_name)
    if not match:
        print('invalid args')
        sys.exit(1)
    image = match.group('image')
    tag = match.group('tag') if match.group('tag') else 'latest'
    run_pull(image, tag)

@click.command()
@click.argument('image_name', required=True)
@click.option('--cpus', help='Number of CPUs', default=None, type=float)
@click.option('--memory',help='Memory limit in bytes.', default=None)
@click.argument('command', required=True, nargs=-1)
def run(image_name, cpus, memory, command):
    match = re.match(IMAGE_TAG_PATTERN, image_name)
    if not match:
        print('invalid args')
        sys.exit(1)
    image = match.group('image')
    tag = match.group('tag') if match.group('tag') else 'latest'
    run_run(image, tag, cpus, memory, command)

cli.add_command(pull)
cli.add_command(run)

if __name__ == '__main__':
    cli()