portal/actions.py

   1 from django.http                 import HttpResponse
   2 from manifold.core.query         import Query
   3 from manifoldapi.manifoldapi        import execute_query,execute_admin_query
   4 from portal.models               import PendingUser, PendingSlice, PendingAuthority
   5 import json
   6
   7 # Thierry: moving this right into the code so
   8 # most people can use myslice without having to install sfa
   9 # XXX tmp sfa dependency, should be moved to SFA gateway
  10 #from sfa.util.xrn                import Xrn
  11
  12
  13 # Get the list of authorities
  14
  15 def authority_get_pis(request, authority_hrn):
  16     query = Query.get('authority').filter_by('authority_hrn', '==', authority_hrn).select('pi_users')
  17     results = execute_admin_query(request, query)
  18     # NOTE: temporarily commented. Because results is giving empty list.
  19     # Needs more debugging
  20     #if not results:
  21     #    raise Exception, "Authority not found: %s" % authority_hrn
  22     #result, = results
  23     #return result['pi_users']
  24     return results
  25
  26 def authority_get_pi_emails(request, authority_hrn):
  27     #return ['jordan.auge@lip6.fr', 'loic.baron@lip6.fr']
  28     pi_users = authority_get_pis(request,authority_hrn)
  29     if any(d['pi_users'] == None for d in pi_users):
  30         return ['support@myslice.info']
  31     else:
  32         pi_user_hrns = [ hrn for x in pi_users for hrn in x['pi_users'] ]
  33         query = Query.get('user').filter_by('user_hrn', 'included', pi_user_hrns).select('email')
  34         results = execute_admin_query(request, query)
  35         print "mails",  [result['email'] for result in results]
  36         return [result['email'] for result in results]
  37
  38 # SFA get record
  39
  40 def sfa_get_user(request, user_hrn, pub):
  41     query_sfa_user = Query.get('user').filter_by('user_hrn', '==', user_hrn)
  42     result_sfa_user = execute_query(request, query_sfa_user)
  43     return result_sfa_user
  44
  45 # SFA add record (user, slice)
  46
  47 def sfa_add_user(request, user_params):
  48     if 'email' in user_params:
  49         user_params['user_email'] = user_params['email']
  50     query = Query.create('user').set(user_params).select('user_hrn')
  51     results = execute_query(request, query)
  52     if not results:
  53         raise Exception, "Could not create %s. Already exists ?" % user_params['hrn']
  54     return results
  55
  56 def sfa_update_user(request, user_hrn, user_params):
  57     # user_params: keys [public_key]
  58     if 'email' in user_params:
  59         user_params['user_email'] = user_params['email']
  60     query = Query.update('user').filter_by('user_hrn', '==', user_hrn).set(user_params).select('user_hrn')
  61     results = execute_query(request,query)
  62     return results
  63
  64 def sfa_add_slice(request, slice_params):
  65     query = Query.create('slice').set(slice_params).select('slice_hrn')
  66     results = execute_query(request, query)
  67     if not results:
  68         raise Exception, "Could not create %s. Already exists ?" % slice_params['hrn']
  69     return results
  70
  71 def sfa_add_authority(request, authority_params):
  72     query = Query.create('authority').set(authority_params).select('authority_hrn')
  73     results = execute_query(request, query)
  74     print "sfa_add_auth results=",results
  75     if not results:
  76         raise Exception, "Could not create %s. Already exists ?" % authority_params['hrn']
  77     return results
  78
  79 def sfa_add_user_to_slice(request, user_hrn, slice_params):
  80 # 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'
  81     query_current_users = Query.get('slice').select('user').filter_by('slice_hrn','==',slice_params['hrn'])
  82     results_current_users = execute_query(request, query_current_users)
  83     slice_params['researcher'] = slice_params['researcher'] | results_current_users
  84     query = Query.update('slice').filter_by('user_hrn', '==', user_hrn).set(slice_params).select('slice_hrn')
  85     results = execute_query(request, query)
  86 # Also possible but not supported yet
  87 # UPDATE myslice:user SET slice=['ple.upmc.agent','ple.upmc.myslicedemo','ple.upmc.tophat'] where user_hrn=='ple.upmc.azerty'
  88     if not results:
  89         raise Exception, "Could not create %s. Already exists ?" % slice_params['hrn']
  90     return results
  91
  92 # Propose hrn
  93
  94 def manifold_add_user(request, user_params):
  95     # user_params: email, password e.g., user_params = {'email':'aa@aa.com','password':'demo'}
  96     query = Query.create('local:user').set(user_params).select('email')
  97     results = execute_admin_query(request, query)
  98     if not results:
  99         raise Exception, "Failed creating manifold user: %s" % user_params['email']
 100     result, = results
 101     return result['email']
 102
 103 def manifold_update_user(request, email, user_params):
 104     # user_params: password, config e.g.,
 105     query = Query.update('local:user').filter_by('email', '==', email).set(user_params).select('email')
 106     results = execute_admin_query(request,query)
 107     # NOTE: results remains empty and goes to Exception. However, it updates the manifold DB.
 108     # That's why I commented the exception part. -- Yasin
 109     #if not results:
 110     #    raise Exception, "Failed updating manifold user: %s" % user_params['email']
 111     #result, = results
 112     return results
 113
 114 def manifold_add_account(request, account_params):
 115     query = Query.create('local:account').set(account_params).select(['user', 'platform'])
 116     results = execute_admin_query(request,query)
 117     if not results:
 118         raise Exception, "Failed creating manifold account on platform %s for user: %s" % (account_params['platform'], account_params['user'])
 119     result, = results
 120     return result['user_id']
 121
 122 def manifold_update_account(request,user_id,account_params):
 123     # account_params: config
 124     query = Query.update('local:account').filter_by('platform', '==', 'myslice').filter_by('user_id', '==', user_id).set(account_params).select('user_id')
 125     results = execute_admin_query(request,query)
 126     return results
 127
 128 #explicitly mention the platform_id
 129 def manifold_delete_account(request, platform_id, user_id, account_params):
 130     query = Query.delete('local:account').filter_by('platform_id', '==', platform_id).filter_by('user_id', '==', user_id).set(account_params).select('user_id')
 131     results = execute_admin_query(request,query)
 132     return results
 133
 134
 135 #not tested
 136 def manifold_add_platform(request, platform_params):
 137     query = Query.create('local:platform').set(platform_params).select(['user', 'platform'])
 138     results = execute_admin_query(request,query)
 139     if not results:
 140         raise Exception, "Failed creating manifold  platform %s for user: %s" % (platform_params['platform'], platform_params['user'])
 141     result, = results
 142     return result['platform_id']
 143
 144
 145 def make_request_user(user):
 146     request = {}
 147     request['type']          = 'user'
 148     request['id']            = user.id
 149     request['timestamp']     = user.created # XXX in DB ?
 150     request['authority_hrn'] = user.authority_hrn
 151     request['first_name']    = user.first_name
 152     request['last_name']     = user.last_name
 153     request['email']         = user.email
 154     request['login']         = user.login
 155     request['keypair']       = user.keypair
 156     return request
 157
 158 def make_request_slice(slice):
 159     request = {}
 160     request['type'] = 'slice'
 161     request['id'] = slice.id
 162     request['user_hrn'] = slice.user_hrn
 163     request['timestamp'] = slice.created
 164     request['authority_hrn'] = slice.authority_hrn
 165     request['slice_name'] = slice.slice_name
 166     request['number_of_nodes'] = slice.number_of_nodes
 167     request['type_of_nodes'] = slice.type_of_nodes
 168     request['purpose'] = slice.purpose
 169     return request
 170
 171 def make_request_authority(authority):
 172     request = {}
 173     request['type']                  = 'authority'
 174     request['id']                    = authority.id
 175     request['site_name']             = authority.site_name
 176     request['site_latitude']         = authority.site_latitude
 177     request['site_longitude']        = authority.site_longitude
 178     request['site_url']              = authority.site_url
 179     request['site_authority']        = authority.site_authority
 180     request['site_abbreviated_name'] = authority.site_abbreviated_name
 181     request['address_line1']         = authority.address_line1
 182     request['address_line2']         = authority.address_line2
 183     request['address_line3']         = authority.address_line3
 184     request['address_city']          = authority.address_city
 185     request['address_postalcode']    = authority.address_postalcode
 186     request['address_state']         = authority.address_state
 187     request['address_country']       = authority.address_country
 188     request['authority_hrn']         = authority.authority_hrn
 189     request['timestamp']             = authority.created
 190     return request
 191
 192 def make_requests(pending_users, pending_slices, pending_authorities):
 193     requests = []
 194     for user in pending_users:
 195         requests.append(make_request_user(user))
 196     for slice in pending_slices:
 197         requests.append(make_request_slice(slice))
 198     for authority in pending_authorities:
 199         requests.append(make_request_authority(authority))
 200     return requests
 201
 202 def get_request_by_id(ids):
 203     sorted_ids = { 'user': [], 'slice': [], 'authority': [] }
 204     for type__id in ids:
 205         type, id = type__id.split('__')
 206         sorted_ids[type].append(id)
 207
 208     if not ids:
 209         pending_users  = PendingUser.objects.all()
 210         pending_slices = PendingSlice.objects.all()
 211         pending_authorities = PendingAuthority.objects.all()
 212     else:
 213         pending_users  = PendingUser.objects.filter(id__in=sorted_ids['user']).all()
 214         pending_slices = PendingSlice.objects.filter(id__in=sorted_ids['slice']).all()
 215         pending_authorities = PendingAuthority.objects.filter(id__in=sorted_ids['authority']).all()
 216
 217     return make_requests(pending_users, pending_slices, pending_authorities)
 218
 219 def get_requests(authority_hrns=None):
 220     print "get_request_by_authority auth_hrns = ", authority_hrns
 221     if not authority_hrns:
 222         pending_users  = PendingUser.objects.all()
 223         pending_slices = PendingSlice.objects.all()
 224         pending_authorities = PendingAuthority.objects.all()
 225     else:
 226         pending_users  = PendingUser.objects.filter(authority_hrn__in=authority_hrns).all()
 227         pending_slices = PendingSlice.objects.filter(authority_hrn__in=authority_hrns).all()
 228         pending_authorities = PendingAuthority.objects.filter(authority_hrn__in=authority_hrns).all()
 229
 230     return make_requests(pending_users, pending_slices, pending_authorities)
 231
 232 # XXX Is it in sync with the form fields ?
 233
 234 def portal_validate_request(wsgi_request, request_ids):
 235     status = {}
 236
 237     if not isinstance(request_ids, list):
 238         request_ids = [request_ids]
 239
 240     requests = get_request_by_id(request_ids)
 241     for request in requests:
 242         # type, id, timestamp, details, allowed -- MISSING: authority_hrn
 243         # CAREFUL about details
 244         # user  : first name, last name, email, password, keypair
 245         # slice : number of nodes, type of nodes, purpose
 246
 247         request_status = {}
 248
 249         print "REQUEST", request
 250         if request['type'] == 'user':
 251
 252             try:
 253                 # XXX tmp user_hrn inside the keypair column of pendiguser table
 254                 hrn = json.loads(request['keypair'])['user_hrn']
 255                 #hrn = "%s.%s" % (request['authority_hrn'], request['login'])
 256                 # XXX tmp sfa dependency
 257                 from sfa.util.xrn import Xrn
 258                 urn = Xrn(hrn, request['type']).get_urn()
 259                 if 'pi' in request:
 260                     auth_pi = request['pi']
 261                 else:
 262                     auth_pi = ''
 263                 sfa_user_params = {
 264                     'hrn'        : hrn,
 265                     'urn'        : urn,
 266                     'type'       : request['type'],
 267                     'keys'       : [json.loads(request['keypair'])['user_public_key']],
 268                     'first_name' : request['first_name'],
 269                     'last_name'  : request['last_name'],
 270                     'email'      : request['email'],
 271                     #'slices'    : None,
 272                     #'researcher': None,
 273                     'pi'         : [auth_pi],
 274                     'enabled'    : True
 275                 }
 276                 # ignored in request: id, timestamp, password
 277
 278                 # ADD USER TO SFA Registry
 279                 sfa_add_user(wsgi_request, sfa_user_params)
 280
 281                 # USER INFO
 282                 user_query  = Query().get('local:user').select('user_id','config','email','status').filter_by('email', '==', request['email'])
 283                 user_details = execute_admin_query(request, user_query)
 284                 #print user_details[0]
 285
 286                 # UPDATE USER STATUS = 2
 287                 manifold_user_params = {
 288                     'status': 2
 289                 }
 290                 manifold_update_user(request, request['email'], manifold_user_params)
 291
 292                 # USER MAIN ACCOUNT != reference
 293                 #print 'USER MAIN ACCOUNT != reference'
 294                 list_accounts_query  = Query().get('local:account').select('user_id','platform_id','auth_type','config')\
 295                     .filter_by('user_id','==',user_details[0]['user_id'])\
 296                     .filter_by('auth_type','!=','reference')
 297                 list_accounts = execute_admin_query(request, list_accounts_query)
 298                 #print "List accounts = ",list_accounts
 299                 for account in list_accounts:
 300                     main_platform_query  = Query().get('local:platform').select('platform_id','platform').filter_by('platform_id','==',account['platform_id'])
 301                     main_platform = execute_admin_query(request, main_platform_query)
 302
 303                 # ADD REFERENCE ACCOUNTS ON SFA ENABLED PLATFORMS
 304                 #print 'ADD REFERENCE ACCOUNTS ON SFA ENABLED PLATFORMS'
 305                 platforms_query  = Query().get('local:platform').filter_by('disabled', '==', '0').filter_by('gateway_type','==','sfa').select('platform_id','gateway_type')
 306                 platforms = execute_admin_query(request, platforms_query)
 307                 #print "platforms SFA ENABLED = ",platforms
 308                 for platform in platforms:
 309                     #print "add reference to platform ",platform
 310                     manifold_account_params = {
 311                         'user_id': user_details[0]['user_id'],
 312                         'platform_id': platform['platform_id'],
 313                         'auth_type': 'reference',
 314                         'config': '{"reference_platform": "' + main_platform[0]['platform'] + '"}',
 315                     }
 316                     manifold_add_account(request, manifold_account_params)
 317
 318                 request_status['SFA user'] = {'status': True }
 319
 320             except Exception, e:
 321                  request_status['SFA user'] = {'status': False, 'description': str(e)}
 322
 323 #            user_params = {'status':2}
 324 #            manifold_update_user(request, request['email'], user_params)
 325
 326             # MANIFOLD user should be added beforehand, during registration
 327             #try:
 328             #    manifold_user_params = { key: request[key] for key in MANIFOLD_USER_KEYS }
 329             #    # XXX # manifold_add_user(manifold_user_params)
 330             #    request_status['MySlice user'] = {'status': True }
 331             #except Exception, e:
 332             #    request_status['MySlice user'] = {'status': False, 'description': str(e)}
 333
 334             # XXX
 335             #manifold_account_params = { key: request[key] for key in MANIFOLD_ACCOUNT_KEYS }
 336             #manifold_add_account(manifold_account_params)
 337             #request_status['MySlice testbed accounts'] = {'status': False }
 338
 339         elif request['type'] == 'slice':
 340             try:
 341                 hrn = "%s.%s" % (request['authority_hrn'], request['slice_name'])
 342                 # XXX tmp sfa dependency
 343                 from sfa.util.xrn import Xrn
 344                 urn = Xrn(hrn, request['type']).get_urn()
 345
 346                 # Add User to Slice if we have the user_hrn in pendingslice table
 347                 if 'user_hrn' in request:
 348                     user_hrn = request['user_hrn']
 349                     print "Slice %s will be created for %s" % (hrn,request['user_hrn'])
 350                 else:
 351                     user_hrn=''
 352                     print "Slice %s will be created without users %s" % (hrn)
 353                 sfa_slice_params = {
 354                     'hrn'        : hrn,
 355                     'urn'        : urn,
 356                     'type'       : request['type'],
 357                     #'slices'    : None,
 358                     'researcher' : [user_hrn],
 359                     #'pi'        : None,
 360                     'enabled'    : True
 361                 }
 362                 # ignored in request: id, timestamp,  number_of_nodes, type_of_nodes, purpose
 363
 364                 sfa_add_slice(wsgi_request, sfa_slice_params)
 365                 #sfa_add_user_to_slice(wsgi_request, user_hrn, sfa_slice_params)
 366                 request_status['SFA slice'] = {'status': True }
 367
 368             except Exception, e:
 369                 request_status['SFA slice'] = {'status': False, 'description': str(e)}
 370
 371         elif request['type'] == 'authority':
 372             try:
 373                 #hrn = "%s.%s" % (request['authority_hrn'], request['site_authority'])
 374                 hrn = request['site_authority']
 375                 # XXX tmp sfa dependency
 376                 from sfa.util.xrn import Xrn
 377                 urn = Xrn(hrn, request['type']).get_urn()
 378
 379                 sfa_authority_params = {
 380                     'hrn'        : hrn,
 381                     'urn'        : urn,
 382                     'type'       : request['type'],
 383                     #'pi'        : None,
 384                     'enabled'    : True
 385                 }
 386                 print "ADD Authority"
 387                 sfa_add_authority(wsgi_request, sfa_authority_params)
 388                 request_status['SFA authority'] = {'status': True }
 389
 390             except Exception, e:
 391                 request_status['SFA authority'] = {'status': False, 'description': str(e)}
 392
 393         # XXX Remove from Pendings in database
 394
 395         status['%s__%s' % (request['type'], request['id'])] = request_status
 396
 397     return status
 398
 399
 400 def validate_action(request, **kwargs):
 401     ids = filter(None, kwargs['id'].split('/'))
 402     status = portal_validate_request(request, ids)
 403     json_answer = json.dumps(status)
 404     return HttpResponse (json_answer, mimetype="application/json")
 405
 406 # Django and ajax
 407 # http://djangosnippets.org/snippets/942/