#!/usr/bin/env python

from mixcoatl.infrastructure.machine_image import MachineImage
import argparse
import sys

if __name__ == '__main__':
    parser = argparse.ArgumentParser(description="Update a machine image's owning group")
    parser.add_argument('--machine_image_id', '-m', type=int, help='Machine Image ID', required=True)
    parser.add_argument('--group_id', '-g', type=int, help='Group ID', required=True)
    cmd_args = parser.parse_args()

    m = MachineImage(cmd_args.machine_image_id)
    results = m.set_owner(cmd_args.group_id)

    if results is True:
        print "Machine Image with ID: %s has changed to Group ID: %s" % (str(cmd_args.machine_image_id), str(cmd_args.group_id))
    else:
        print results