Modification on slabimporter to avoid creating an account when importing

[sfa.git] / sfa / importer / slabimporter.py

diff --git a/sfa/importer/slabimporter.py b/sfa/importer/slabimporter.py
index e74fbf1..d9e34bc 100644 (file)
--- a/sfa/importer/slabimporter.py
+++ b/sfa/importer/slabimporter.py
@@ -4,7 +4,7 @@ from sfa.util.config import Config
 from sfa.util.xrn import Xrn, get_authority, hrn_to_urn
 
 from sfa.senslab.slabdriver import SlabDriver
-from sfa.senslab.slabpostgres import SliceSenslab, slab_dbsession
+from sfa.senslab.slabpostgres import SenslabXP, slab_dbsession
 
 from sfa.trust.certificate import Keypair,convert_public_key
 from sfa.trust.gid import create_uuid
@@ -63,9 +63,9 @@ class SlabImporter:
         
         #Create special slice table for senslab 
         
-        if not slabdriver.db.exists('slice_senslab'):
+        if not slabdriver.db.exists('slab_xp'):
             slabdriver.db.createtable()
-            self.logger.info ("SlabImporter.run:  slice_senslab table created ")
+            self.logger.info ("SlabImporter.run:  slab_xp table created ")
 
         #retrieve all existing SFA objects
         all_records = dbsession.query(RegRecord).all()
@@ -75,6 +75,9 @@ class SlabImporter:
        
         self.records_by_type_hrn = \
             dict ( [ ( (record.type,record.hrn) , record ) for record in all_records ] )
+        print>>sys.stderr,"\r\n SLABIMPORT \t all_records[0] %s all_records[0].email %s \r\n" %(all_records[0].type, all_records[0])
+        self.users_rec_by_email = \
+            dict ( [ (record.email, record) for record in all_records if record.type == 'user' ] )
             
         # create hash by (type,pointer) 
         self.records_by_type_pointer = \
@@ -92,7 +95,7 @@ class SlabImporter:
         print>>sys.stderr,"\r\n SLABIMPORT \t ldap_person_listdict %s \r\n" %(ldap_person_listdict)
         slices_listdict = slabdriver.GetSlices()
         try:
-            slices_by_userid = dict ( [ (one_slice['record_id_user'], one_slice ) for one_slice in slices_listdict ] )
+            slices_by_userid = dict ( [ (one_slice['reg_researchers']['record_id'], one_slice ) for one_slice in slices_listdict ] )
         except TypeError:
              self.logger.log_exc("SlabImporter: failed to create list of slices by user id.") 
              pass
@@ -167,6 +170,8 @@ class SlabImporter:
                     
         # import persons
         for person in ldap_person_listdict : 
+            
+
             print>>sys.stderr,"SlabImporter: person: %s" %(person['hrn'])
             if 'ssh-rsa' not in person['pkey']:
                 #people with invalid ssh key (ssh-dss, empty, bullshit keys...)
@@ -178,8 +183,21 @@ class SlabImporter:
             # xxx suspicious again
             if len(person_hrn) > 64: person_hrn = person_hrn[:64]
             person_urn = hrn_to_urn(person_hrn, 'user')
-
+            
+            
+            print>>sys.stderr," \r\n SlabImporter:  HEYYYYYYYYYY" , self.users_rec_by_email
+            
+            #Check if user using person['email'] form LDAP is already registered
+            #in SFA. One email = one person. Inb this case, do not create another
+            #record for this person
+            #person_hrn  returned by GetPErson based on senslab root auth + uid ldap
             user_record = self.find_record_by_type_hrn('user', person_hrn)
+            if not user_record and  person['email'] in self.users_rec_by_email:
+                user_record = self.users_rec_by_email[person['email']]
+                person_hrn = user_record.hrn
+                person_urn = hrn_to_urn(person_hrn, 'user')
+                
+            
             slice_record = self.find_record_by_type_hrn ('slice', slice_hrn)
             
             # return a tuple pubkey (a plc key object) and pkey (a Keypair object)
@@ -257,8 +275,7 @@ class SlabImporter:
                         
                 dbsession.commit()
 
-                user_record.stale=False
-                print>>sys.stderr,"SlabImporter: STALE!! PERSON : %s" %user_record
+                user_record.stale = False
             except:
                 self.logger.log_exc("SlabImporter: failed to import person  %s"%(person) )       
             
@@ -284,11 +301,11 @@ class SlabImporter:
                     #Get it
                     sl_rec = dbsession.query(RegSlice).filter(RegSlice.hrn.match(slice_hrn)).all()
                     
-                    slab_slice = SliceSenslab( slice_hrn = slice_hrn, record_id_slice=sl_rec[0].record_id, record_id_user= user_record.record_id)
-                    print>>sys.stderr, "\r\n \r\n SLAB IMPORTER SLICE IMPORT NOTslice_record %s \r\n slab_slice %s" %(sl_rec,slab_slice)
-                    slab_dbsession.add(slab_slice)
-                    slab_dbsession.commit()
-                    self.logger.info("SlabImporter: imported slice: %s" % slice_record)  
+                    #slab_slice = SenslabXP( slice_hrn = slice_hrn, record_id_slice=sl_rec[0].record_id, record_id_user= user_record.record_id)
+                    #print>>sys.stderr, "\r\n \r\n SLAB IMPORTER SLICE IMPORT NOTslice_record %s \r\n slab_slice %s" %(sl_rec,slab_slice)
+                    #slab_dbsession.add(slab_slice)
+                    #slab_dbsession.commit()
+                    #self.logger.info("SlabImporter: imported slice: %s" % slice_record)  
                     self.update_just_added_records_dict ( slice_record )
 
                 except:
@@ -328,10 +345,10 @@ class SlabImporter:
                 self.logger.warning("stale not found with %s"%record)
             if stale:
                 self.logger.info("SlabImporter: deleting stale record: %s" % record)
-                if record.type == 'user':
-                    rec = slab_dbsession.query(SliceSenslab).filter_by(record_id_user = record.record_id).first()
-                    slab_dbsession.delete(rec)
-                    slab_dbsession.commit()
+                #if record.type == 'user':
+                    #rec = slab_dbsession.query(SenslabXP).filter_by(record_id_user = record.record_id).first()
+                    #slab_dbsession.delete(rec)
+                    #slab_dbsession.commit()
                 dbsession.delete(record)
                 dbsession.commit()