Reactivated PI get emails
[myslice.git] / portal / actions.py
index 3be1176..3ee568c 100644 (file)
@@ -36,6 +36,8 @@ def authority_get_pi_emails(request, authority_hrn):
 # SFA add record (user, slice)
 
 def sfa_add_user(request, user_params):
+    if 'email' in user_params:
+        user_params['user_email'] = user_params['email']
     query = Query.create('user').set(user_params).select('user_hrn')
     results = execute_query(request, query)
     if not results:
@@ -44,6 +46,8 @@ def sfa_add_user(request, user_params):
 
 def sfa_update_user(request, user_hrn, user_params):
     # user_params: keys [public_key] 
+    if 'email' in user_params:
+        user_params['user_email'] = user_params['email']
     query = Query.update('user').filter_by('user_hrn', '==', user_hrn).set(user_params).select('user_hrn')
     results = execute_query(request,query)
     return results
@@ -64,8 +68,14 @@ def sfa_add_authority(request, authority_params):
     return results
 
 def sfa_add_user_to_slice(request, user_hrn, slice_params):
+# UPDATE myslice:slice SET researcher=['ple.upmc.jordan_auge','ple.inria.thierry_parmentelat','ple.upmc.loic_baron','ple.upmc.ciro_scognamiglio','ple.upmc.mohammed-yasin_rahman','ple.upmc.azerty'] where slice_hrn=='ple.upmc.myslicedemo'
+    query_current_users = Query.get('slice').select('user').filter_by('slice_hrn','==',slice_params['hrn'])
+    results_current_users = execute_query(request, query_current_users)
+    slice_params['researcher'] = slice_params['researcher'] | results_current_users
     query = Query.update('slice').filter_by('user_hrn', '==', user_hrn).set(slice_params).select('slice_hrn')
     results = execute_query(request, query)
+# Also possible but not supported yet
+# UPDATE myslice:user SET slice=['ple.upmc.agent','ple.upmc.myslicedemo','ple.upmc.tophat'] where user_hrn=='ple.upmc.azerty'
     if not results:
         raise Exception, "Could not create %s. Already exists ?" % slice_params['hrn']
     return results
@@ -145,7 +155,7 @@ def make_request_slice(slice):
     request = {}
     request['type'] = 'slice'
     request['id'] = slice.id
-    request['user_email'] = slice.user_email
+    request['user_hrn'] = slice.user_hrn
     request['timestamp'] = slice.created
     request['authority_hrn'] = slice.authority_hrn
     request['slice_name'] = slice.slice_name
@@ -202,7 +212,7 @@ def get_request_by_id(ids):
 
     return make_requests(pending_users, pending_slices, pending_authorities)
 
-def get_request_by_authority(authority_hrns):
+def get_requests(authority_hrns=None):
     print "get_request_by_authority auth_hrns = ", authority_hrns
     if not authority_hrns:
         pending_users  = PendingUser.objects.all()
@@ -214,7 +224,7 @@ def get_request_by_authority(authority_hrns):
         pending_authorities = PendingAuthority.objects.filter(authority_hrn__in=authority_hrns).all()
 
     return make_requests(pending_users, pending_slices, pending_authorities)
-    
+
 # XXX Is it in sync with the form fields ?
 
 def portal_validate_request(wsgi_request, request_ids):
@@ -242,7 +252,10 @@ def portal_validate_request(wsgi_request, request_ids):
                 # XXX tmp sfa dependency
                 from sfa.util.xrn import Xrn 
                 urn = Xrn(hrn, request['type']).get_urn()
-
+                if 'pi' in request:
+                    auth_pi = request['pi']
+                else:
+                    auth_pi = ''
                 sfa_user_params = {
                     'hrn'        : hrn, 
                     'urn'        : urn,
@@ -253,30 +266,58 @@ def portal_validate_request(wsgi_request, request_ids):
                     'email'      : request['email'],
                     #'slices'    : None,
                     #'researcher': None,
-                    #'pi'        : None,
+                    'pi'         : [auth_pi],
                     'enabled'    : True
                 }
                 # ignored in request: id, timestamp, password
+                
+                # ADD USER TO SFA Registry
+                sfa_add_user(wsgi_request, sfa_user_params)
 
-                # UPDATE user status = 2 = validated
-                user_query  = Query().get('local:user').select('config','email','status').filter_by('email', '==', request['email'])
+                # USER INFO
+                user_query  = Query().get('local:user').select('user_id','config','email','status').filter_by('email', '==', request['email'])
                 user_details = execute_admin_query(request, user_query)
-                print user_details[0]
+                #print user_details[0]
+
+                # UPDATE USER STATUS = 2
                 manifold_user_params = {
                     'status': 2
                 }
                 manifold_update_user(request, request['email'], manifold_user_params) 
-                sfa_add_user(wsgi_request, sfa_user_params)
-               # XXX Remove from database
 
+                # USER MAIN ACCOUNT != reference
+                #print 'USER MAIN ACCOUNT != reference'
+                list_accounts_query  = Query().get('local:account').select('user_id','platform_id','auth_type','config')\
+                    .filter_by('user_id','==',user_details[0]['user_id'])\
+                    .filter_by('auth_type','!=','reference')    
+                list_accounts = execute_admin_query(request, list_accounts_query)
+                #print "List accounts = ",list_accounts
+                for account in list_accounts:
+                    main_platform_query  = Query().get('local:platform').select('platform_id','platform').filter_by('platform_id','==',account['platform_id'])
+                    main_platform = execute_admin_query(request, main_platform_query)
+
+                # ADD REFERENCE ACCOUNTS ON SFA ENABLED PLATFORMS                        
+                #print 'ADD REFERENCE ACCOUNTS ON SFA ENABLED PLATFORMS'
+                platforms_query  = Query().get('local:platform').filter_by('disabled', '==', '0').filter_by('gateway_type','==','sfa').select('platform_id','gateway_type')
+                platforms = execute_admin_query(request, platforms_query)
+                #print "platforms SFA ENABLED = ",platforms
+                for platform in platforms:
+                    #print "add reference to platform ",platform
+                    manifold_account_params = {
+                        'user_id': user_details[0]['user_id'],
+                        'platform_id': platform['platform_id'],
+                        'auth_type': 'reference',
+                        'config': '{"reference_platform": "' + main_platform[0]['platform'] + '"}',
+                    }
+                    manifold_add_account(request, manifold_account_params)
+        
                 request_status['SFA user'] = {'status': True }
 
             except Exception, e:
-                request_status['SFA user'] = {'status': False, 'description': str(e)}
-            
-            user_params = {'status':2}
-            manifold_update_user(request, request['email'],user_params)
+                 request_status['SFA user'] = {'status': False, 'description': str(e)}
+                       
+#            user_params = {'status':2}
+#            manifold_update_user(request, request['email'], user_params)
 
             # MANIFOLD user should be added beforehand, during registration
             #try:
@@ -297,13 +338,20 @@ def portal_validate_request(wsgi_request, request_ids):
                 # XXX tmp sfa dependency
                 from sfa.util.xrn import Xrn 
                 urn = Xrn(hrn, request['type']).get_urn()
-
+                
+                # Add User to Slice if we have the user_hrn in pendingslice table
+                if 'user_hrn' in request:
+                    user_hrn = request['user_hrn']
+                    print "Slice %s will be created for %s" % (hrn,request['user_hrn'])
+                else:
+                    user_hrn=''
+                    print "Slice %s will be created without users %s" % (hrn)
                 sfa_slice_params = {
                     'hrn'        : hrn, 
                     'urn'        : urn,
                     'type'       : request['type'],
                     #'slices'    : None,
-                    #'researcher': None,
+                    'researcher' : [user_hrn],
                     #'pi'        : None,
                     'enabled'    : True
                 }