#!/usr/bin/env python
from mixcoatl.infrastructure.server import Server
import argparse
import sys

if __name__ == '__main__':
    parser = argparse.ArgumentParser(description="Update a server's billing code/budget")
    parser.add_argument('--server_id', '-s', type=int, help='Server ID', required=True)
    parser.add_argument('--billing_code', '-b', type=int, help='Billing code, also known as budget ID', required=True)
    cmd_args = parser.parse_args()

    s = Server(cmd_args.server_id)
    results = s.set_billing_code(cmd_args.billing_code)

    if results is True:
        print "Server with ID: %s has changed to Billing Code/Budged ID: %s" % (str(cmd_args.server_id), str(cmd_args.billing_code))
    else:
        print results
