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
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
13 # Get the list of authorities
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
21 # raise Exception, "Authority not found: %s" % authority_hrn
23 #return result['pi_users']
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']
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]
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
45 # SFA add record (user, slice)
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)
53 raise Exception, "Could not create %s. Already exists ?" % user_params['hrn']
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)
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)
68 raise Exception, "Could not create %s. Already exists ?" % slice_params['hrn']
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
76 raise Exception, "Could not create %s. Already exists ?" % authority_params['hrn']
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'
89 raise Exception, "Could not create %s. Already exists ?" % slice_params['hrn']
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)
99 raise Exception, "Failed creating manifold user: %s" % user_params['email']
101 return result['email']
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
110 # raise Exception, "Failed updating manifold user: %s" % user_params['email']
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)
118 raise Exception, "Failed creating manifold account on platform %s for user: %s" % (account_params['platform'], account_params['user'])
120 return result['user_id']
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)
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)
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)
140 raise Exception, "Failed creating manifold platform %s for user: %s" % (platform_params['platform'], platform_params['user'])
142 return result['platform_id']
145 def make_request_user(user):
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
158 def make_request_slice(slice):
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
171 def make_request_authority(authority):
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
192 def make_requests(pending_users, pending_slices, pending_authorities):
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))
202 def get_request_by_id(ids):
203 sorted_ids = { 'user': [], 'slice': [], 'authority': [] }
205 type, id = type__id.split('__')
206 sorted_ids[type].append(id)
209 pending_users = PendingUser.objects.all()
210 pending_slices = PendingSlice.objects.all()
211 pending_authorities = PendingAuthority.objects.all()
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()
217 return make_requests(pending_users, pending_slices, pending_authorities)
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()
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()
230 return make_requests(pending_users, pending_slices, pending_authorities)
232 # XXX Is it in sync with the form fields ?
234 def portal_validate_request(wsgi_request, request_ids):
237 if not isinstance(request_ids, list):
238 request_ids = [request_ids]
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
249 print "REQUEST", request
250 if request['type'] == 'user':
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()
260 auth_pi = request['pi']
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'],
276 # ignored in request: id, timestamp, password
278 # ADD USER TO SFA Registry
279 sfa_add_user(wsgi_request, sfa_user_params)
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]
286 # UPDATE USER STATUS = 2
287 manifold_user_params = {
290 manifold_update_user(request, request['email'], manifold_user_params)
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)
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'] + '"}',
316 manifold_add_account(request, manifold_account_params)
318 request_status['SFA user'] = {'status': True }
321 request_status['SFA user'] = {'status': False, 'description': str(e)}
323 # user_params = {'status':2}
324 # manifold_update_user(request, request['email'], user_params)
326 # MANIFOLD user should be added beforehand, during registration
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)}
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 }
339 elif request['type'] == 'slice':
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()
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'])
352 print "Slice %s will be created without users %s" % (hrn)
356 'type' : request['type'],
358 'researcher' : [user_hrn],
362 # ignored in request: id, timestamp, number_of_nodes, type_of_nodes, purpose
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 }
369 request_status['SFA slice'] = {'status': False, 'description': str(e)}
371 elif request['type'] == 'authority':
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()
379 sfa_authority_params = {
382 'type' : request['type'],
386 print "ADD Authority"
387 sfa_add_authority(wsgi_request, sfa_authority_params)
388 request_status['SFA authority'] = {'status': True }
391 request_status['SFA authority'] = {'status': False, 'description': str(e)}
393 # XXX Remove from Pendings in database
395 status['%s__%s' % (request['type'], request['id'])] = request_status
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")
407 # http://djangosnippets.org/snippets/942/