focker/focker.py

from argparse import ArgumentParser
import yaml
import os
from weir import zfs, process
from .build import build


def clone(self, name, props={}, force=False):
    url = zfs._urlsplit(self.name)
    url_1 = zfs._urlsplit(name)
    if url.netloc != url_1.netloc:
        raise ValueError('Clone has to happen on the same host')
    cmd = ['zfs', 'clone']
    for prop, value in props.items():
        cmd.append('-o')
        cmd.append(prop + '=' + str(value))
    cmd.append(url.path)
    cmd.append(url_1.path)
    process.check_call(cmd, netloc=url.netloc)

zfs.ZFSSnapshot.clone = clone


def command_build(args):
    fname = os.path.join(args.focker_dir, 'Fockerfile.yml')
    print('fname:', fname)
    if not os.path.exists(fname):
        raise ValueError('No Fockerfile.yml could be found in the specified directory')
    with open(fname, 'r') as f:
        spec = yaml.safe_load(f)
    print('spec:', spec)
    build(spec)


def run(args):
    pass


def create_parser():
    parser = ArgumentParser()
    subparsers = parser.add_subparsers()
    parser_build = subparsers.add_parser('build')
    parser_build.set_defaults(func=command_build)
    parser_build.add_argument('focker_dir', type=str)
    parser_run = subparsers.add_parser('run')
    parser_run.set_defaults(func=run)
    parser_rm = subparsers.add_parser('rm')
    parser_rmi = subparsers.add_parser('rmi')
    parser_ps = subparsers.add_parser('ps')
    parser_images = subparsers.add_parser('images')
    return parser


parser = create_parser()
args = parser.parse_args()
if not hasattr(args, 'func'):
    raise ValueError('You must choose the mode')
args.func(args)