Skip to content

HTTPS clone URL

Subversion checkout URL

You can clone with HTTPS or Subversion.

Download ZIP
Fetching contributors…

Cannot retrieve contributors at this time

executable file 228 lines (217 sloc) 11.051 kb
#!/usr/bin/python
from optparse import OptionParser
from lib import swarmtoolscore
import sys
import httplib
import json
import ast
def usage(script_name):
print "%s [create|update|destroy|list|get_info|add_resource|remove_resource|list_resources] \n"%(script_name)
print "Use '%s [method] --help for a method's usage and options."%(script_name)
sys.exit()
def create(hostname, api_key, name, description, public, resources):
conn = httplib.HTTPConnection(hostname)
create_swarm = {"name": name, "description": description}
if public != None:
if public == "true":
create_swarm["public"] = True
elif public == "false":
create_swarm["public"] = False
if resources != None:
resources_dict = ast.literal_eval(resources)
resources_list = []
for key, value in resources_dict.iteritems():
if value == "both":
producer_dict = {"resource_id": key, "resource_type": "producer"}
consumer_dict = {"resource_id": key, "resource_type": "consumer"}
resources_list.append(producer_dict)
resources_list.append(consumer_dict)
else:
add_resource_dict = {"resource_id": key, "resource_type": value}
resources_list.append(add_resource_dict)
create_swarm["resources"] = resources_list
create_swarm_json = json.dumps(create_swarm)
conn.request("POST", "/swarms", create_swarm_json, {"x-bugswarmapikey":api_key})
resp = conn.getresponse()
txt = resp.read()
conn.close()
print json.dumps(json.loads(txt), sort_keys=True, indent=4)
def update(hostname, api_key, swarm_id, name, description, public):
conn = httplib.HTTPConnection(hostname)
update_swarm = {}
if name != None:
update_swarm["name"] = name
if description != None:
update_swarm["description"] = description
if public != None:
if public == "true":
update_swarm["public"] = True
elif public == "false":
update_swarm["public"] = False
update_swarm_json = json.dumps(update_swarm)
conn.request("PUT", "/swarms/%s"%(swarm_id), update_swarm_json, {"x-bugswarmapikey":api_key})
resp = conn.getresponse()
txt = resp.read()
conn.close()
print json.dumps(json.loads(txt), sort_keys=True, indent=4)
def destroy(hostname, api_key, swarm_id):
conn = httplib.HTTPConnection(hostname)
conn.request("DELETE", "/swarms/%s"%(swarm_id), None, {"x-bugswarmapikey":api_key})
resp = conn.getresponse()
txt = resp.status
conn.close()
if str(txt) == "204":
print "Great success! :)"
else:
print txt
print "Something went wrong! :("
def list_swarms(hostname, api_key):
conn = httplib.HTTPConnection(hostname)
conn.request("GET", "/swarms", None, {"x-bugswarmapikey":api_key})
resp = conn.getresponse()
txt = resp.read()
conn.close()
print json.dumps(json.loads(txt), sort_keys=True, indent=4)
def get_info(hostname, api_key, swarm_id):
conn = httplib.HTTPConnection(hostname)
conn.request("GET", "/swarms/%s"%(swarm_id), None, {"x-bugswarmapikey":api_key})
resp = conn.getresponse()
txt = resp.read()
conn.close()
print json.dumps(json.loads(txt), sort_keys=True, indent=4)
def add_resource(hostname, api_key, swarm_id, resource_id, resource_type):
add_resource = {"resource_id": resource_id, "resource_type": resource_type}
add_resource_json = json.dumps(add_resource)
conn = httplib.HTTPConnection(hostname)
conn.request("POST", "/swarms/%s/resources"%(swarm_id), add_resource_json, {"x-bugswarmapikey":api_key})
resp = conn.getresponse()
txt = resp.read()
conn.close()
if txt != "Created":
print json.dumps(json.loads(txt), sort_keys=True, indent=4)
else:
print "Great success! :)"
def remove_resource(hostname, api_key, swarm_id, resource_id, resource_type):
remove_resource = {"resource_id": resource_id, "resource_type": resource_type}
remove_resource_json = json.dumps(remove_resource)
conn = httplib.HTTPConnection(hostname)
conn.request("DELETE", "/swarms/%s/resources"%(swarm_id), remove_resource_json, {"x-bugswarmapikey":api_key})
resp = conn.getresponse()
txt = resp.read()
conn.close()
if txt != "":
print json.dumps(json.loads(txt), sort_keys=True, indent=4)
else:
print "Great success! :)"
def list_resources(hostname, api_key, swarm_id, resource_type):
conn = httplib.HTTPConnection(hostname)
if resource_type != None:
conn.request("GET", "/swarms/%s/resources?type=%s"%(swarm_id, resource_type), None, {"x-bugswarmapikey":api_key})
else:
conn.request("GET", "/swarms/%s/resources"%(swarm_id), None, {"x-bugswarmapikey":api_key})
resp = conn.getresponse()
txt = resp.read()
conn.close()
print json.dumps(json.loads(txt), sort_keys=True, indent=4)
def main():
server_info = swarmtoolscore.get_server_info()
keys = swarmtoolscore.get_keys()
if len(sys.argv) == 1:
usage(sys.argv[0])
elif sys.argv[1] == "create":
opt_usage = "usage: \n %s NAME DESCRIPTION [options]"%(sys.argv[1])
opt_usage += "\n\n *NAME: The name of the swarm being created." \
+"\n *DESCRIPTION: The description of the swarm being created."
parser = OptionParser(usage = opt_usage)
parser.add_option("-p", "--public", dest="public", help="Set whether the swarm is public or not. Valid types; 'true', 'false'.", metavar="PUBLIC")
parser.add_option("-r", "--resources", dest="resources", help="Input resources you wish to add to the swarm by default. Format the resources as a dictionary of \"resource_id\":\"resource_type\" key:value pairs. If you wish for a resource to be added as both a consumer and a producer, set the \"resource_type\" to \"both\". Example: '{\"c35b44419f7b2717f25c6022d04af72a492cd892\":\"consumer\",\"b09df0ea0323073b05a1ddd9b2d44c769d8fa7c2\":\"both\"}'", metavar="RESOURCES")
(options, args) = parser.parse_args()
if len(args) != 3:
print "Invalid number of args. See --help for correct usage."
sys.exit()
name = args[1]
description = args[2]
create(server_info["hostname"], keys["configuration"], name, description, options.public, options.resources)
elif sys.argv[1] == "update":
opt_usage = "usage: \n %s SWARM_ID [options]"%(sys.argv[1])
opt_usage += "\n\n *SWARM_ID: The ID of the swarm being updated."
parser = OptionParser(usage = opt_usage)
parser.add_option("-n", "--name", dest="name", help="Set the swarm's name", metavar="NAME")
parser.add_option("-d", "--description", dest="description", help="Set the swarm's description", metavar="DESCRIPTION")
parser.add_option("-p", "--public", dest="public", help="Set whether the swarm is public or not. Valid types; 'true', 'false'.", metavar="PUBLIC")
(options, args) = parser.parse_args()
if len(args) != 2:
print "Invalid number of args. See --help for correct usage."
sys.exit()
swarm_id = args[1]
update(server_info["hostname"], keys["configuration"], swarm_id, options.name, options.description, options.public)
elif sys.argv[1] == "destroy":
opt_usage = "usage: \n %s SWARM_ID"%(sys.argv[1])
opt_usage += "\n\n *SWARM_ID: The ID of the swarm to destroy."
parser = OptionParser(usage = opt_usage)
(options, args) = parser.parse_args()
if len(args) != 2:
print "Invalid number of args. See --help for correct usage."
sys.exit()
swarm_id = args[1]
destroy(server_info["hostname"], keys["configuration"], swarm_id)
elif sys.argv[1] == "list":
opt_usage = "usage: \n %s"%(sys.argv[1])
parser = OptionParser(usage = opt_usage)
(options, args) = parser.parse_args()
if len(args) != 1:
print "Invalid number of args. See --help for correct usage."
sys.exit()
list_swarms(server_info["hostname"], keys["configuration"])
elif sys.argv[1] == "get_info":
opt_usage = "usage: \n %s SWARM_ID"%(sys.argv[1])
opt_usage += "\n\n *SWARM_ID: The ID of the swarm who's info is desired."
parser = OptionParser(usage = opt_usage)
(options, args) = parser.parse_args()
if len(args) != 2:
print "Invalid number of args. See --help for correct usage."
sys.exit()
swarm_id = args[1]
get_info(server_info["hostname"], keys["configuration"], swarm_id)
elif sys.argv[1] == "add_resource":
opt_usage = "usage: \n %s SWARM_ID RESOURCE_ID RESOURCE_TYPE"%(sys.argv[1])
opt_usage += "\n\n *SWARM_ID: The ID of the swarm to add to." \
+"\n *RESOURCE_ID: The ID of the resource to add." \
+"\n *RESOURCE_TYPE: The type of the resource to add. Valid types; 'producer', 'consumer'."
parser = OptionParser(usage = opt_usage)
(options, args) = parser.parse_args()
if len(args) != 4:
print "Invalid number of args. See --help for correct usage."
sys.exit()
swarm_id = args[1]
resource_id = args[2]
resource_type = args[3]
add_resource(server_info["hostname"], keys["configuration"], swarm_id, resource_id, resource_type)
elif sys.argv[1] == "remove_resource":
opt_usage = "usage: \n %s SWARM_ID RESOURCE_ID RESOURCE_TYPE"%(sys.argv[1])
opt_usage += "\n\n *SWARM_ID: The ID of the rwarm remove from." \
+"\n *RESOURCE_ID: The ID of the resource to remove." \
+"\n *RESOURCE_TYPE: The type of the resource to remove. Valid types; 'producer', 'consumer'."
parser = OptionParser(usage = opt_usage)
(options, args) = parser.parse_args()
if len(args) != 4:
print "Invalid number of args. See --help for correct usage."
sys.exit()
swarm_id = args[1]
resource_id = args[2]
resource_type = args[3]
remove_resource(server_info["hostname"], keys["configuration"], swarm_id, resource_id, resource_type)
elif sys.argv[1] == "list_resources":
opt_usage = "usage: \n %s SWARM_ID [options]"%(sys.argv[1])
opt_usage += "\n\n *SWARM_ID: The ID of the swarm who's resources will be listed."
parser = OptionParser(usage = opt_usage)
parser.add_option("-t", "--type", dest="type", help="Limit the list. Valid types; 'producer', 'consumer'.", metavar="TYPE")
(options, args) = parser.parse_args()
if len(args) != 2:
print "Invalid number of args. See --help for correct usage."
sys.exit()
swarm_id = args[1]
list_resources(server_info["hostname"], keys["configuration"], swarm_id, options.type)
else:
usage(sys.argv[0])
main()
Jump to Line
Something went wrong with that request. Please try again.