portal/actions.py

   1 from django.http                 import HttpResponse
   2 from manifold.core.query         import Query
   3 from manifold.manifoldapi        import execute_query,execute_admin_query
   4 from portal.models               import PendingUser, PendingSlice
   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_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
  29     pi_users = authority_get_pis(request,authority_hrn)
  30     pi_user_hrns = [ hrn for x in pi_users for hrn in x['pi_users'] ]
  31     query = Query.get('user').filter_by('user_hrn', 'included', pi_user_hrns).select('email')
  32     results = execute_query(request, query)
  33     print "mails",  [result['email'] for result in results]
  34     return [result['email'] for result in results]
  35
  36 # SFA add record (user, slice)
  37
  38 def sfa_add_user(request, user_params):
  39     query = Query.create('user').set(user_params).select('user_hrn')
  40     results = execute_query(request, query)
  41     if not results:
  42         raise Exception, "Could not create %s. Already exists ?" % user_params['hrn']
  43     return results
  44
  45 def sfa_update_user(request, user_hrn, user_params):
  46     # user_params: keys [public_key]
  47     query = Query.update('user').filter_by('user_hrn', '==', user_hrn).set(user_params).select('user_hrn')
  48     results = execute_query(request,query)
  49     return results
  50
  51 def sfa_add_slice(request, slice_params):
  52     query = Query.create('slice').set(slice_params).select('slice_hrn')
  53     results = execute_query(request, query)
  54     if not results:
  55         raise Exception, "Could not create %s. Already exists ?" % slice_params['hrn']
  56     return results
  57
  58 def sfa_add_user_to_slice(request, user_hrn, slice_params):
  59     query = Query.update('slice').filter_by('user_hrn', '==', user_hrn).set(slice_params).select('slice_hrn')
  60     results = execute_query(request, query)
  61     if not results:
  62         raise Exception, "Could not create %s. Already exists ?" % slice_params['hrn']
  63     return results
  64
  65 # Propose hrn
  66
  67 def manifold_add_user(request, user_params):
  68     # user_params: email, password e.g., user_params = {'email':'aa@aa.com','password':'demo'}
  69     query = Query.create('local:user').set(user_params).select('email')
  70     results = execute_admin_query(request, query)
  71     if not results:
  72         raise Exception, "Failed creating manifold user: %s" % user_params['email']
  73     result, = results
  74     return result['email']
  75
  76 def manifold_update_user(request, email, user_params):
  77     # user_params: password, config e.g.,
  78     query = Query.update('local:user').filter_by('email', '==', email).set(user_params).select('email')
  79     results = execute_admin_query(request,query)
  80     # NOTE: results remains empty and goes to Exception. However, it updates the manifold DB.
  81     # That's why I commented the exception part. -- Yasin
  82     #if not results:
  83     #    raise Exception, "Failed updating manifold user: %s" % user_params['email']
  84     #result, = results
  85     return results
  86
  87 def manifold_add_account(request, account_params):
  88     query = Query.create('local:account').set(account_params).select(['user', 'platform'])
  89     results = execute_admin_query(request,query)
  90     if not results:
  91         raise Exception, "Failed creating manifold account on platform %s for user: %s" % (account_params['platform'], account_params['user'])
  92     result, = results
  93     return result['user_id']
  94
  95 def manifold_update_account(request,account_params):
  96     # account_params: config
  97     query = Query.update('local:account').filter_by('platform', '==', 'myslice').set(account_params).select('user_id')
  98     results = execute_query(request,query)
  99     # NOTE: results remains empty and goes to Exception. However, it updates the manifold DB.
 100     # That's why I commented the exception part. -- Yasin
 101     #if not results:
 102     #    raise Exception, "Failed updating manifold account: config %s" % account_params['config']
 103     #result, = results
 104     return results
 105
 106 #explicitly mention the platform_id
 107 def manifold_delete_account(request, platform_id, account_params):
 108     query = Query.delete('local:account').filter_by('platform_id', '==', platform_id).set(account_params).select('user_id')
 109     results = execute_query(request,query)
 110     return results
 111
 112
 113 #not tested
 114 def manifold_add_platform(request, platform_params):
 115     query = Query.create('local:platform').set(platform_params).select(['user', 'platform'])
 116     results = execute_admin_query(request,query)
 117     if not results:
 118         raise Exception, "Failed creating manifold  platform %s for user: %s" % (platform_params['platform'], platform_params['user'])
 119     result, = results
 120     return result['platform_id']
 121
 122
 123 def make_request_user(user):
 124     request = {}
 125     request['type']          = 'user'
 126     request['id']            = user.id
 127     request['timestamp']     = user.created # XXX in DB ?
 128     request['authority_hrn'] = user.authority_hrn
 129     request['first_name']    = user.first_name
 130     request['last_name']     = user.last_name
 131     request['email']         = user.email
 132     request['login']         = user.login
 133     request['keypair']       = user.keypair
 134     return request
 135
 136 def make_request_slice(slice):
 137     request = {}
 138     request['type'] = 'slice'
 139     request['id'] = slice.id
 140     request['user_email'] = slice.user_email
 141     request['timestamp'] = slice.created
 142     request['authority_hrn'] = slice.authority_hrn
 143     request['slice_name'] = slice.slice_name
 144     request['number_of_nodes'] = slice.number_of_nodes
 145     request['type_of_nodes'] = slice.type_of_nodes
 146     request['purpose'] = slice.purpose
 147     return request
 148
 149 def make_requests(pending_users, pending_slices):
 150     print "pending users =", pending_users
 151     print "pending slices =", pending_slices
 152
 153     requests = []
 154     for user in pending_users:
 155         requests.append(make_request_user(user))
 156     for slice in pending_slices:
 157         requests.append(make_request_slice(slice))
 158     return requests
 159
 160 def get_request_by_id(ids):
 161     sorted_ids = { 'user': [], 'slice': [] }
 162     for type__id in ids:
 163         type, id = type__id.split('__')
 164         sorted_ids[type].append(id)
 165
 166     if not ids:
 167         pending_users  = PendingUser.objects.all()
 168         pending_slices = PendingSlice.objects.all()
 169     else:
 170         pending_users  = PendingUser.objects.filter(id__in=sorted_ids['user']).all()
 171         pending_slices = PendingSlice.objects.filter(id__in=sorted_ids['slice']).all()
 172
 173     return make_requests(pending_users, pending_slices)
 174
 175 def get_request_by_authority(authority_hrns):
 176     print "get_request_by_authority auth_hrns = ", authority_hrns
 177     if not authority_hrns:
 178         pending_users  = PendingUser.objects.all()
 179         pending_slices = PendingSlice.objects.all()
 180     else:
 181         pending_users  = PendingUser.objects.filter(authority_hrn__in=authority_hrns).all()
 182         pending_slices = PendingSlice.objects.filter(authority_hrn__in=authority_hrns).all()
 183
 184     return make_requests(pending_users, pending_slices)
 185
 186 # XXX Is it in sync with the form fields ?
 187
 188 def portal_validate_request(wsgi_request, request_ids):
 189     status = {}
 190
 191     if not isinstance(request_ids, list):
 192         request_ids = [request_ids]
 193
 194     requests = get_request_by_id(request_ids)
 195     for request in requests:
 196         # type, id, timestamp, details, allowed -- MISSING: authority_hrn
 197         # CAREFUL about details
 198         # user  : first name, last name, email, password, keypair
 199         # slice : number of nodes, type of nodes, purpose
 200
 201         request_status = {}
 202
 203         print "REQUEST", request
 204         if request['type'] == 'user':
 205
 206             try:
 207                 # XXX tmp user_hrn inside the keypair column of pendiguser table
 208                 hrn = json.loads(request['keypair'])['user_hrn']
 209                 #hrn = "%s.%s" % (request['authority_hrn'], request['login'])
 210                 # XXX tmp sfa dependency
 211                 from sfa.util.xrn import Xrn
 212                 urn = Xrn(hrn, request['type']).get_urn()
 213
 214                 sfa_user_params = {
 215                     'hrn'        : hrn,
 216                     'urn'        : urn,
 217                     'type'       : request['type'],
 218                     'keys'       : [json.loads(request['keypair'])['user_public_key']],
 219                     'first_name' : request['first_name'],
 220                     'last_name'  : request['last_name'],
 221                     'email'      : request['email'],
 222                     #'slices'    : None,
 223                     #'researcher': None,
 224                     #'pi'        : None,
 225                     'enabled'    : True
 226                 }
 227                 # ignored in request: id, timestamp, password
 228
 229                 sfa_add_user(wsgi_request, sfa_user_params)
 230
 231                 # XXX Remove from database
 232
 233
 234                 request_status['SFA user'] = {'status': True }
 235
 236             except Exception, e:
 237                 request_status['SFA user'] = {'status': False, 'description': str(e)}
 238
 239             # MANIFOLD user should be added beforehand, during registration
 240             #try:
 241             #    manifold_user_params = { key: request[key] for key in MANIFOLD_USER_KEYS }
 242             #    # XXX # manifold_add_user(manifold_user_params)
 243             #    request_status['MySlice user'] = {'status': True }
 244             #except Exception, e:
 245             #    request_status['MySlice user'] = {'status': False, 'description': str(e)}
 246
 247             # XXX
 248             #manifold_account_params = { key: request[key] for key in MANIFOLD_ACCOUNT_KEYS }
 249             #manifold_add_account(manifold_account_params)
 250             #request_status['MySlice testbed accounts'] = {'status': False }
 251
 252         elif request['type'] == 'slice':
 253             try:
 254                 hrn = "%s.%s" % (request['authority_hrn'], request['slice_name'])
 255                 # XXX tmp sfa dependency
 256                 from sfa.util.xrn import Xrn
 257                 urn = Xrn(hrn, request['type']).get_urn()
 258
 259                 sfa_slice_params = {
 260                     'hrn'        : hrn,
 261                     'urn'        : urn,
 262                     'type'       : request['type'],
 263                     #'slices'    : None,
 264                     #'researcher': None,
 265                     #'pi'        : None,
 266                     'enabled'    : True
 267                 }
 268                 # ignored in request: id, timestamp,  number_of_nodes, type_of_nodes, purpose
 269
 270                 sfa_add_slice(wsgi_request, sfa_slice_params)
 271                 #sfa_add_user_to_slice(wsgi_request, user_hrn, sfa_slice_params)
 272
 273                 # XXX Remove from database
 274
 275
 276                 request_status['SFA slice'] = {'status': True }
 277
 278             except Exception, e:
 279                 request_status['SFA slice'] = {'status': False, 'description': str(e)}
 280
 281         status['%s__%s' % (request['type'], request['id'])] = request_status
 282
 283     return status
 284
 285
 286 def validate_action(request, **kwargs):
 287     ids = filter(None, kwargs['id'].split('/'))
 288     status = portal_validate_request(request, ids)
 289     json_answer = json.dumps(status)
 290     return HttpResponse (json_answer, mimetype="application/json")
 291
 292 # Django and ajax
 293 # http://djangosnippets.org/snippets/942/