merge from improvements on the 1.0 branch:

[monitor.git] / getsshkeys.py

diff --git a/getsshkeys.py b/getsshkeys.py
index fbfc65e..68d2945 100755 (executable)
--- a/getsshkeys.py
+++ b/getsshkeys.py
@@ -6,7 +6,7 @@ import string
 import time
 import xml, xmlrpclib
 try:
-       import config
+       from monitor import config
        auth = {'Username'   : config.API_AUTH_USER,
                'AuthMethod' : "password",
                        'AuthString' : config.API_AUTH_PASSWORD}
@@ -17,7 +17,12 @@ except:
 
 args = {}
 args['known_hosts'] =  os.environ['HOME'] + os.sep + ".ssh" + os.sep + "known_hosts"
-args['XMLRPC_SERVER'] = 'https://boot.planet-lab.org/PLCAPI/'
+try:
+       import config
+       args['XMLRPC_SERVER'] = config.API_SERVER
+except:
+       args['XMLRPC_SERVER'] = 'https://boot.planet-lab.org/PLCAPI/'
+       print "Using default API server %s" %  args['XMLRPC_SERVER']
 
 class SSHKnownHosts:
        def __init__(self, args = args):