# Author: William Lam
# Website: www.williamlam.com
# Product: VMware vSphere + vSphere Flash Read Cache (vFRC)
# Description: Python script that calls vSphere MOB to add Virtual Flash Resource
# Reference: http://www.williamlam.com/2013/11/how-to-automate-configuration-of-vfrc.html

import sys,re,os,urllib,urllib2,base64

if len(sys.argv) == 1:
	print "\n\tUsage: " + str(sys.argv[0]) + " [VFFS-UUID]\n"
	sys.exit(1)
else:
	vffsUuid = str(sys.argv[1])
 
# mob url
url = "https://localhost/mob/?moid=ha-vflash-manager&method=configureVFlashResource"
 
# mob login credentials
username = "root"
password = "vmware123"
 
# Create global variables
global passman,authhandler,opener,req,page,page_content,nonce,headers,cookie,params,e_params
 
# Code to build opener with HTTP Basic Authentication
passman = urllib2.HTTPPasswordMgrWithDefaultRealm()
passman.add_password(None,url,username,password)
authhandler = urllib2.HTTPBasicAuthHandler(passman)
opener = urllib2.build_opener(authhandler)
urllib2.install_opener(opener)
 
### Code to capture required page data and cookie required for post back to meet CSRF requirements  ###
try:
	req = urllib2.Request(url)
	page = urllib2.urlopen(req)
	page_content= page.read()
except IOError, e:
	opener.close()
	sys.exit(1)
else:
	print "Successfully connected to vSphere MOB"
 
# regex to get the vmware-session-nonce value from the hidden form entry
reg = re.compile('name="vmware-session-nonce" type="hidden" value="?([^\s^"]+)"')
nonce = reg.search(page_content).group(1)
 
# get the page headers to capture the cookie
headers = page.info()
cookie = headers.get("Set-Cookie")

# Code to create HostVFlashManagerVFlashResourceConfigSpec
xml = '<spec xsi:type="HostVFlashManagerVFlashResourceConfigSpec"><vffsUuid>' + vffsUuid + '</vffsUuid></spec>'

try :
	params = {'vmware-session-nonce':nonce,'spec':xml}
	e_params = urllib.urlencode(params)
	req = urllib2.Request(url, e_params, headers={"Cookie":cookie})
	page = urllib2.urlopen(req).read()
except IOError, e:
	opener.close()
	sys.exit(1)
else:
	print "Successfully issued configureVFlashResource() with VFFS UUID " + vffsUuid + "\n"