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