Merge branch 'master' of github.com:shahaan/ansible_cluster_in_a_box

createNode

+#!/usr/bin/env python
+import sys, os, string, subprocess, socket, ansible.runner, re
+import copy, shlex,uuid, random, multiprocessing, time, shutil
+import novaclient.v1_1.client as nvclient
+import novaclient.exceptions as nvexceptions
+import glanceclient.v2.client as glclient
+import keystoneclient.v2_0.client as ksclient
+
+class Authenticate:
+	
+	def __init__(self, username, passwd):
+		self.username=username
+		self.passwd=passwd
+		self.tenantName= os.environ['OS_TENANT_NAME']
+		self.authUrl="https://keystone.rc.nectar.org.au:5000/v2.0"
+		kc = ksclient.Client(   auth_url=self.authUrl,
+					username=self.username,
+					password=self.passwd)
+		self.tenantList=kc.tenants.list()
+		self.novaSemaphore = multiprocessing.BoundedSemaphore(value=1)
+	
+	def createNovaObject(self,tenantName):
+		for tenant in self.tenantList:
+			if tenant.name == tenantName:
+				try:
+					nc = nvclient.Client(	auth_url=self.authUrl,
+						username=self.username,
+						api_key=self.passwd,
+						project_id=tenant.name,
+						tenant_id=tenant.id,
+						service_type="compute"
+						)
+					return nc
+				except nvexceptions.ClientException:
+					raise
+	
+	def gatherInfo(self):
+
+		for tenant in self.tenantList: print tenant.name
+		tenantName = raw_input("Please select a project: (Default MCC-On-R@CMON):")
+		if not tenantName or tenantName not in [tenant.name for tenant in self.tenantList]: 
+			tenantName = "MCC_On_R@CMON"
+		print tenantName,"selected\n"
+		
+		## Fetch the Nova Object
+
+		nc = self.createNovaObject(tenantName)
+		
+		## Get the Flavor
+		flavorList = nc.flavors.list()
+		for flavor in flavorList: print flavor.name
+		flavorName = raw_input("Please select a Flavor Name: (Default m1.xxlarge):")
+		if not flavorName or flavorName not in [flavor.name for flavor in flavorList]:
+			flavorName = "m1.xxlarge"
+		print flavorName,"selected\n"
+
+		
+		## Get the Availability Zones
+		az_p1 = subprocess.Popen(shlex.split\
+		("nova availability-zone-list"),stdout=subprocess.PIPE)
+		az_p2 = subprocess.Popen(shlex.split\
+		("""awk '{if ($2 && $2 != "Name")print $2}'"""),\
+		stdin=az_p1.stdout,stdout=subprocess.PIPE)
+		availabilityZonesList =  subprocess.Popen(shlex.split\
+		("sort"),stdin=az_p2.stdout,stdout=subprocess.PIPE).communicate()[0]
+		print  availabilityZonesList
+		availabilityZone = raw_input("Please select an availability zone: (Default monash-01):")
+		if not availabilityZone or \
+		availabilityZone not in [ zone for zone in availabilityZonesList.split()]:
+			availabilityZone = "monash-01"
+		print availabilityZone,"selected\n"
+		
+		## Get the number of instances to spawn
+		numberOfInstances = raw_input\
+		("Please specify the number of instances to launch: (Default 1):")
+		if not numberOfInstances or \
+		not isinstance(int(numberOfInstances), int):
+			numberOfInstances = 1
+		subprocess.call(['clear'])
+		flavorObj = nc.flavors.find(name=flavorName)
+		print "Creating",numberOfInstances,\
+		"instance(s) in",availabilityZone,"zone..."
+		instanceList = []
+		for counter in range(0,int(numberOfInstances)):
+			nodeName = "MCC-Node"+str(random.randrange(1,1000))
+			try:
+				novaInstance =  nc.servers.create\
+				(name=nodeName,image="ddc13ccd-483c-4f5d-a5fb-4b968aaf385b",\
+				flavor=flavorObj,key_name="shahaan",\
+				availability_zone=availabilityZone)
+				instanceList.append(novaInstance)
+			except nvexceptions.ClientException:
+				raise
+				continue
+				
+		while 'BUILD' in [novaInstance.status \
+		for novaInstance in instanceList]:
+			for count in range(0,len(instanceList)):
+				time.sleep(5)
+				if instanceList[count].status != 'BUILD': 
+					continue
+				else:
+					try:
+						instanceList[count] = nc.servers.get(instanceList[count].id)
+					except nvexceptions.ClientException or \
+					nvexceptions.ConnectionRefused or \
+					nvexceptions.InstanceInErrorState:
+						raise
+						del instanceList[count]
+						continue
+		activeHostsList = []
+		SSHports = []
+		for novaInstance in instanceList:
+			if novaInstance.status == 'ACTIVE':
+				hostname = socket.gethostbyaddr(novaInstance.networks.values()[0][0])[0]
+				activeHostsList.append(hostname)
+				SSHDict = {}
+				SSHDict['IP'] = novaInstance.networks.values()[0][0]
+				SSHDict['status'] = 'CLOSED'
+				SSHports.append(SSHDict) 
+		print "Scanning if port 22 is open..."
+		sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+		while 'CLOSED' in [host['status'] for host in SSHports]:
+			for instance in range(0,len(SSHports)):
+				sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+				if SSHports[instance]['status'] == 'CLOSED' and not sock.connect_ex((SSHports[instance]['IP'], 22)):
+					SSHports[instance]['status'] = 'OPEN'
+					print "Port 22, opened for IP:",SSHports[instance]['IP']
+				else:
+					time.sleep(5)
+				sock.close()
+				
+		fr = open('/etc/ansible/hosts.rpmsave','r+')
+		fw = open('hosts.temp','w+')
+		lines = fr.readlines()
+		for line in lines:
+			fw.write(line)
+			if re.search('\[new-servers\]',line):
+				for host in activeHostsList: fw.write(host+'\n')
+		fr.close()
+		fw.close()
+		shutil.move('hosts.temp','/etc/ansible/hosts')
+		print "Building the Nodes now..."
+		subprocess.call(shlex.split("/mnt/nectar-nfs/root/swStack/ansible/bin/ansible-playbook /mnt/nectar-nfs/root/ansible-config-root/mcc-nectar-dev/buildNew.yml -v"))	
+
+if __name__ == "__main__":
+	username = os.environ['OS_USERNAME']
+	passwd = os.environ['OS_PASSWORD']
+	choice = raw_input(username + " ? (y/n):")
+	while choice and choice not in ("n","y"):
+		print "y or n please"
+		choice = raw_input()
+	if choice == "n":
+		username = raw_input("username :")
+		passwd = raw_input("password :")
+	auth = Authenticate(username, passwd)
+	auth.gatherInfo()