iotlab driver fixed for OneLab in Production
[sfa.git] / sfa / iotlab / iotlabdriver.py
index bbe362f..855dc35 100644 (file)
@@ -135,7 +135,7 @@ class IotlabDriver(Driver):
 
 
 
-    def AddPerson(self, record, add_to_ldap = True):
+    def AddPerson(self, record):
         """
 
         Adds a new account. Any fields specified in records are used,
@@ -150,20 +150,16 @@ class IotlabDriver(Driver):
         :rtype: dict
 
         """
-        if not add_to_ldap:
-            ret = self.__add_person_to_db(record)
-            return ret
-
         ret = self.testbed_shell.ldap.LdapAddUser(record)
 
         if ret['bool'] is True:
-            record['hrn'] = self.testbed_shell.root_auth + '.' + ret['uid']
+            #record['hrn'] = self.testbed_shell.root_auth + '.' + ret['uid']
             logger.debug("IOTLAB_API AddPerson return code %s record %s  "
                          % (ret, record))
-            self.__add_person_to_db(record)
+            #self.__add_person_to_db(record)
         return ret
 
-    def __add_person_to_db(self, user_dict):
+    def add_person_to_db(self, user_dict):
         """
         Add a federated user straight to db when the user issues a lease
         request with iotlab nodes and that he has not registered with iotlab
@@ -177,7 +173,6 @@ class IotlabDriver(Driver):
         """
         query = self.api.dbsession().query(RegUser)
         check_if_exists = query.filter_by(email = user_dict['email']).first()
-        logger.debug("LOIC __add_person_to_db %s" % check_if_exists)
         #user doesn't exists
         if not check_if_exists:
             logger.debug("__add_person_to_db \t Adding %s \r\n \r\n \
@@ -196,8 +191,9 @@ class IotlabDriver(Driver):
 
             if pubkey is not None and pkey is not None :
                 hierarchy = Hierarchy()
+                # We fake the parent in order to be able to create a valid GID
                 person_gid = hierarchy.create_gid(person_urn, create_uuid(), \
-                                pkey)
+                                pkey, force_parent='iotlab')
                 if user_dict['email']:
                     logger.debug("__add_person_to_db \r\n \r\n \
                         IOTLAB IMPORTER PERSON EMAIL OK email %s "\
@@ -310,7 +306,6 @@ class IotlabDriver(Driver):
         .. seealso::_sql_get_slice_info_from_user
         .. seealso:: _sql_get_slice_info
         """
-        logger.debug("JORDAN get_slice_records slice_filter=%r slice_filter_type=%r" % (slice_filter, slice_filter_type))
 
         #Get list of slices based on the slice hrn
         if slice_filter_type == 'slice_hrn':
@@ -369,10 +364,8 @@ class IotlabDriver(Driver):
 
         #First try to get information on the slice based on the filter provided
         if slice_filter_type in authorized_filter_types_list:
-            logger.debug("JORDAN GET SLICES 1")
             fixed_slicerec_dict = self._get_slice_records(slice_filter,
                                                     slice_filter_type)
-            logger.debug("JORDAN GET SLICE RECORDS %r" % fixed_slicerec_dict)
             # if the slice was not found in the sfa db
             if fixed_slicerec_dict is None:
                 return return_slicerec_dictlist
@@ -394,14 +387,12 @@ class IotlabDriver(Driver):
             #If no job is running or no job scheduled
             #return only the slice record
             if leases_list == [] and fixed_slicerec_dict:
-                logger.debug("JORDAN CASE 1")
                 return_slicerec_dictlist.append(fixed_slicerec_dict)
 
             # if the jobs running don't belong to the user/slice we are looking
             # for
             leases_hrn = [lease['slice_hrn'] for lease in leases_list]
             if slice_hrn not in leases_hrn:
-                logger.debug("JORDAN CASE 2")
                 return_slicerec_dictlist.append(fixed_slicerec_dict)
             #If several jobs for one slice , put the slice record into
             # each lease information dict
@@ -446,7 +437,6 @@ class IotlabDriver(Driver):
 
 
         else:
-            logger.debug("JORDAN GET SLICES 2")
             #Get all slices from the iotlab sfa database ,
             #put them in dict format
             #query_slice_list = dbsession.query(RegRecord).all()
@@ -723,10 +713,7 @@ class IotlabDriver(Driver):
         self.api.dbsession().add(sfa_record)
         self.api.dbsession().commit()
         #Update the reg-researchers dependency table
-        if user_record is not None:
-            sfa_record.reg_researchers = [user_record]
-        else:
-            sfa_record.reg_researchers = slice_record['reg-researchers']
+        sfa_record.reg_researchers = [user_record]
         self.api.dbsession().commit()
 
         return
@@ -1102,7 +1089,7 @@ class IotlabDriver(Driver):
 
 
 
-    def delete(self, slice_urns, options={}):
+    def delete(self, slice_urns, options=None):
         """
         Deletes the lease associated with the slice hrn and the credentials
             if the slice belongs to iotlab. Answer to DeleteSliver.
@@ -1120,6 +1107,7 @@ class IotlabDriver(Driver):
         .. note:: creds are unused, and are not used either in the dummy driver
              delete_sliver .
         """
+        if options is None: options={}
         # collect sliver ids so we can update sliver allocation states after
         # we remove the slivers.
         aggregate = IotlabAggregate(self)
@@ -1412,7 +1400,8 @@ class IotlabDriver(Driver):
             'geni_ad_rspec_versions': ad_rspec_versions}
 
     # first 2 args are None in case of resource discovery
-    def list_resources (self, version=None, options={}):
+    def list_resources (self, version=None, options=None):
+        if options is None: options={}
         aggregate = IotlabAggregate(self)
         rspec =  aggregate.list_resources(version=version, options=options)
         return rspec
@@ -1421,7 +1410,8 @@ class IotlabDriver(Driver):
         aggregate = IotlabAggregate(self)
         return aggregate.describe(urns, version=version, options=options)
 
-    def status (self, urns, options={}):
+    def status (self, urns, options=None):
+        if options is None: options={}
         aggregate = IotlabAggregate(self)
         desc =  aggregate.describe(urns, version='GENI 3')
         status = {'geni_urn': desc['geni_urn'],
@@ -1429,7 +1419,8 @@ class IotlabDriver(Driver):
         return status
 
 
-    def allocate (self, urn, rspec_string, expiration, options={}):
+    def allocate (self, urn, rspec_string, expiration, options=None):
+        if options is None: options={}
         xrn = Xrn(urn)
         aggregate = IotlabAggregate(self)
 
@@ -1480,15 +1471,6 @@ class IotlabDriver(Driver):
         # ensure site record exists
 
         # ensure person records exists
-
-        # oui c'est degueulasse, le slice_record se retrouve modifie
-        # dans la methode avec les infos du user, els infos sont propagees
-        # dans verify_slice_leases
-        logger.debug("IOTLABDRIVER.PY  BEFORE slices.verify_persons")
-        # XXX JORDAN XXX slice_record devrait recevoir le caller_xrn...
-        logger.debug("LOIC users = %r" % users)
-        # XXX LOIC XXX Trying here to bypass the verify_persons function
-        # But it doesn't work, It seems we have to add users in LDAP anyway...
         for user in users:
             # XXX LOIC using hrn is a workaround because the function 
             # Xrn.get_urn returns 'urn:publicid:IDN+onelab:upmc+timur_friedman'
@@ -1505,23 +1487,31 @@ class IotlabDriver(Driver):
                 #slice_record['login'] = user['login']
                 slice_record['user']=user
 
-        # XXX LOIC XXX Need to re-activate this function and understand exactly what is required
+        # oui c'est degueulasse, le slice_record se retrouve modifie
+        # dans la methode avec les infos du user, els infos sont propagees
+        # dans verify_slice_leases
+        logger.debug("IOTLABDRIVER.PY  BEFORE slices.verify_persons")
 
-        persons = slices.verify_persons(xrn.hrn, slice_record, [slice_record['user']], options=options)
+        # XXX JORDAN XXX slice_record devrait recevoir le caller_xrn...
+        # LOIC maintenant c'est fait au dessus
+        logger.debug("IOTLABDRIVER.PY - LOIC - slice_record[user][hrn] = %s" % slice_record['user']['hrn'])
+        logger.debug("IOTLABDRIVER.PY - LOIC - slice_record[reg-researchers] = %s" % slice_record['reg-researchers'])
+        persons = slices.verify_persons(xrn.hrn, slice_record, users,
+                                        options=options)
         logger.debug("IOTLABDRIVER.PY  AFTER slices.verify_persons")
-        logger.debug("LOIC - IOTLABDRIVER.PY - AFTER slices.verify_persons slice_record = %r" % slice_record)
-        # ensure slice attributes exists
-        # XXX LOIC XXX This method doesn't exist !!!
-        #slices.verify_slice_attributes(slice, requested_attributes, options=options)
+        logger.debug("LOIC - slice_record[user] = %s" % slice_record['user'])
+        logger.debug("IOTLABDRIVER.PY - LOIC - slice_record[reg-researchers] = %s" % slice_record['reg-researchers'])
 
-        # XXX LOIC !!! Essayons d'inverser : d'abord verify_persons ensuite verify_slice
-        
         # ensure slice record exists
-        # XXX LOIC !!! verify_slice uniquement pour iotlab - les slices externes ne doivent pas être vérifiés = Ils ne sont pas dans le registry IOTLAB
-        current_slice = None
-        #current_slice = slices.verify_slice(xrn.hrn, slice_record, sfa_peer)
-        #logger.debug("IOTLABDRIVER.PY \t ===============allocate \t\
-        #                    \r\n \r\n  current_slice %s" % (current_slice))
+        current_slice = slices.verify_slice(xrn.hrn, slice_record, sfa_peer)
+        logger.debug("LOIC - AFTER verify_slice - slice_record[user] = %s" % slice_record['user'])
+        logger.debug("IOTLABDRIVER.PY - LOIC - slice_record[reg-researchers] = %s" % slice_record['reg-researchers'])
+        logger.debug("IOTLABDRIVER.PY \t ===============allocate \t\
+                            \r\n \r\n  current_slice %s" % (current_slice))
+
+        # ensure slice attributes exists
+        # slices.verify_slice_attributes(slice, requested_attributes,
+                                    # options=options)
 
         # add/remove slice from nodes
         # XXX JORDAN ensure requested_xp_dict returns a dict with all new leases
@@ -1531,6 +1521,7 @@ class IotlabDriver(Driver):
                      % (requested_xp_dict))
         request_nodes = rspec.version.get_nodes_with_slivers()
 
+
         # JORDAN: nodes_list will contain a list of newly allocated nodes
         nodes_list = []
         for start_time in requested_xp_dict:
@@ -1552,43 +1543,28 @@ class IotlabDriver(Driver):
         logger.debug("IOTLABDRIVER.PY \tallocate leases  %s \
                         rspec_requested_leases %s" % (leases,
                         rspec_requested_leases))
-
-        # XXX LOIC !!! What is in the slice_record?
-        # Where is affected reg_researchers value???
-        logger.debug("LOIC - IOTLABDRIVER.PY - After verify_slice_leases slice_record = %r" %slice_record)
         # update sliver allocations
         # JORDAN Here we loop over newly allocated nodes
         for hostname in nodes_list:
-            logger.debug("FORLOOP JORDAN hostname=%r" % hostname)
             client_id = hostname
             node_urn = xrn_object(self.testbed_shell.root_auth, hostname).urn
             component_id = node_urn
-            if current_slice is not None:
-                if 'reg-urn' in current_slice:
-                    slice_urn = current_slice['reg-urn']
-                else:
-                    slice_urn = current_slice['urn']
+            if 'reg-urn' in current_slice:
+                slice_urn = current_slice['reg-urn']
             else:
-                slice_urn = slice_record['urn']
+                slice_urn = current_slice['urn']
+
             # JORDAN: We loop over leases previously in the slice
-            for lease in leases: # rspec_requested_leases ?????? XXX
-                logger.debug("FOR LEASE LOOP JORDAN lease=%r" % lease)
-                logger.debug("JORDAN hostname=%r lease['reserved_nodes']=%r, bool=%r" % (hostname, lease['reserved_nodes'], hostname in lease['reserved_nodes']))
+            for lease in leases:
                 if hostname in lease['reserved_nodes']:
-                    logger.debug("JORDAN IF OK")
                     index = lease['reserved_nodes'].index(hostname)
-                    logger.debug("JORDAN index=%r" % index)
                     sliver_hrn = '%s.%s-%s' % (self.hrn, lease['lease_id'],
                                    lease['resource_ids'][index] )
-
-                    logger.debug("LOIC sliver_hrn=%r" % sliver_hrn)
                     sliver_id = Xrn(sliver_hrn, type='sliver').urn
-
-                    logger.debug("LOIC sliver_id=%r" % sliver_id)
                     record = SliverAllocation(sliver_id=sliver_id, client_id=client_id,
-                                              component_id=component_id,
-                                              slice_urn = slice_urn,
-                                              allocation_state='geni_allocated')
+                                      component_id=component_id,
+                                      slice_urn = slice_urn,
+                                      allocation_state='geni_allocated')
                     record.sync(self.api.dbsession())
 
         # JORDAN : added describe_options which was not specified at all
@@ -1598,13 +1574,13 @@ class IotlabDriver(Driver):
         }
         return aggregate.describe([xrn.get_urn()], version=rspec.version, options=describe_options)
 
-    def provision(self, urns, options={}):
+    def provision(self, urns, options=None):
+        if options is None: options={}
         # update users
         slices = IotlabSlices(self)
         aggregate = IotlabAggregate(self)
         slivers = aggregate.get_slivers(urns)
         current_slice = slivers[0]
-        logger.debug("Provision current slice: %r" % (current_slice,))
         peer = slices.get_peer(current_slice['hrn'])
         sfa_peer = slices.get_sfa_peer(current_slice['hrn'])
         users = options.get('geni_users', [])