portal/actions.py

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