--- /dev/null
+#!/usr/bin/python
+
+from monitor.wrapper import plc
+import sys
+import time
+
+email_list = [ 'soltesz@cs.princeton.edu',
+ # 'soltesz@cs.princeton.edu',
+ # 'justin@cs.arizona.edu',
+ # 'bakers@cs.arizona.edu',
+ # 'jhh@cs.arizona.edu',
+ # 'mfreed@cs.nyu.edu',
+ # 'kyoungso@cs.princeton.edu',
+ # 'nspring@cs.umd.edu',
+ # 'vivek@cs.princeton.edu',
+ ]
+
+from monitor import config
+api = plc.PLC(plc.Auth(config.API_AUTH_USER, config.API_AUTH_PASSWORD).auth, config.API_SERVER)
+api06 = plc.PLC(plc.Auth(config.API06_AUTH_USER, config.API06_AUTH_PASSWORD).auth, config.API06_SERVER)
+
+# add planetlab-15.cs.princeton.edu, and use the key on the CD.
+#id = api06.AddNode(1, {'boot_state': 'rins', 'model': 'Dell Optiplex',
+# 'hostname' : 'planetlab-15.cs.princeton.edu',
+# 'version' : '3.3'})
+#api06.AddNodeNetwork(id, {'ip': '128.112.139.39',
+# 'type' : 'ipv4',
+## 'is_primary' : True,
+# 'method' : 'dhcp', })
+#api06.UpdateNode(id, {'key': "wptNagk8SgRxzN1lXfKMAjUYhQbOBymKnKg9Uv0LwGM"})
+#
+
+#print "adding vsys attributes"
+#api06.AddSliceAttribute('princeton_slicestat', 'vsys', 'pl-ps')
+#api06.AddSliceAttribute('princeton_slicestat', 'vsys', 'vtop')
+#api06.AddSliceAttribute('pl_netflow', 'vsys', 'pfmount')
+
+#print "preserve princeton_chopstix"
+
+#sys.exit(1)
+
+#attr_types = api.GetSliceAttributeTypes()
+#attr_types06 = api06.GetSliceAttributeTypes()
+#attr_types06_names = [a['name'] for a in attr_types06]
+#for type in attr_types:
+# if type['name'] not in attr_types06_names:
+# print "adding %s " % type
+# api06.AddSliceAttributeType(type)
+def person_exists(user):
+ try:
+ x = api06.GetPersons({'email':user['email']})
+ if len(x) == 0:
+ return False
+ else:
+ return True
+ except:
+ return False
+
+def site_exists(site):
+ try:
+ x = api06.GetSites({'login_base':site['login_base']})
+ if len(x) == 0:
+ return False
+ else:
+ return True
+ except:
+ return False
+
+def slice_exists(slice):
+ try:
+ x = api06.GetSlices({'name':slice['name']})
+ if len(x) == 0:
+ return False
+ else:
+ return True
+ except:
+ return False
+
+def node_exists(node):
+ x = api06.GetNodes({'hostname':node['hostname']})
+ if len(x) == 0:
+ return False
+ else:
+ return True
+
+# Renew slices
+slices = api06.GetSlices()
+for slice in slices:
+ print "Updating expiration of %s" % slice['name']
+ api06.UpdateSlice(slice['name'], {'expires': int(time.time()) + 7*24*60*60})
+
+#sys.exit(1)
+
+for email in email_list:
+ user = api.GetPersons({'email': email})
+ if len(user) == 0:
+ print "User not found: %s" % email
+ continue
+
+ user = user[0]
+
+ print "adding person %s " % user['email']
+ if not person_exists(user):
+ api06.AddPerson(user)
+
+ api06.UpdatePerson(email, {'enabled': True})
+
+ print "adding person keys %s " % user['email']
+ key = api.GetKeys({'person_id': user['person_id']})[0]
+ key06 = {'key': key['key'], 'key_type': key['key_type']}
+ api06.AddPersonKey(user['email'], key06)
+
+ print "updating person roles: ",
+ for role in user['roles']:
+ print "%s" % role,
+ api06.AddRoleToPerson(role, user['email'])
+ sys.stdout.flush()
+ print ""
+
+ sites = api.GetSites(user['site_ids'])
+ print "Adding sites:",
+ for site in sites:
+ nodes = api.GetNodes(site['node_ids'])
+
+ if not site_exists(site):
+ print "%s" % site['login_base']
+ api06.AddSite(site)
+ api06.AddPersonToSite(user['email'], site['login_base'])
+ sys.stdout.flush()
+
+ for node in nodes:
+ if not node_exists(node):
+ id = api06.AddNode(site['login_base'], {'boot_state': node['boot_state'],
+ 'model': node['model'],
+ 'hostname' : node['hostname'],
+ 'version' : node['version']})
+ nnets = api.GetInterfaces(node['interface_ids'])
+ for nnet in nnets:
+ del nnet['interface_id']
+ del nnet['nodenetwork_setting_ids']
+ api06.AddNodeNetwork(id, nnet)
+ print ""
+
+ nodes = api06.GetNodes()
+ slices = api.GetSlices(user['slice_ids'])
+ for slice in slices:
+ # create slice
+ no_add = True
+ if not slice_exists(slice):
+ print "Adding slice %s" % slice['name']
+ try:
+ api06.AddSlice(slice)
+ attr = api.GetSliceAttributes({'slice_attribute_id' :
+ slice['slice_attribute_ids']})
+ print "adding attributes:",
+ added_attr = []
+ for a in attr:
+ print "%s" % a['name'],
+ #if a['name'] not in added_attr:
+ api06.AddSliceAttribute(slice['name'], a['name'], a['value'])
+ # added_attr.append(a['name'])
+ sys.stdout.flush()
+ print ""
+ except:
+ no_add = False
+ print "error with ", slice['name']
+
+ if no_add:
+ print "adding nodes and %s to slice %s" % (user['email'], slice['name'])
+ # add all api06 nodes to slice
+ api06.AddSliceToNodes(slice['name'], [n['hostname'] for n in nodes])
+ # add user to slice
+ api06.AddPersonToSlice(user['email'], slice['name'])
git://git.onelab.eu / monitor.git / blobdiff