blob: 551a238ca842d91e6e7d0be9673d6aa4717bb3a6 (
plain)
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
|
#!/usr/bin/python
#
# this script is used to update the /etc/hosts file in the host/dom0 so that the
# guestname will be resolved to the guest's IP address.
#
import sys
import os
import time
import xmlrpclib
from lxml import etree
from cStringIO import StringIO
from optparse import OptionParser
import socket
#start here
usage = "usage: %prog --recipeid "
parser = OptionParser()
parser.add_option("-r", "--recipeid", dest="recipeid", help="recipe id of the guest to hostname of.")
(options, args) = parser.parse_args()
recipeid = options.recipeid
xml_tries = 1
interval = 300
proxy = xmlrpclib.ServerProxy('http://%s:8000/RPC2' % os.environ['LAB_CONTROLLER'])
while xml_tries < 5:
try:
recipe_xml = proxy.get_my_recipe(dict(recipe_id=recipeid))
break
except:
print "Couldn't get guestinfo from %s . sleeping %i secs" % (os.environ['LAB_CONTROLLER'] , interval)
time.sleep(interval)
xml_tries += 1
if xml_tries == 5:
print "Can't get guestinfo from %s" % os.environ['LAB_CONTROLLER']
sys.exit(1)
thetree = etree.parse(StringIO(recipe_xml))
for node in thetree.xpath('/job/recipeSet/recipe/guestrecipe'):
print "in node"
if node.attrib['id'] == recipeid:
guestname = node.attrib['guestname']
guesthost = node.attrib['system']
guestip = socket.gethostbyname(guesthost)
print "guestname: " + guestname + " guesthost: " + guesthost + " ip: " + guestip
# update /etc/hosts with this info
fh = open('/etc/hosts', 'a')
fh.write(guestip+" "+guestname+"\n")
fh.close()
sys.exit(0)
|
