rest_interface.py

#!/bin/python

import sys, getopt, purestorage, json

def usage():
 print '{} [ -a | --array ] <array_name> [ -p | --password ] <api token> [ -f | --function ] <function name> [-v | --verbose]'.format(sys.argv[0])
 print 'where <function name> can be:'
 print ' list_luns [-l | --lun_name] <comma separated list of lun names (no spaces!)>'
 print ' list_hosts [-h | --host_name] <comma separated list of hostnames (no spaces!)>'
 print ' list_hostgroups'
 print ''
 print ' create_lun [-l | --lun_name] <lun name> [-s | --lun_size] <lun size>'
 print ' where <lun size> can be in G or T (eg. 1T, 20G)'
 print ' create_host [ -h | --host_name ] <hostname> [ -w | --wwn ] <comma separated list of WWNs (no spaces!)>'
 print ' create_hostgroup [ -g | --group_name ] <hostgroup_name>'
 print ''
 print ' add_host_to_hostgroup [ -h | --host_name ] <comma separated list of hostnames (no spaces!)> [ -g | --group_name ] <hostgroup_name>'
 print ' add_lun_to_hostgroup [ -l | --lun_name ] <lun name> [ -g | --group_name ] <hostgroup_name>'
 print ' add_lun_to_host [ -l | --lun_name ] <lun_name> [ -h | --host_name ] <hostname>'
 print ''
 print ' remove_host_from_hostgroup [ -h | --host_name ] <comma separated list of hostnames (no spaces!)> [ -g | --group_name ] <hostgroup_name>'
 print ' remove_lun_from_host [ -l | --lun_name ] <lun name> [ -h | --host_name ] <hostname>'
 print ' remove_lun_from_hostgroup [ -l | --lun_name ] <lun name> [ -g | --group_name ] <hostgroup_name>'
 print ''
 print ' delete_host [-h --host_name ] <hostname>'
 print ' delete_hostgroup [-g | --group_name ] <hostgroup_name>'
 print ' delete_lun [ -l | --lun_name ] <lun_name>'
 print ''
 print '-v | --verbose: will give you more detail relating to that function.'
 print ' without "verbose" any successful actions (creates, adds etc) will simply exit with a 0 return code'
 sys.exit(2)

def main(argv):
 array_name = ''
 api_token = ''
 verbose = False
 try:
 opts, args = getopt.getopt(argv,"?h:l:a:p:f:s:vw:g:",["help", "host_name=", "lun_name=", "array=","password=","function=","lun_size=","verbose","wwn=","group_name="])
 except:
 print str(getopt.GetoptError)
 usage()
 for opt, arg in opts:
 if opt in ("-?", "--help"):
 usage()
 elif opt in ("-a", "--array"):
 array_name = arg
 elif opt in ("-p", "--password"):
 api_token = arg
 elif opt in ("-f", "--function"):
 function = arg
 elif opt in ("-h", "--host_name"):
 host_name = arg
 elif opt in ("-w", "--wwn"):
 wwn_list = arg
 elif opt in ("-l", "--lun_name"):
 lun_name = arg
 elif opt in ("-g", "--group_name"):
 group_name = arg
 elif opt in ("-s", "--lun_size"):
 lun_size = arg
 elif opt in ("-v", "--verbose"):
 verbose = True

# If no function is specified then send a usage message
 try:
 function
 except:
 print "ERROR: You must specify a function to perform"
 usage()
 
 # We need the following to prevent a security warning
 import requests 
 from requests.packages.urllib3.exceptions import InsecureRequestWarning
 requests.packages.urllib3.disable_warnings(InsecureRequestWarning)

 array = purestorage.FlashArray(array_name, api_token=api_token)
# 
 if function == 'list_luns':
 try:
 lun_name
 except:
 response=array.list_volumes()
 if verbose:
 print("%-28s%-24s%-26s%s" % ("Name", "Size in GB", "Created", "Serial"))
 for x in range(len(response)):
 print("%-28s%-24s%-26s%s" % (response[x]['name'], response[x]['size']/1024/1024/1024, response[x]['created'], response[x]['serial']))
 else:
 for x in range(len(response)):
 print response[x]['name']
 else:
 response=array.list_volumes(names=lun_name, connect=True)
 if verbose:
 print("%-28s%-24s%-26s%s" % ("Name", "Size in GB", "Host Name", "Host Group"))
 for x in range(len(response)):
 print("%-28s%-24s%-26s%s" % (response[x]['name'], response[x]['size']/1024/1024/1024, response[x]['host'], response[x]['hgroup']))
 else:
 print "If you're wanting information on specific lun(s) then use the '-v' option"
#
 elif function == 'list_hosts':
# if host_name has been passed then let's just get info about this else do all hosts
 try:
 host_name
 except:
 response=array.list_hosts(all=True)
 else:
 response=array.list_hosts(names=host_name, all=True)
 if verbose:
 print("%-16s%-26s%-26s%-28s" % ("Name", "WWN", "Host Group", "Presented LUN Name"))
 for x in range(len(response)):
 print("%-16s%-26s%-26s%-28s" % (response[x]['name'], response[x]['host_wwn'], response[x]['hgroup'], response[x]['vol']))
# response[x]['name']=""
# response[x]['hgroup']=""
 else:
 response=array.list_hosts()
 for x in range(len(response)):
 print("%-16s%-26s" % (response[x]['name'], response[x]['hgroup']))
### print response[x]['name']

#
 elif function == 'list_hostgroups':
 response=array.list_hgroups()
 if verbose:
 print("%-28s%-26s" % ("Name", "Hosts"))
 for x in range(len(response)):
 for y in range(len(response[x]['hosts'])):
 print("%-28s%s" % (response[x]['name'], response[x]['hosts'][y]))
 response[x]['name']=""
 else:
 for x in range(len(response)):
 print response[x]['name']
# 
 elif function == 'create_lun':
 try:
 response=array.create_volume(lun_name, lun_size)
 if verbose:
 print("%-28s%-24s%-26s%s" % ("Name", "Size in GB", "Created", "Serial"))
 print("%-28s%-24s%-26s%s" % (response['name'], response['size']/1024/1024/1024, response['created'], response['serial']))
 sys.exit(0)
 except purestorage.PureHTTPError as response:
 y=json.loads(response.text)
 print "LUN Creation failed: {}: {}".format(lun_name, y[0]['msg'])
 sys.exit(1)
#
 elif function == 'create_host':
 try:
 response=array.create_host(host_name)
 if verbose:
 print "host creation succeeded: {}".format(host_name)
 except purestorage.PureHTTPError as response:
 y=json.loads(response.text)
 print "host creation failed: {}: {}".format(host_name, y[0]['msg'])
 sys.exit(1)
 try:
 wwn_list=wwn_list.split(',')
 response=array.set_host(host_name, wwnlist=wwn_list)
 if verbose:
 print "WWNs added"
 for y in range(len(response['wwn'])):
 print "{}".format(response['wwn'][y])
 sys.exit(0)
 except purestorage.PureHTTPError as response:
 print response
 y=json.loads(response.text)
 print "Failed to allocate WWNs to {}: {}".format(host_name, y[0]['msg'])
 sys.exit(1)
#
 elif function == 'create_hostgroup':
 try:
 response=array.create_hgroup(group_name)
 if verbose:
 print "hostgroup creation succeeded: {}".format(group_name)
 sys.exit(0)
 except purestorage.PureHTTPError as response:
 y=json.loads(response.text)
 print "hostgroup creation failed: {}: {}".format(group_name, y[0]['msg'])
 sys.exit(1)
#
 elif function == 'add_host_to_hostgroup':
 try:
 host_list=host_name.split(',')
 response=array.set_hgroup(group_name, addhostlist=host_list)
 if verbose:
 print "Succesfully added {} to hostgroup {}".format(host_list, group_name)
 sys.exit(0)
 except purestorage.PureHTTPError as response:
 y=json.loads(response.text)
 print "Adding {} to hostgroup {} failed: {}".format(host_list, group_name, y[0]['msg'])
 sys.exit(1)
#
 elif function == 'add_lun_to_hostgroup':
 try:
 response=array.connect_hgroup(group_name, lun_name)
 if verbose:
 print "Succesfully added lun {} to hostgroup {}".format(lun_name, group_name)
 sys.exit(0)
 except purestorage.PureHTTPError as response:
 y=json.loads(response.text)
 print "Adding lun {} to hostgroup {} failed: {}".format(lun_name, group_name, y[0]['msg'])
 sys.exit(1)
#
 elif function == 'add_lun_to_host':
 try:
 response=array.connect_host(host_name, lun_name)
 if verbose:
 print "Succesfully added lun {} to host {}: lun={}".format(lun_name, host_name, response['lun'])
 sys.exit(0)
 except purestorage.PureHTTPError as response:
 y=json.loads(response.text)
 print "Adding lun {} to host {} failed: {}".format(lun_name, host_name, y[0]['msg'])
 sys.exit(1)
#
 elif function == 'delete_host':
 try:
 response=array.delete_host(host_name)
 if verbose:
 print "Succesfully deleted host {}".format(host_name)
 sys.exit(0)
 except purestorage.PureHTTPError as response:
 y=json.loads(response.text)
 print "Deletion of host {} failed: {}".format(host_name, y[0]['msg'])
 sys.exit(1)
#
 elif function == 'delete_hostgroup':
 try:
 response=array.delete_hgroup(group_name)
 if verbose:
 print "Succesfully deleted hostgroup {}".format(group_name)
 sys.exit(0)
 except purestorage.PureHTTPError as response:
 y=json.loads(response.text)
 print "Deletion of hostgroup {} failed: {}".format(group_name, y[0]['msg'])
 sys.exit(1)
#
 elif function == 'delete_lun':
 try:
 response=array.destroy_volume(lun_name)
 if verbose:
 print "Succesfully deleted volume {}".format(lun_name)
 sys.exit(0)
 except purestorage.PureHTTPError as response:
 y=json.loads(response.text)
 print "Deletion of lun {} failed: {}".format(lun_name, y[0]['msg'])
 sys.exit(1)
 try:
 response=array.eradicate_volume(lun_name)
 if verbose:
 print "Succesfully eradicated volume {}".format(lun_name)
 sys.exit(0)
 except purestorage.PureHTTPError as response:
 y=json.loads(response.text)
 print "eradication of lun {} failed: {}".format(lun_name, y[0]['msg'])
 sys.exit(1)
#
 elif function == 'remove_host_from_hostgroup':
 try:
 host_list=host_name.split(',')
 response=array.set_hgroup(group_name, remhostlist=host_list)
 if verbose:
 print "Succesfully removed () from hostgroup {}".format(host_list, group_name)
 sys.exit(0)
 except purestorage.PureHTTPError as response:
 y=json.loads(response.text)
 print "Removal of hosts {} from hostgroup {} failed: {}".format(host_list, group_name, y[0]['msg'])
 sys.exit(1)
#
### print ' remove_lun_from_host [ -l | --lun_name ] <lun name> [ -h | --host_name ] <hostname>'
 elif function == 'delete_hostgroup':
 try:
 response=array.delete_hgroup(group_name)
 if verbose:
 print "Succesfully deleted hostgroup {}".format(group_name)
 sys.exit(0)
 except purestorage.PureHTTPError as response:
 y=json.loads(response.text)
 print "Deletion of hostgroup {} failed: {}".format(group_name, y[0]['msg'])
 sys.exit(1)
#
### print ' remove_lun_from_hostgroup [ -l | --lun_name ] <lun name> [ -g | --group_name ] <hostgroup_name>'
### disconnect_hgroup(hgroup, volume)

 elif function == 'delete_hostgroup':
 try:
 response=array.delete_hgroup(group_name)
 if verbose:
 print "Succesfully deleted hostgroup {}".format(group_name)
 sys.exit(0)
 except purestorage.PureHTTPError as response:
 y=json.loads(response.text)
 print "Deletion of hostgroup {} failed: {}".format(group_name, y[0]['msg'])
 sys.exit(1)
#

 #disconnect from API
 array.invalidate_cookie()

if __name__ == "__main__":
 main(sys.argv[1:])