#!/usr/bin/env python

from mixcoatl.infrastructure.volume import Volume
import argparse
import sys

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

    s = Volume(cmd_args.volume_id)
    results = s.set_owner(cmd_args.group_id)

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