python-scripts/Space_Report.py at master · PureStorage-OpenConnect/python-scripts · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
frompurestorageimportpurestorage
importsys
importcsv
importrequests
importtime
importos
importargparse

# Disable certificate warnings

requests.packages.urllib3.disable_warnings()

starttime=time.time()

# Set array variables

flasharray=''
api_token= '’

# Set script variables

time_stamp= (time.strftime('%m-%d-%y'))
report_file='space-report-'+time_stamp+'.csv'
interval_options='1h, 3h, 24h, 7d, 30d, 90d, 1y'

deffa_connect(flasharray, api_token):

globalarray

try:
array=purestorage.FlashArray(flasharray, api_token=api_token)
print('Successfully connected to ', flasharray)
exceptExceptionase:
print(e)
sys.exit('Exiting: Unable to establish session')

defwrite_output(interval):
withopen(report_file, 'w') ascsvfile:
fieldnames= ['Volume Name', 'Current Data Reduction', 'Data Reduction '+interval, 'Current Size(GB)',
'Size '+interval+' Ago(GB)', interval+' Growth(GB)']
writer=csv.DictWriter(csvfile, fieldnames=fieldnames)
writer.writeheader()

print('Parsing volume data.')

# Loop through all volumes to get historical space data

forcurrentvolinallvolumes:
thisvol=array.get_volume(currentvol['name'], space='True', historical=interval)
volname=thisvol[0]['name']
volcurdr=round(thisvol[0]['data_reduction'], 2)
volstartdr=round(thisvol[len(thisvol) -1]['data_reduction'], 2)
volstartsize=round(thisvol[0]['volumes'] /1024/1024/1024, 2)
volcursize=round(thisvol[len(thisvol) -1]['volumes'] /1024/1024/1024, 2)
volsizedif=volcursize-volstartsize
volsizedif=round(volsizedif, 2)
writer.writerow(
 {'Volume Name': volname, 'Current Data Reduction': volcurdr, 'Data Reduction '+interval: volstartdr,
'Current Size(GB)': volcursize, 'Size '+interval+' Ago(GB)': volstartsize, interval+' Growth(GB)': volsizedif})

defarg_parser():

globalinterval

parser=argparse.ArgumentParser('Add interval')
parser.add_argument('interval', type=str, help='('+interval_options+')')

args=parser.parse_args()
interval=args.interval

ifintervalnotininterval_options:
print('Please enter a valid interval: '+interval_options)
sys.exit()

returninterval

defmain():

globalallvolumes, interval

# Check to see if FlashArray and API key have been set

ifflasharray=="":
sys.exit('Please edit this file and set the IP or DNS name for the variable: flasharray')

ifapi_token=="":
sys.exit('Please edit this file and assign an API token for the variable: api_token for the FlashArray '+flasharray)

# Call argument parser to check for command line options

arg_parser()

# Call function to connect to the FlashArray

fa_connect(flasharray, api_token)

# Gather all volumes on the array

allvolumes=array.list_volumes()
print('Gathering all volumes.')

# Call function to parse and write output to CSV file

write_output(interval)

print('Script completed in ', round(time.time()-starttime,2), ' seconds.')
print('Output file: ', report_file, ' located at: ', os.getcwd())

main()