Various improvements
[myslice.git] / 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 from django.contrib.auth.models  import User
8 from django.contrib.sites.models import Site
9 from django.contrib.auth        import get_user_model
10 from django.template.loader     import render_to_string
11 from django.core.mail           import EmailMultiAlternatives, send_mail
12
13 from myslice.theme              import ThemeView
14 from myslice.configengine       import ConfigEngine
15
16
17 theme = ThemeView()
18
19 import activity.slice
20
21 # Thierry: moving this right into the code so 
22 # most people can use myslice without having to install sfa
23 # XXX tmp sfa dependency, should be moved to SFA gateway
24 #from sfa.util.xrn                import Xrn 
25
26
27 # Get the list of pis in a given authority
28 def authority_get_pis(request, authority_hrn):
29
30     # REGISTRY ONLY TO BE REMOVED WITH MANIFOLD-V2
31     query = Query.get('authority').filter_by('authority_hrn', '==', authority_hrn).select('pi_users')
32     results = execute_admin_query(request, query)
33     print "authority_get_pis = %s" % results
34     # NOTE: temporarily commented. Because results is giving empty list. 
35     # Needs more debugging
36     #if not results:
37     #    raise Exception, "Authority not found: %s" % authority_hrn
38     #result, = results
39     #return result['pi_users']
40     return results
41
42 #check the user is pi or not in the registry
43 def authority_check_pis(request, user_email):
44     try:
45         user_query  = Query().get('local:user').filter_by('email', '==', user_email).select('user_id','email','password','config')
46         user_details = execute_admin_query(request, user_query)
47     
48         # getting the authority_hrn
49         for user_detail in user_details:
50             user_id = user_detail['user_id']
51             if user_detail['config']:
52                 config = json.loads(user_detail['config'])
53                 authority_hrn = config.get('authority','Unknown Authority')
54  
55         account_query  = Query().get('local:account').filter_by('user_id', '==', user_id).select('user_id','platform_id','auth_type','config')
56         account_details = execute_admin_query(request, account_query)
57     
58         platform_query  = Query().get('local:platform').select('platform_id','platform')
59         platform_details = execute_admin_query(request, platform_query)
60     
61         for account_detail in account_details:
62             for platform_detail in platform_details:
63                 if platform_detail['platform_id'] == account_detail['platform_id']:
64                     if 'myslice' in platform_detail['platform']:
65                         account_config = json.loads(account_detail['config'])
66                         user_hrn = account_config.get('user_hrn','N/A')
67
68         pi_status = False
69         pis = authority_get_pis (request, authority_hrn)
70         for pi in pis:
71             pi_list = pi['pi_users']
72
73         if user_hrn in pi_list:
74             pi_status = True
75         return pi_status
76
77     except Exception,e:
78         print "Exception in actions.py in authority_check_pis %s" % e
79         return None
80
81
82 def authority_add_pis(request, authority_hrn,user_hrn):
83     try:
84         # getting pis of the authority of the user
85         pis = authority_get_pis (request, authority_hrn)
86         for pi in pis:
87             pi_list = pi['pi_users']
88    
89         updated_pi_list = pi_list.append(user_hrn) 
90         query = Query.update('authority').filter_by('authority_hrn', '==', authority_hrn).set({'pi_users':pi_list})
91         results = execute_admin_query(request,query)
92         newpis = authority_get_pis (request, authority_hrn)
93         return newpis
94     except Exception,e: 
95         print "Exception in actions.py in authority_add_pis %s" % e
96         return None
97
98
99 def authority_remove_pis(request, authority_hrn,user_hrn):
100     try:
101         # getting pis of the authority of the user
102         pis = authority_get_pis (request, authority_hrn)
103         for pi in pis:
104             pi_list = pi['pi_users']
105  
106         updated_pi_list = pi_list.remove(user_hrn) 
107         query = Query.update('authority').filter_by('authority_hrn', '==', authority_hrn).set({'pi_users':pi_list})
108         results = execute_admin_query(request,query)
109         newpis = authority_get_pis (request, authority_hrn)
110         return newpis
111     except Exception,e: 
112         print "Exception in actions.py in authority_remove_pis %s" % e
113         return None
114
115
116
117 def authority_get_pi_emails(request, authority_hrn):
118     pi_users = authority_get_pis(request,authority_hrn)
119     print "pi_users = %s" % pi_users
120
121     if any(pi['pi_users'] == None or not pi['pi_users']  for pi in pi_users):
122         #theme.template_name = 'email_default_recipients.txt' 
123         #default_email = render_to_string(theme.template, request)
124         #default_email = default_email.replace('\n', '')
125         #return default_email
126         # the above doesn't work
127         return ['support@onelab.eu']
128     else:
129         pi_user_hrns = [ hrn for x in pi_users for hrn in x['pi_users'] ]
130
131         # REGISTRY ONLY TO BE REMOVED WITH MANIFOLD-V2
132         query = Query.get('myslice:user').filter_by('user_hrn', 'included', pi_user_hrns).select('user_email')
133         results = execute_admin_query(request, query)
134         return [result['user_email'] for result in results]
135
136 #clear user credentials
137 def clear_user_creds(request, user_email):
138     try:
139         user_query  = Query().get('local:user').filter_by('email', '==', user_email).select('user_id','email','password','config')
140         user_details = execute_admin_query(request, user_query)
141     
142         # getting the user_id from the session
143         for user_detail in user_details:
144             user_id = user_detail['user_id']
145             user_email = user_detail['email']
146     
147         account_query  = Query().get('local:account').filter_by('user_id', '==', user_id).select('user_id','platform_id','auth_type','config')
148         account_details = execute_admin_query(request, account_query)
149     
150         platform_query  = Query().get('local:platform').select('platform_id','platform')
151         platform_details = execute_admin_query(request, platform_query)
152     
153         for account_detail in account_details:
154             for platform_detail in platform_details:
155                 if platform_detail['platform_id'] == account_detail['platform_id']:
156                     if 'myslice' in platform_detail['platform']:
157                         account_config = json.loads(account_detail['config'])
158                         #user_cred = account_config.get('delegated_user_credential','N/A')
159                         user_cred = account_config.get('user_credential','N/A')
160                         if 'N/A' not in user_cred:
161                             user_hrn = account_config.get('user_hrn','N/A')
162                             user_pub_key = json.dumps(account_config.get('user_public_key','N/A'))
163                             user_priv_key = json.dumps(account_config.get('user_private_key','N/A'))
164                             updated_config = '{"user_public_key":'+ user_pub_key + ', "user_private_key":'+ user_priv_key + ', "user_hrn":"'+ user_hrn + '"}'
165                             user_params = { 'config': updated_config}
166                             manifold_update_account(request, user_id,user_params)
167                             return user_email
168                         else:
169                             return None
170
171     except Exception,e:
172         print "Exception in actions.py in clear_user_creds %s" % e
173         return None
174
175 def is_pi(wsgi_request, user_hrn, authority_hrn):
176     # XXX could be done in a single query !
177
178     # seauthorities from user where user_hrn == "ple.upmc.jordan_auge"
179
180     # REGISTRY ONLY TO BE REMOVED WITH MANIFOLD-V2
181     query = Query.get('myslice:user').filter_by('user_hrn', '==', user_hrn).select('pi_authorities')
182     results = execute_admin_query(wsgi_request, query)
183     if not results:
184         # XXX Warning ?
185         return False
186     result = results[0]
187     user_authority_hrns = result.get('pi_authorities', [])
188     return authority_hrn in user_authority_hrns
189     
190 # SFA get record
191
192 def sfa_get_user(request, user_hrn, pub):
193
194     # REGISTRY ONLY TO BE REMOVED WITH MANIFOLD-V2
195     query_sfa_user = Query.get('myslice:user').filter_by('user_hrn', '==', user_hrn)
196     result_sfa_user = execute_admin_query(request, query_sfa_user)
197     return result_sfa_user                        
198
199 def sfa_update_user(request, user_hrn, user_params):
200     # user_params: keys [public_key] 
201     if 'email' in user_params:
202         user_params['user_email'] = user_params['email']
203
204     # REGISTRY ONLY TO BE REMOVED WITH MANIFOLD-V2
205     query = Query.update('myslice:user').filter_by('user_hrn', '==', user_hrn).set(user_params).select('user_hrn')
206     results = execute_admin_query(request,query)
207     return results
208
209 def sfa_add_authority(request, authority_params):
210
211     # REGISTRY ONLY TO BE REMOVED WITH MANIFOLD-V2
212     query = Query.create('authority').set(authority_params).select('authority_hrn')
213     results = execute_query(request, query)
214     print "sfa_add_auth results=",results
215     if not results:
216         raise Exception, "Could not create %s. Already exists ?" % authority_params['hrn']
217     return results
218
219 def sfa_add_user_to_slice(request, user_hrn, slice_params):
220 # 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'
221
222     # REGISTRY ONLY TO BE REMOVED WITH MANIFOLD-V2
223     query_current_users = Query.get('myslice:slice').select('user').filter_by('slice_hrn','==',slice_params['hrn'])
224     results_current_users = execute_query(request, query_current_users)
225     slice_params['researcher'] = slice_params['researcher'] | results_current_users
226
227     # REGISTRY ONLY TO BE REMOVED WITH MANIFOLD-V2
228     query = Query.update('myslice:slice').filter_by('user_hrn', '==', user_hrn).set(slice_params).select('slice_hrn')
229     results = execute_query(request, query)
230 # Also possible but not supported yet
231 # UPDATE myslice:user SET slice=['ple.upmc.agent','ple.upmc.myslicedemo','ple.upmc.tophat'] where user_hrn=='ple.upmc.azerty'
232     if not results:
233         raise Exception, "Could not create %s. Already exists ?" % slice_params['hrn']
234     return results
235
236 # Propose hrn
237
238 def manifold_add_user(wsgi_request, request):
239     """Add a Manifold user corresponding to a user request.
240
241     Args:
242         wsgi_request: a WSGIRequest instance
243         request (dict): a dictionary containing the user request built from the
244             form.
245
246     Returns:
247         The user_id of the inserted user.
248
249     Raises:
250         ?
251     
252     """
253     USER_CONFIG = '{"firstname": "%(first_name)s", "lastname": "%(last_name)s", "authority": "%(authority_hrn)s"}'
254
255     user_params = {
256         'email'     : request['email'],
257         'password'  : request['password'],
258         'config'    : USER_CONFIG % request,
259         'status'    : 1,
260     }
261
262     query = Query.create('local:user').set(user_params).select('email')
263     results = execute_admin_query(request, query)
264     if not results:
265         raise Exception, "Failed creating manifold user: %s" % user_params['email']
266     result = results[0]
267     return result['email']
268
269 def manifold_update_user(request, email, user_params):
270     # user_params: password, config e.g., 
271     query = Query.update('local:user').filter_by('email', '==', email).set(user_params).select('email')
272     results = execute_admin_query(request,query)
273     # NOTE: results remains empty and goes to Exception. However, it updates the manifold DB.
274     # That's why I commented the exception part. -- Yasin 
275     #if not results:
276     #    raise Exception, "Failed updating manifold user: %s" % user_params['email']
277     #result, = results
278     return results
279
280 def manifold_add_account(request, account_params):
281     query = Query.create('local:account').set(account_params).select(['user', 'platform'])
282     results = execute_admin_query(request,query)
283     if not results:
284         raise Exception, "Failed creating manifold account on platform %s for user: %s" % (account_params['platform'], account_params['user'])
285     result, = results
286     return result['user_id']
287
288 def manifold_update_account(request,user_id,account_params):
289     # account_params: config
290     query = Query.update('local:account').filter_by('platform', '==', 'myslice').filter_by('user_id', '==', user_id).set(account_params).select('user_id')
291     results = execute_admin_query(request,query)
292     return results
293
294 #explicitly mention the platform_id
295 def manifold_delete_account(request, platform_id, user_id, account_params):
296     query = Query.delete('local:account').filter_by('platform_id', '==', platform_id).filter_by('user_id', '==', user_id).set(account_params).select('user_id')
297     results = execute_admin_query(request,query)
298     return results
299
300 def manifold_delete_user(request, user_id, user_params):
301     query = Query.delete('local:user').filter_by('user_id', '==', user_id).set(user_params).select('user_id')
302     results = execute_admin_query(request,query)
303     return results
304
305
306 #not tested
307 def manifold_add_platform(request, platform_params):
308     query = Query.create('local:platform').set(platform_params).select(['user', 'platform'])
309     results = execute_admin_query(request,query)
310     if not results:
311         raise Exception, "Failed creating manifold platform %s for user: %s" % (platform_params['platform'], platform_params['user'])
312     result, = results
313     return result['platform_id']
314
315
316 def make_request_user(user):
317     request = {}
318     request['type']          = 'user'
319     request['id']            = user.id
320     request['timestamp']     = user.created # XXX in DB ?
321     request['authority_hrn'] = user.authority_hrn
322     request['first_name']    = user.first_name
323     request['last_name']     = user.last_name
324     request['email']         = user.email
325     request['login']         = user.login
326     request['user_hrn']      = user.user_hrn
327     request['public_key']    = user.public_key
328     request['private_key']   = user.private_key
329     return request
330
331 def make_request_slice(slice):
332     request = {}
333     request['type'] = 'slice'
334     request['id'] = slice.id
335     request['user_hrn'] = slice.user_hrn
336     request['timestamp'] = slice.created
337     request['authority_hrn'] = slice.authority_hrn
338     request['slice_name'] = slice.slice_name
339     request['number_of_nodes'] = slice.number_of_nodes
340     request['type_of_nodes'] = slice.type_of_nodes
341     request['purpose'] = slice.purpose
342     return request
343
344 def make_request_authority(authority):
345     request = {}
346     request['type']                  = 'authority'
347     request['id']                    = authority.id
348     request['site_name']             = authority.site_name
349     request['site_latitude']         = authority.site_latitude
350     request['site_longitude']        = authority.site_longitude
351     request['site_url']              = authority.site_url
352     request['site_authority']        = authority.site_authority
353     request['site_abbreviated_name'] = authority.site_abbreviated_name
354     request['address_line1']         = authority.address_line1
355     request['address_line2']         = authority.address_line2
356     request['address_line3']         = authority.address_line3
357     request['address_city']          = authority.address_city
358     request['address_postalcode']    = authority.address_postalcode
359     request['address_state']         = authority.address_state
360     request['address_country']       = authority.address_country
361     request['authority_hrn']         = authority.authority_hrn
362     request['timestamp']             = authority.created
363     return request
364
365 def make_requests(pending_users, pending_slices, pending_authorities):
366     requests = []
367     for user in pending_users:
368         requests.append(make_request_user(user))
369     for slice in pending_slices:
370         requests.append(make_request_slice(slice))
371     for authority in pending_authorities:
372         requests.append(make_request_authority(authority))
373     return requests   
374
375 def get_request_by_id(ids):
376     sorted_ids = { 'user': [], 'slice': [], 'authority': [] }
377     for type__id in ids:
378         type, id = type__id.split('__')
379         sorted_ids[type].append(id)
380         
381     if not ids:
382         pending_users  = PendingUser.objects.all()
383         pending_slices = PendingSlice.objects.all()
384         pending_authorities = PendingAuthority.objects.all()
385     else:
386         pending_users  = PendingUser.objects.filter(id__in=sorted_ids['user']).all()
387         pending_slices = PendingSlice.objects.filter(id__in=sorted_ids['slice']).all()
388         pending_authorities = PendingAuthority.objects.filter(id__in=sorted_ids['authority']).all()
389
390     return make_requests(pending_users, pending_slices, pending_authorities)
391
392 def get_requests(authority_hrns=None):
393     print "get_request_by_authority auth_hrns = ", authority_hrns
394     if not authority_hrns:
395         ## get those pending users who have confirmed their emails
396         pending_users  = PendingUser.objects.filter(status__iexact = 'True')
397         pending_slices = PendingSlice.objects.all()
398         pending_authorities = PendingAuthority.objects.all()
399     else:
400         pending_users  = PendingUser.objects
401         pending_slices = PendingSlice.objects
402         pending_authorities = PendingAuthority.objects
403         from django.db.models import Q
404         list_user_Q = list()
405         list_slice_Q = list()
406         list_auth_Q = list()
407         for hrn in authority_hrns:
408             list_user_Q.append(Q(authority_hrn__startswith=hrn, status__iexact = 'True'))
409             list_slice_Q.append(Q(authority_hrn__startswith=hrn))
410             list_auth_Q.append(Q(site_authority__startswith=hrn))
411             print "startswith hrn = ",hrn
412         from operator import __or__ as OR
413         pending_users        = pending_users.filter(reduce(OR, list_user_Q))
414         pending_slices       = pending_slices.filter(reduce(OR, list_slice_Q))
415         pending_authorities  = pending_authorities.filter(reduce(OR, list_auth_Q))
416         #pending_authorities  = pending_authorities.all() #filter(reduce(OR, list_Q))
417
418     return make_requests(pending_users, pending_slices, pending_authorities)
419
420 # XXX Is it in sync with the form fields ?
421
422 def portal_validate_request(wsgi_request, request_ids):
423     status = {}
424
425     if not isinstance(request_ids, list):
426         request_ids = [request_ids]
427
428     requests = get_request_by_id(request_ids)
429     for request in requests:
430         # type, id, timestamp, details, allowed -- MISSING: authority_hrn
431         # CAREFUL about details
432         # user  : first name, last name, email, password, keypair
433         # slice : number of nodes, type of nodes, purpose
434         
435         request_status = {}
436
437         if request['type'] == 'user':
438
439             try:
440                 create_user(wsgi_request, request)
441                 request_status['SFA user'] = {'status': True }
442                 PendingUser.objects.get(id=request['id']).delete()
443             except Exception, e:
444                  request_status['SFA user'] = {'status': False, 'description': str(e)}
445                        
446 #            user_params = {'status':2}
447 #            manifold_update_user(request, request['email'], user_params)
448
449             # MANIFOLD user should be added beforehand, during registration
450             #try:
451             #    manifold_user_params = { key: request[key] for key in MANIFOLD_USER_KEYS }
452             #    # XXX # manifold_add_user(manifold_user_params)
453             #    request_status['MySlice user'] = {'status': True }
454             #except Exception, e:
455             #    request_status['MySlice user'] = {'status': False, 'description': str(e)}
456
457             # XXX
458             #manifold_account_params = { key: request[key] for key in MANIFOLD_ACCOUNT_KEYS }
459             #manifold_add_account(manifold_account_params)
460             #request_status['MySlice testbed accounts'] = {'status': False }
461
462         elif request['type'] == 'slice':
463             try:
464                 create_slice(wsgi_request, request)
465                 request_status['SFA slice'] = {'status': True }
466                 PendingSlice.objects.get(id=request['id']).delete()
467
468             except Exception, e:
469                 request_status['SFA slice'] = {'status': False, 'description': str(e)}
470
471         elif request['type'] == 'authority':
472             try:
473                 #hrn = "%s.%s" % (request['authority_hrn'], request['site_authority'])
474                 hrn = request['site_authority']
475                 # XXX tmp sfa dependency
476                 from sfa.util.xrn import Xrn 
477                 urn = Xrn(hrn, request['type']).get_urn()
478                 
479                 # Only hrn is required for Manifold Query 
480                 sfa_authority_params = {
481                     'authority_hrn'        : hrn,
482                     #'authority_urn'        : urn,
483                     #'type'       : request['type'],
484                     #'pi'        : None,
485                     #'enabled'    : True
486                 }
487                 print "ADD Authority"
488                 sfa_add_authority(wsgi_request, sfa_authority_params)
489                 request_status['SFA authority'] = {'status': True }
490                 PendingAuthority.objects.get(id=request['id']).delete()
491
492             except Exception, e:
493                 request_status['SFA authority'] = {'status': False, 'description': str(e)}
494
495         # XXX Remove from Pendings in database
496
497         status['%s__%s' % (request['type'], request['id'])] = request_status
498
499     return status
500
501 def validate_action(request, **kwargs):
502     ids = filter(None, kwargs['id'].split('/'))
503     status = portal_validate_request(request, ids)
504     json_answer = json.dumps(status)
505     return HttpResponse (json_answer, mimetype="application/json")
506
507
508 def reject_action(request, **kwargs):
509     ids = filter(None, kwargs['id'].split('/'))
510     status = portal_reject_request(request, ids)
511     json_answer = json.dumps(status)
512     return HttpResponse (json_answer, mimetype="application/json")
513
514
515 def portal_reject_request(wsgi_request, request_ids):
516     status = {}
517     # get the domain url    
518     current_site = Site.objects.get_current()
519     current_site = current_site.domain
520
521
522     if not isinstance(request_ids, list):
523         request_ids = [request_ids]
524
525     requests = get_request_by_id(request_ids)
526     for request in requests:
527         # type, id, timestamp, details, allowed -- MISSING: authority_hrn
528         # CAREFUL about details
529         # user  : first name, last name, email, password, keypair
530         # slice : number of nodes, type of nodes, purpose
531         
532         request_status = {}
533
534         if request['type'] == 'user':
535             try:
536                 request_status['SFA user'] = {'status': True }
537                 # getting user email based on id 
538                 ## RAW SQL queries on Django DB- https://docs.djangoproject.com/en/dev/topics/db/sql/
539                 for user in PendingUser.objects.raw('SELECT * FROM portal_pendinguser WHERE id = %s', [request['id']]):
540                     user_email= user.email
541                     first_name = user.first_name
542                     last_name = user.last_name
543
544                 ctx = {
545                     'first_name'    : first_name, 
546                     'last_name'     : last_name, 
547                     'portal_url'    : current_site,
548                     }
549                 try:
550                     theme.template_name = 'user_request_denied.txt'
551                     text_content = render_to_string(theme.template, ctx)
552                     theme.template_name = 'user_request_denied.html'
553                     html_content = render_to_string(theme.template, ctx)
554                     theme.template_name = 'email_default_sender.txt'
555                     sender =  render_to_string(theme.template, ctx)
556                     sender = sender.replace('\n', '')
557                                
558                     subject = 'User request denied.'
559
560                     msg = EmailMultiAlternatives(subject, text_content, sender, [user_email])
561                     msg.attach_alternative(html_content, "text/html")
562                     msg.send()
563                 except Exception, e:
564                     print "Failed to send email, please check the mail templates and the SMTP configuration of your server"   
565             
566                 # removing from Django auth_user
567                 UserModel = get_user_model()
568                 UserModel._default_manager.filter(email__iexact = user_email).delete()
569                 # removing from Django portal_pendinguser
570                 PendingUser.objects.get(id=request['id']).delete()
571                 # removing from manifold
572                 # removing manifold account
573                 user_query = Query().get('local:user') \
574                     .filter_by('email', '==', user_email)           \
575                     .select('user_id')
576                 user = execute_admin_query(wsgi_request, user_query)
577                 user_id = user[0]['user_id']
578         
579                 platform_query = Query().get('local:platform') \
580                     .filter_by('platform', '==', 'myslice')           \
581                     .select('platform_id')
582                 platform = execute_admin_query(wsgi_request, platform_query)
583                 platform_id = platform[0]['platform_id']
584                 account_params = {'user_id':user_id}
585                 manifold_delete_account(request, platform_id, user_id, account_params)           
586              
587                 # removing manifold user
588                 user_params = {'user_id':user_id}
589                 manifold_delete_user(request, user_id, user_params)
590             except Exception, e:
591                 request_status['SFA authority'] = {'status': False, 'description': str(e)}
592                       
593         elif request['type'] == 'slice':
594             request_status['SFA slice'] = {'status': True } 
595
596             # getting user email based on id 
597             ## RAW SQL queries on Django DB- https://docs.djangoproject.com/en/dev/topics/db/sql/
598             for user in PendingSlice.objects.raw('SELECT * FROM portal_pendingslice WHERE id = %s', [request['id']]):
599                 user_email= user.type_of_nodes # XXX type_of_nodes field contains the email [shd be renamed in DB]
600                 slice_name = user.slice_name
601                 purpose = user.purpose
602                 url = user.number_of_nodes
603
604             ctx = {
605                 'slice_name': slice_name,
606                 'purpose': purpose,
607                 'url': url,
608                 'portal_url': current_site,
609                 }
610             try:
611                 theme.template_name = 'slice_request_denied.txt'
612                 text_content = render_to_string(theme.template, ctx)
613                 theme.template_name = 'slice_request_denied.html'
614                 html_content = render_to_string(theme.template, ctx)
615                 theme.template_name = 'email_default_sender.txt'
616                 sender =  render_to_string(theme.template, ctx)
617                 sender = sender.replace('\n', '')
618                                
619                 subject = 'Slice request denied.'
620
621                 msg = EmailMultiAlternatives(subject, text_content, sender, [user_email])
622                 msg.attach_alternative(html_content, "text/html")
623                 msg.send()
624             except Exception, e:
625                 print "Failed to send email, please check the mail templates and the SMTP configuration of your server"
626                       
627             PendingSlice.objects.get(id=request['id']).delete()
628
629         elif request['type'] == 'authority':
630             request_status['SFA authority'] = {'status': True }
631             
632             # getting user email based on id 
633             ## RAW SQL queries on Django DB- https://docs.djangoproject.com/en/dev/topics/db/sql/
634             for user in PendingAuthority.objects.raw('SELECT * FROM portal_pendingauthority WHERE id = %s', [request['id']]):
635                 user_email= user.address_line1 # XXX address_line1 field contains the email [shd be renamed in DB]
636                 site_name = user.site_name
637                 city = user.address_city
638                 country = user.address_country
639                 short_name = user.site_abbreviated_name
640                 url = user.site_url
641
642             ctx = { 
643                 'site_name': site_name,
644                 'short_name': short_name,
645                 'url': url,
646                 'city': city,
647                 'country': country,                          
648                 'portal_url'    : current_site,
649                 }
650                 
651             try:
652                 theme.template_name = 'authority_request_denied.txt'
653                 text_content = render_to_string(theme.template, ctx)
654                 theme.template_name = 'authority_request_denied.html'
655                 html_content = render_to_string(theme.template, ctx)
656                 theme.template_name = 'email_default_sender.txt'
657                 sender =  render_to_string(theme.template, ctx)
658                 sender = sender.replace('\n', '')
659                 subject = 'Authority request denied.'
660                 msg = EmailMultiAlternatives(subject, text_content, sender, [user_email])
661                 msg.attach_alternative(html_content, "text/html")
662                 msg.send()
663             except Exception, e:
664                 print "Failed to send email, please check the mail templates and the SMTP configuration of your server"
665
666             PendingAuthority.objects.get(id=request['id']).delete()
667
668         status['%s__%s' % (request['type'], request['id'])] = request_status
669
670     return status
671
672 # Django and ajax
673 # http://djangosnippets.org/snippets/942/
674
675
676
677 #-------------------------------------------------------------------------------
678 # REQUESTS - Slices
679 #-------------------------------------------------------------------------------
680
681 def create_slice(wsgi_request, request):
682     """
683     Arguments:
684         wsgi_request (~ WSGIRequest) : 
685         request (dict) : the slice request in our own dict format
686
687     Raises:
688         Exception
689     """
690     hrn = "%s.%s" % (request['authority_hrn'], request['slice_name'])
691     # XXX tmp sfa dependency
692     from sfa.util.xrn import Xrn 
693     urn = Xrn(hrn, request['type']).get_urn()
694     
695     # Add User to Slice if we have the user_hrn in pendingslice table
696     user_hrn = request.get('user_hrn', None)
697     user_hrns = list([user_hrn]) if user_hrn else list()
698     
699     # REGISTRY ONLY TO BE REMOVED WITH MANIFOLD-V2
700     user_query  = Query().get('myslice:user').select('user_hrn','user_email').filter_by('user_hrn','==',user_hrn)
701     user_details_sfa = execute_admin_query(wsgi_request, user_query)
702     if not user_details_sfa:
703         raise Exception, "User %s doesn't exist, validate user before validating slice" % user_hrn
704     for user in user_details_sfa:
705         user_email = user['user_email']
706
707     # XXX LOIC Quick fix because this is totally inconsistent
708     if not 'number_of_nodes' in request:
709         request['number_of_nodes']=""
710
711     # XXX We should create a slice with Manifold terminology
712     slice_params = {
713         'slice_hrn'        : hrn, 
714         'slice_urn'        : urn,
715         'slice_type'       : request['type'],
716         'url'              : request['number_of_nodes'],
717         'users'            : user_hrns,
718         'slice_enabled'    : True
719     }
720     # ignored in request: id, timestamp,  number_of_nodes, type_of_nodes, purpose
721
722     # REGISTRY ONLY TO BE REMOVED WITH MANIFOLD-V2
723     query = Query.create('myslice:slice').set(slice_params).select('slice_hrn')
724     results = execute_query(wsgi_request, query)
725     if not results:
726         raise Exception, "Could not create %s. Already exists ?" % slice_params['hrn']
727     else:
728         clear_user_creds(wsgi_request,user_email)
729         # log user activity
730         activity.slice.validate(request, { "slice" : hrn })
731         try:
732             theme.template_name = 'slice_request_validated.txt'
733             text_content = render_to_string(theme.template, request)
734             theme.template_name = 'slice_request_validated.html'
735             html_content = render_to_string(theme.template, request)
736         
737             theme.template_name = 'email_default_sender.txt'
738             sender =  render_to_string(theme.template, request)
739             sender = sender.replace('\n', '')
740
741             subject = 'Slice request validated'
742
743             msg = EmailMultiAlternatives(subject, text_content, sender, [user_email])
744             msg.attach_alternative(html_content, "text/html")
745             msg.send()
746         except Exception, e:
747             print "Failed to send email, please check the mail templates and the SMTP configuration of your server"
748        
749     return results
750
751 def create_pending_slice(wsgi_request, request, email):
752     """
753     """
754
755     # Insert an entry in the PendingSlice table
756     s = PendingSlice(
757         slice_name      = request['slice_name'],
758         user_hrn        = request['user_hrn'],
759         authority_hrn   = request['authority_hrn'],
760         number_of_nodes = request['url'], # field needs to be renamed
761         purpose         = request['purpose'],
762         type_of_nodes   = request['email'] # field needs to be renamed 
763     )
764     s.save()
765
766     try:
767         # Send an email: the recipients are the PI of the authority
768         recipients = authority_get_pi_emails(wsgi_request, request['authority_hrn'])
769
770         theme.template_name = 'slice_request_email.txt' 
771         text_content = render_to_string(theme.template, request)
772     
773         theme.template_name = 'slice_request_email.html' 
774         html_content = render_to_string(theme.template, request)
775     
776         theme.template_name = 'slice_request_email_subject.txt'
777         subject = render_to_string(theme.template, request)
778         subject = subject.replace('\n', '')
779     
780         sender = email
781         msg = EmailMultiAlternatives(subject, text_content, sender, recipients)
782         msg.attach_alternative(html_content, "text/html")
783         msg.send()
784     except Exception, e:
785         print "Failed to send email, please check the mail templates and the SMTP configuration of your server"
786
787 #-------------------------------------------------------------------------------
788 # REQUESTS - Users
789 #-------------------------------------------------------------------------------
790
791 def manifold_add_reference_user_accounts(wsgi_request, request):
792     """When a new user is created, add reference accounts to the reference platform.
793     """
794     # XXX XXX XXX The rest of this function has to be checked XXX XXX XXX
795
796     # Retrieve user information
797     user_query  = Query().get('local:user')             \
798         .select('user_id', 'config', 'email', 'status') \
799         .filter_by('email', '==', request['email'])
800     user_details = execute_admin_query(wsgi_request, user_query)
801
802     # USER MAIN ACCOUNT != reference
803     #print 'USER MAIN ACCOUNT != reference'
804     list_accounts_query = Query().get('local:account')              \
805         .select('user_id', 'platform_id', 'auth_type', 'config')    \
806         .filter_by('user_id', '==', user_details[0]['user_id'])     \
807         .filter_by('auth_type', '!=', 'reference')
808     list_accounts = execute_admin_query(wsgi_request, list_accounts_query)
809
810     # XXX main_platform is being erased several times ???
811     for account in list_accounts:
812         main_platform_query = Query().get('local:platform')         \
813             .select('platform_id', 'platform')                      \
814             .filter_by('platform_id', '==', account['platform_id'])
815         main_platform = execute_admin_query(wsgi_request, main_platform_query)
816
817     # Add reference accounts on SFA enabled platforms
818     platforms_query = Query().get('local:platform') \
819         .filter_by('disabled', '==', '0')           \
820         .filter_by('gateway_type', '==', 'sfa')     \
821         .select('platform_id', 'gateway_type')
822     platforms = execute_admin_query(wsgi_request, platforms_query)
823     for platform in platforms:
824         #print "add reference to platform ",platform
825         manifold_account_params = {
826             'user_id'       : user_details[0]['user_id'],
827             'platform_id'   : platform['platform_id'],
828             'auth_type'     : 'reference',
829             'config'        : '{"reference_platform": "' + main_platform[0]['platform'] + '"}',
830         }
831         manifold_add_account(wsgi_request, manifold_account_params)
832
833 def sfa_create_user(wsgi_request, request, namespace = None, as_admin = False):
834     """
835     Arguments:
836         wsgi_request (~ WSGIRequest) : 
837         request (dict) : the user request in our own dict format
838
839     Raises:
840         Exception
841     """
842     from sfa.util.xrn import Xrn 
843
844     auth_pi = request.get('pi', None)
845     auth_pi = list([auth_pi]) if auth_pi else list()
846
847     # We create a user request with Manifold terminology
848     sfa_user_params = {
849         'user_hrn'          : request['user_hrn'],
850         'user_email'        : request['email'],
851         'user_urn'          : Xrn(request['user_hrn'], request['type']).get_urn(),
852         'user_type'         : request['type'],
853         'keys'              : request['public_key'],
854         'user_first_name'   : request['first_name'],
855         'user_last_name'    : request['last_name'],
856         'pi_authorities'    : auth_pi,
857         'user_enabled'      : True
858     }
859
860     if namespace is not None:
861         query = Query.create('%s:user' % namespace).set(sfa_user_params).select('user_hrn')
862     else:
863         # REGISTRY ONLY TO BE REMOVED WITH MANIFOLD-V2
864         query = Query.create('myslice:user').set(sfa_user_params).select('user_hrn')
865
866     if as_admin:
867         results = execute_admin_query(wsgi_request, query)
868     else:
869         results = execute_query(wsgi_request, query)
870
871     if not results:
872         raise Exception, "Could not create %s. Already exists ?" % sfa_user_params['user_hrn']
873     else:
874         try:
875             theme.template_name = 'user_request_validated.txt'
876             text_content = render_to_string(theme.template, request)
877             theme.template_name = 'user_request_validated.html'
878             html_content = render_to_string(theme.template, request)
879         
880             theme.template_name = 'email_default_sender.txt'
881             sender =  render_to_string(theme.template, request)
882             sender = sender.replace('\n', '')
883
884
885             subject = 'User validated'
886
887             msg = EmailMultiAlternatives(subject, text_content, sender, [request['email']])
888             msg.attach_alternative(html_content, "text/html")
889             msg.send()
890         except Exception, e:
891             print "Failed to send email, please check the mail templates and the SMTP configuration of your server"
892
893     return results
894
895 def iotlab_create_user (wsgi_request, request, namespace = None, as_admin=False):
896    
897     import requests
898     import time
899     from requests.auth import HTTPBasicAuth
900     
901     URL_REST = ConfigEngine.default_iotlab_url
902     LOGIN_ADMIN = ConfigEngine.default_iotlab_admin_user
903     PASSWORD_ADMIN = ConfigEngine.default_iotlab_admin_password
904
905     auth = HTTPBasicAuth(LOGIN_ADMIN,PASSWORD_ADMIN)
906     headers = {'content-type': 'application/json'}
907
908     for user in PendingUser.objects.raw('SELECT * FROM portal_pendinguser WHERE email = %s', [request['email']]):
909         password= user.password
910
911
912     iotlab_user_params = {
913         "type"          : "SA",
914         #"login"         : request['email'], #auto generated by iotlab
915         "password"      : password,
916         "firstName"     : request['first_name'],
917         "lastName"      : request['last_name'],
918         "email"         : request['email'],
919         "structure"     : request['authority_hrn'],
920         "city"          : "N/A",
921         "country"       : "N/A",
922         "sshPublicKey"  : request['public_key'],
923         "motivations"   : "SFA federation",
924     }    
925    
926     iotlab_user_params1 = json.dumps(iotlab_user_params)
927     r=requests.post(url=URL_REST, data=iotlab_user_params1, headers=headers, auth=auth)
928     print 'Create iotlab user : ', r.status_code, r.text
929     return r.text
930
931 def create_user(wsgi_request, request, namespace = None, as_admin = False):
932     # XXX This has to be stored centrally
933     USER_STATUS_ENABLED = 2
934
935     # NOTE : if we were to create a user directly (just like we create slices,
936     # we would have to perform the steps in create_pending_user too
937
938     # Add the user to the SFA registry
939     sfa_create_user(wsgi_request, request, namespace, as_admin)
940
941     # Update Manifold user status
942     manifold_update_user(wsgi_request, request['email'], {'status': USER_STATUS_ENABLED})
943
944     # Add reference accounts for platforms
945     manifold_add_reference_user_accounts(wsgi_request, request)
946
947     # Add the user to iotlab portal if theme is set to onelab
948     if theme.theme == 'onelab':
949         iotlab_create_user (wsgi_request, request)
950
951 def create_pending_user(wsgi_request, request, user_detail):
952     """
953     """
954
955     # Insert an entry in the PendingUser table
956     b = PendingUser(
957         first_name    = request['first_name'],
958         last_name     = request['last_name'],
959         authority_hrn = request['authority_hrn'],
960         email         = request['email'],
961         password      = request['password'],
962         public_key    = request['public_key'],
963         private_key   = request['private_key'],
964         user_hrn      = request['user_hrn'],
965         pi            = request['pi'],
966         email_hash    = request['email_hash'],
967         status        = 'False',
968     )
969     b.save()
970     # sends email to user to activate the email
971     theme.template_name = 'activate_user.html'
972     html_content = render_to_string(theme.template, request)
973     theme.template_name = 'activate_user.txt'
974     text_content = render_to_string(theme.template, request)
975     theme.template_name = 'activate_user_email_subject.txt'
976     subject = render_to_string(theme.template, request)
977     subject = subject.replace('\n', '')
978     theme.template_name = 'email_default_sender.txt'
979     sender =  render_to_string(theme.template, request)
980     sender = sender.replace('\n', '')
981     recipient = [request['email']]
982     #recipient = recipient.append(request['email'])
983
984     msg = EmailMultiAlternatives(subject, text_content, sender, recipient)
985     msg.attach_alternative(html_content, "text/html")
986     msg.send()
987    
988     # saves the user to django auth_user table [needed for password reset]
989     user = User.objects.create_user(request['email'], request['email'], request['password'])
990
991     # Creating a manifold user
992     user_id = manifold_add_user(wsgi_request, request)
993
994     # Creating a Manifold account on the MySlice platform
995     # Note the JSON representation of public and private keys already includes quotes
996     account_config = {
997         'user_hrn'          : request['user_hrn'],
998         'user_public_key'   : request['public_key'],
999     }
1000     if request['private_key']:
1001         account_config['user_private_key'] = request['private_key']
1002
1003     user_id = user_detail['user_id'] + 1 # the user_id for the newly created user in local:user
1004
1005     # XXX TODO: Require a myslice platform
1006     # ALERT: this will disapear with ROUTERV2 of Manifold
1007     # We have to consider the case where several registries can be used
1008     # Removed hardcoded platform = 5
1009     # This platform == 'myslice' is a TMP FIX !!
1010     try:
1011         reg_platform_query = Query().get('local:platform') \
1012             .filter_by('platform', '==', 'myslice')           \
1013             .select('platform_id')
1014         reg_platform = execute_admin_query(wsgi_request, reg_platform_query)
1015
1016         reg_platform_id = reg_platform[0]['platform_id']
1017         account_params = {
1018             'platform_id'   : reg_platform_id, # XXX ALERT !!
1019             'user_id'       : user_id, 
1020             'auth_type'     : request['auth_type'], 
1021             'config'        : json.dumps(account_config),
1022         }
1023         manifold_add_account(wsgi_request, account_params)
1024     except Exception, e:
1025         print "Failed creating manifold account on platform %s for user: %s" % ('myslice', request['email'])
1026
1027     try:
1028         # Send an email: the recipients are the PI of the authority
1029         # If No PI is defined for this Authority, send to a default email (different for each theme)
1030         recipients = authority_get_pi_emails(wsgi_request, request['authority_hrn'])
1031         
1032         theme.template_name = 'user_request_email.html'
1033         html_content = render_to_string(theme.template, request)
1034  
1035         theme.template_name = 'user_request_email.txt'
1036         text_content = render_to_string(theme.template, request)
1037     
1038         theme.template_name = 'user_request_email_subject.txt'
1039         subject = render_to_string(theme.template, request)
1040         subject = subject.replace('\n', '')
1041     
1042         theme.template_name = 'email_default_sender.txt'
1043         sender =  render_to_string(theme.template, request)
1044         sender = sender.replace('\n', '')
1045     
1046         msg = EmailMultiAlternatives(subject, text_content, sender, recipients)
1047         msg.attach_alternative(html_content, "text/html")
1048         msg.send()
1049     except Exception, e:
1050         print "Failed to send email, please check the mail templates and the SMTP configuration of your server"
1051         import traceback
1052         traceback.print_exc()