some fixes and files moved

[planetlab-umts-tools.git] / utils / add_to_nodegroup.py

diff --git a/utils/add_to_nodegroup.py b/utils/add_to_nodegroup.py
new file mode 100644 (file)
index 0000000..6209a99
--- /dev/null
+++ b/utils/add_to_nodegroup.py
@@ -0,0 +1,62 @@
+#!/usr/bin/python
+
+# adds a node to a nodegroup - to be called with an argument (the ip or hostname of the plc)
+
+import sys
+import xmlrpclib
+
+import sys
+import getpass
+
+
+sys.path.append('/usr/share/plc_api')
+
+
+if (len(sys.argv)<2):
+       print('I need the first argument (the pi or hostname address of the plc)')
+       exit(1)
+
+plc_ip=sys.argv[1]
+
+
+user = raw_input('Insert your user:')
+password= getpass.getpass('Insert your password:')
+#password = raw_input('Insert your password:')
+hostname = raw_input('Insert the node hostname:')
+nodegroup=raw_input('Insert the group:')
+
+
+auth = {}
+
+auth['AuthMethod'] = 'password'
+auth['Username'] = user
+auth['AuthString'] = password
+
+
+plc = xmlrpclib.ServerProxy('https://'+plc_ip+'/PLCAPI/', allow_none=True)
+
+       
+authorized = plc.AuthCheck(auth)
+
+if authorized:
+       print 'We are authorized!'
+       
+node_fields = {'nodegroups' : [nodegroup] }
+
+nodes = plc.GetNodes(auth, hostname, ['node_id'])
+
+if len(nodes)>1:
+       print 'I found  %d node with that hostname' % len(nodes)
+       exit(1)
+       
+if len(nodes)==0:
+       print 'I haven\'t found any nodes with that hostname'
+       exit(1)
+
+node_id=nodes[0]['node_id']
+
+if plc.UpdateNode (auth, node_id, node_fields) != 1:
+       print "Couldn't set the nodegroup - error in UpdateNode api";
+else:
+       print 'Successfully added node to nodegroup!'
+