1 from django.http import HttpResponse
2 from manifold.core.query import Query
3 from manifoldapi.manifoldapi import execute_query,execute_admin_query
4 from portal.models import PendingUser, PendingSlice, PendingAuthority
7 from django.contrib.auth.models import User
8 from django.template.loader import render_to_string
9 from django.core.mail import EmailMultiAlternatives, send_mail
11 from myslice.theme import ThemeView
15 # Thierry: moving this right into the code so
16 # most people can use myslice without having to install sfa
17 # XXX tmp sfa dependency, should be moved to SFA gateway
18 #from sfa.util.xrn import Xrn
21 # Get the list of authorities
23 def authority_get_pis(request, authority_hrn):
24 query = Query.get('authority').filter_by('authority_hrn', '==', authority_hrn).select('pi_users')
25 results = execute_admin_query(request, query)
26 print "authority_get_pis = %s" % results
27 # NOTE: temporarily commented. Because results is giving empty list.
28 # Needs more debugging
30 # raise Exception, "Authority not found: %s" % authority_hrn
32 #return result['pi_users']
35 def authority_get_pi_emails(request, authority_hrn):
36 pi_users = authority_get_pis(request,authority_hrn)
37 print "pi_users = %s" % pi_users
39 if any(pi['pi_users'] == None or not pi['pi_users'] for pi in pi_users):
40 #theme.template_name = 'email_default_recipients.txt'
41 #default_email = render_to_string(theme.template, request)
42 #default_email = default_email.replace('\n', '')
44 # the above doesn't work
45 return ['support@myslice.info']
47 pi_user_hrns = [ hrn for x in pi_users for hrn in x['pi_users'] ]
48 query = Query.get('user').filter_by('user_hrn', 'included', pi_user_hrns).select('user_email')
49 results = execute_admin_query(request, query)
50 return [result['user_email'] for result in results]
52 #clear user credentials
53 def clear_user_creds(request, user_email):
55 user_query = Query().get('local:user').filter_by('email', '==', user_email).select('user_id','email','password','config')
56 user_details = execute_admin_query(request, user_query)
58 # getting the user_id from the session
59 for user_detail in user_details:
60 user_id = user_detail['user_id']
61 user_email = user_detail['email']
63 account_query = Query().get('local:account').filter_by('user_id', '==', user_id).select('user_id','platform_id','auth_type','config')
64 account_details = execute_admin_query(request, account_query)
66 platform_query = Query().get('local:platform').select('platform_id','platform')
67 platform_details = execute_admin_query(request, platform_query)
69 for account_detail in account_details:
70 for platform_detail in platform_details:
71 if platform_detail['platform_id'] == account_detail['platform_id']:
72 if 'myslice' in platform_detail['platform']:
73 account_config = json.loads(account_detail['config'])
74 #user_cred = account_config.get('delegated_user_credential','N/A')
75 user_cred = account_config.get('user_credential','N/A')
76 if 'N/A' not in user_cred:
77 user_hrn = account_config.get('user_hrn','N/A')
78 user_pub_key = json.dumps(account_config.get('user_public_key','N/A'))
79 user_priv_key = json.dumps(account_config.get('user_private_key','N/A'))
80 updated_config = '{"user_public_key":'+ user_pub_key + ', "user_private_key":'+ user_priv_key + ', "user_hrn":"'+ user_hrn + '"}'
81 user_params = { 'config': updated_config}
82 manifold_update_account(request, user_id,user_params)
88 print "Exception in actions.py in clear_user_creds %s" % e
91 def is_pi(wsgi_request, user_hrn, authority_hrn):
92 # XXX could be done in a single query !
94 # select pi_authorities from user where user_hrn == "ple.upmc.jordan_auge"
95 query = Query.get('user').filter_by('user_hrn', '==', user_hrn).select('pi_authorities')
96 results = execute_admin_query(wsgi_request, query)
101 user_authority_hrns = result.get('pi_authorities', [])
102 return authority_hrn in user_authority_hrns
106 def sfa_get_user(request, user_hrn, pub):
107 query_sfa_user = Query.get('user').filter_by('user_hrn', '==', user_hrn)
108 result_sfa_user = execute_query(request, query_sfa_user)
109 return result_sfa_user
111 def sfa_update_user(request, user_hrn, user_params):
112 # user_params: keys [public_key]
113 if 'email' in user_params:
114 user_params['user_email'] = user_params['email']
115 query = Query.update('user').filter_by('user_hrn', '==', user_hrn).set(user_params).select('user_hrn')
116 results = execute_query(request,query)
119 def sfa_add_authority(request, authority_params):
120 query = Query.create('authority').set(authority_params).select('authority_hrn')
121 results = execute_query(request, query)
122 print "sfa_add_auth results=",results
124 raise Exception, "Could not create %s. Already exists ?" % authority_params['hrn']
127 def sfa_add_user_to_slice(request, user_hrn, slice_params):
128 # 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'
129 query_current_users = Query.get('slice').select('user').filter_by('slice_hrn','==',slice_params['hrn'])
130 results_current_users = execute_query(request, query_current_users)
131 slice_params['researcher'] = slice_params['researcher'] | results_current_users
132 query = Query.update('slice').filter_by('user_hrn', '==', user_hrn).set(slice_params).select('slice_hrn')
133 results = execute_query(request, query)
134 # Also possible but not supported yet
135 # UPDATE myslice:user SET slice=['ple.upmc.agent','ple.upmc.myslicedemo','ple.upmc.tophat'] where user_hrn=='ple.upmc.azerty'
137 raise Exception, "Could not create %s. Already exists ?" % slice_params['hrn']
142 def manifold_add_user(wsgi_request, request):
143 """Add a Manifold user corresponding to a user request.
146 wsgi_request: a WSGIRequest instance
147 request (dict): a dictionary containing the user request built from the
151 The user_id of the inserted user.
157 USER_CONFIG = '{"firstname": "%(first_name)s", "lastname": "%(last_name)s", "authority": "%(authority_hrn)s"}'
160 'email' : request['email'],
161 'password' : request['password'],
162 'config' : USER_CONFIG % request,
166 query = Query.create('local:user').set(user_params).select('email')
167 results = execute_admin_query(request, query)
169 raise Exception, "Failed creating manifold user: %s" % user_params['email']
171 return result['email']
173 def manifold_update_user(request, email, user_params):
174 # user_params: password, config e.g.,
175 query = Query.update('local:user').filter_by('email', '==', email).set(user_params).select('email')
176 results = execute_admin_query(request,query)
177 # NOTE: results remains empty and goes to Exception. However, it updates the manifold DB.
178 # That's why I commented the exception part. -- Yasin
180 # raise Exception, "Failed updating manifold user: %s" % user_params['email']
184 def manifold_add_account(request, account_params):
185 query = Query.create('local:account').set(account_params).select(['user', 'platform'])
186 results = execute_admin_query(request,query)
188 raise Exception, "Failed creating manifold account on platform %s for user: %s" % (account_params['platform'], account_params['user'])
190 return result['user_id']
192 def manifold_update_account(request,user_id,account_params):
193 # account_params: config
194 query = Query.update('local:account').filter_by('platform', '==', 'myslice').filter_by('user_id', '==', user_id).set(account_params).select('user_id')
195 results = execute_admin_query(request,query)
198 #explicitly mention the platform_id
199 def manifold_delete_account(request, platform_id, user_id, account_params):
200 query = Query.delete('local:account').filter_by('platform_id', '==', platform_id).filter_by('user_id', '==', user_id).set(account_params).select('user_id')
201 results = execute_admin_query(request,query)
206 def manifold_add_platform(request, platform_params):
207 query = Query.create('local:platform').set(platform_params).select(['user', 'platform'])
208 results = execute_admin_query(request,query)
210 raise Exception, "Failed creating manifold platform %s for user: %s" % (platform_params['platform'], platform_params['user'])
212 return result['platform_id']
215 def make_request_user(user):
217 request['type'] = 'user'
218 request['id'] = user.id
219 request['timestamp'] = user.created # XXX in DB ?
220 request['authority_hrn'] = user.authority_hrn
221 request['first_name'] = user.first_name
222 request['last_name'] = user.last_name
223 request['email'] = user.email
224 request['login'] = user.login
225 request['user_hrn'] = user.user_hrn
226 request['public_key'] = user.public_key
227 request['private_key'] = user.private_key
230 def make_request_slice(slice):
232 request['type'] = 'slice'
233 request['id'] = slice.id
234 request['user_hrn'] = slice.user_hrn
235 request['timestamp'] = slice.created
236 request['authority_hrn'] = slice.authority_hrn
237 request['slice_name'] = slice.slice_name
238 request['number_of_nodes'] = slice.number_of_nodes
239 request['type_of_nodes'] = slice.type_of_nodes
240 request['purpose'] = slice.purpose
243 def make_request_authority(authority):
245 request['type'] = 'authority'
246 request['id'] = authority.id
247 request['site_name'] = authority.site_name
248 request['site_latitude'] = authority.site_latitude
249 request['site_longitude'] = authority.site_longitude
250 request['site_url'] = authority.site_url
251 request['site_authority'] = authority.site_authority
252 request['site_abbreviated_name'] = authority.site_abbreviated_name
253 request['address_line1'] = authority.address_line1
254 request['address_line2'] = authority.address_line2
255 request['address_line3'] = authority.address_line3
256 request['address_city'] = authority.address_city
257 request['address_postalcode'] = authority.address_postalcode
258 request['address_state'] = authority.address_state
259 request['address_country'] = authority.address_country
260 request['authority_hrn'] = authority.authority_hrn
261 request['timestamp'] = authority.created
264 def make_requests(pending_users, pending_slices, pending_authorities):
266 for user in pending_users:
267 requests.append(make_request_user(user))
268 for slice in pending_slices:
269 requests.append(make_request_slice(slice))
270 for authority in pending_authorities:
271 requests.append(make_request_authority(authority))
274 def get_request_by_id(ids):
275 sorted_ids = { 'user': [], 'slice': [], 'authority': [] }
277 type, id = type__id.split('__')
278 sorted_ids[type].append(id)
281 pending_users = PendingUser.objects.all()
282 pending_slices = PendingSlice.objects.all()
283 pending_authorities = PendingAuthority.objects.all()
285 pending_users = PendingUser.objects.filter(id__in=sorted_ids['user']).all()
286 pending_slices = PendingSlice.objects.filter(id__in=sorted_ids['slice']).all()
287 pending_authorities = PendingAuthority.objects.filter(id__in=sorted_ids['authority']).all()
289 return make_requests(pending_users, pending_slices, pending_authorities)
291 def get_requests(authority_hrns=None):
292 print "get_request_by_authority auth_hrns = ", authority_hrns
293 if not authority_hrns:
294 ## get those pending users who have confirmed their emails
295 pending_users = PendingUser.objects.filter(status__iexact = 'True')
296 pending_slices = PendingSlice.objects.all()
297 pending_authorities = PendingAuthority.objects.all()
299 pending_users = PendingUser.objects.filter(authority_hrn__in=authority_hrns).all()
300 pending_slices = PendingSlice.objects.filter(authority_hrn__in=authority_hrns).all()
301 pending_authorities = PendingAuthority.objects.filter(authority_hrn__in=authority_hrns).all()
303 return make_requests(pending_users, pending_slices, pending_authorities)
305 # XXX Is it in sync with the form fields ?
307 def portal_validate_request(wsgi_request, request_ids):
310 if not isinstance(request_ids, list):
311 request_ids = [request_ids]
313 requests = get_request_by_id(request_ids)
314 for request in requests:
315 # type, id, timestamp, details, allowed -- MISSING: authority_hrn
316 # CAREFUL about details
317 # user : first name, last name, email, password, keypair
318 # slice : number of nodes, type of nodes, purpose
322 if request['type'] == 'user':
325 create_user(wsgi_request, request)
326 request_status['SFA user'] = {'status': True }
327 PendingUser.objects.get(id=request['id']).delete()
329 request_status['SFA user'] = {'status': False, 'description': str(e)}
331 # user_params = {'status':2}
332 # manifold_update_user(request, request['email'], user_params)
334 # MANIFOLD user should be added beforehand, during registration
336 # manifold_user_params = { key: request[key] for key in MANIFOLD_USER_KEYS }
337 # # XXX # manifold_add_user(manifold_user_params)
338 # request_status['MySlice user'] = {'status': True }
339 #except Exception, e:
340 # request_status['MySlice user'] = {'status': False, 'description': str(e)}
343 #manifold_account_params = { key: request[key] for key in MANIFOLD_ACCOUNT_KEYS }
344 #manifold_add_account(manifold_account_params)
345 #request_status['MySlice testbed accounts'] = {'status': False }
347 elif request['type'] == 'slice':
349 create_slice(wsgi_request, request)
350 request_status['SFA slice'] = {'status': True }
351 PendingSlice.objects.get(id=request['id']).delete()
354 request_status['SFA slice'] = {'status': False, 'description': str(e)}
356 elif request['type'] == 'authority':
358 #hrn = "%s.%s" % (request['authority_hrn'], request['site_authority'])
359 hrn = request['site_authority']
360 # XXX tmp sfa dependency
361 from sfa.util.xrn import Xrn
362 urn = Xrn(hrn, request['type']).get_urn()
364 # Only hrn is required for Manifold Query
365 sfa_authority_params = {
366 'authority_hrn' : hrn,
367 #'authority_urn' : urn,
368 #'type' : request['type'],
372 print "ADD Authority"
373 sfa_add_authority(wsgi_request, sfa_authority_params)
374 request_status['SFA authority'] = {'status': True }
375 PendingAuthority.objects.get(id=request['id']).delete()
378 request_status['SFA authority'] = {'status': False, 'description': str(e)}
380 # XXX Remove from Pendings in database
382 status['%s__%s' % (request['type'], request['id'])] = request_status
387 def validate_action(request, **kwargs):
388 ids = filter(None, kwargs['id'].split('/'))
389 status = portal_validate_request(request, ids)
390 json_answer = json.dumps(status)
391 return HttpResponse (json_answer, mimetype="application/json")
394 # http://djangosnippets.org/snippets/942/
398 #-------------------------------------------------------------------------------
400 #-------------------------------------------------------------------------------
402 def create_slice(wsgi_request, request):
405 wsgi_request (~ WSGIRequest) :
406 request (dict) : the slice request in our own dict format
411 hrn = "%s.%s" % (request['authority_hrn'], request['slice_name'])
412 # XXX tmp sfa dependency
413 from sfa.util.xrn import Xrn
414 urn = Xrn(hrn, request['type']).get_urn()
416 # Add User to Slice if we have the user_hrn in pendingslice table
417 user_hrn = request.get('user_hrn', None)
418 user_hrns = list([user_hrn]) if user_hrn else list()
420 user_query = Query().get('user').select('user_hrn','user_email').filter_by('user_hrn','==',user_hrn)
421 user_details_sfa = execute_admin_query(wsgi_request, user_query)
422 if not user_details_sfa:
423 raise Exception, "User %s doesn't exist, validate user before validating slice" % user_hrn
424 for user in user_details_sfa:
425 user_email = user['user_email']
427 # XXX We should create a slice with Manifold terminology
431 'slice_type' : request['type'],
432 'url' : request['number_of_nodes'],
434 'slice_enabled' : True
436 # ignored in request: id, timestamp, number_of_nodes, type_of_nodes, purpose
438 query = Query.create('slice').set(slice_params).select('slice_hrn')
439 results = execute_query(wsgi_request, query)
441 raise Exception, "Could not create %s. Already exists ?" % slice_params['hrn']
443 clear_user_creds(wsgi_request,user_email)
444 subject = 'Slice created'
445 msg = 'A manager of your institution has validated your slice request. You can now add resources to the slice and start experimenting.'
446 send_mail(subject, msg, 'support@onelab.eu',[user_email], fail_silently=False)
450 def create_pending_slice(wsgi_request, request, email):
454 # Insert an entry in the PendingSlice table
456 slice_name = request['slice_name'],
457 user_hrn = request['user_hrn'],
458 authority_hrn = request['authority_hrn'],
459 number_of_nodes = request['url'],
460 purpose = request['purpose'],
465 # Send an email: the recipients are the PI of the authority
466 recipients = authority_get_pi_emails(wsgi_request, request['authority_hrn'])
468 theme.template_name = 'slice_request_email.txt'
469 text_content = render_to_string(theme.template, request)
471 theme.template_name = 'slice_request_email.html'
472 html_content = render_to_string(theme.template, request)
474 theme.template_name = 'slice_request_email_subject.txt'
475 subject = render_to_string(theme.template, request)
476 subject = subject.replace('\n', '')
479 msg = EmailMultiAlternatives(subject, text_content, sender, recipients)
480 msg.attach_alternative(html_content, "text/html")
483 print "Failed to send email, please check the mail templates and the SMTP configuration of your server"
485 #-------------------------------------------------------------------------------
487 #-------------------------------------------------------------------------------
489 def manifold_add_reference_user_accounts(wsgi_request, request):
490 """When a new user is created, add reference accounts to the reference platform.
492 # XXX XXX XXX The rest of this function has to be checked XXX XXX XXX
494 # Retrieve user information
495 user_query = Query().get('local:user') \
496 .select('user_id', 'config', 'email', 'status') \
497 .filter_by('email', '==', request['email'])
498 user_details = execute_admin_query(wsgi_request, user_query)
500 # USER MAIN ACCOUNT != reference
501 #print 'USER MAIN ACCOUNT != reference'
502 list_accounts_query = Query().get('local:account') \
503 .select('user_id', 'platform_id', 'auth_type', 'config') \
504 .filter_by('user_id', '==', user_details[0]['user_id']) \
505 .filter_by('auth_type', '!=', 'reference')
506 list_accounts = execute_admin_query(wsgi_request, list_accounts_query)
508 # XXX main_platform is being erased several times ???
509 for account in list_accounts:
510 main_platform_query = Query().get('local:platform') \
511 .select('platform_id', 'platform') \
512 .filter_by('platform_id', '==', account['platform_id'])
513 main_platform = execute_admin_query(wsgi_request, main_platform_query)
515 # Add reference accounts on SFA enabled platforms
516 platforms_query = Query().get('local:platform') \
517 .filter_by('disabled', '==', '0') \
518 .filter_by('gateway_type', '==', 'sfa') \
519 .select('platform_id', 'gateway_type')
520 platforms = execute_admin_query(wsgi_request, platforms_query)
521 for platform in platforms:
522 #print "add reference to platform ",platform
523 manifold_account_params = {
524 'user_id' : user_details[0]['user_id'],
525 'platform_id' : platform['platform_id'],
526 'auth_type' : 'reference',
527 'config' : '{"reference_platform": "' + main_platform[0]['platform'] + '"}',
529 manifold_add_account(wsgi_request, manifold_account_params)
531 def sfa_create_user(wsgi_request, request, namespace = None, as_admin = False):
534 wsgi_request (~ WSGIRequest) :
535 request (dict) : the user request in our own dict format
540 from sfa.util.xrn import Xrn
542 auth_pi = request.get('pi', None)
543 auth_pi = list([auth_pi]) if auth_pi else list()
545 # We create a user request with Manifold terminology
547 'user_hrn' : request['user_hrn'],
548 'user_email' : request['email'],
549 'user_urn' : Xrn(request['user_hrn'], request['type']).get_urn(),
550 'user_type' : request['type'],
551 'keys' : request['public_key'],
552 'user_first_name' : request['first_name'],
553 'user_last_name' : request['last_name'],
554 'pi_authorities' : auth_pi,
555 'user_enabled' : True
558 if namespace is not None:
559 query = Query.create('%s:user' % namespace).set(sfa_user_params).select('user_hrn')
561 query = Query.create('user').set(sfa_user_params).select('user_hrn')
564 results = execute_admin_query(wsgi_request, query)
566 results = execute_query(wsgi_request, query)
569 raise Exception, "Could not create %s. Already exists ?" % sfa_user_params['user_hrn']
571 subject = 'User validated'
572 msg = 'A manager of your institution has validated your account. You have now full user access to the portal.'
573 send_mail(subject, msg, 'support@onelab.eu',[request['email']], fail_silently=False)
576 def create_user(wsgi_request, request, namespace = None, as_admin = False):
577 # XXX This has to be stored centrally
578 USER_STATUS_ENABLED = 2
580 # NOTE : if we were to create a user directly (just like we create slices,
581 # we would have to perform the steps in create_pending_user too
583 # Add the user to the SFA registry
584 sfa_create_user(wsgi_request, request, namespace, as_admin)
586 # Update Manifold user status
587 manifold_update_user(wsgi_request, request['email'], {'status': USER_STATUS_ENABLED})
589 # Add reference accounts for platforms
590 manifold_add_reference_user_accounts(wsgi_request, request)
592 def create_pending_user(wsgi_request, request, user_detail):
596 # Insert an entry in the PendingUser table
598 first_name = request['first_name'],
599 last_name = request['last_name'],
600 authority_hrn = request['authority_hrn'],
601 email = request['email'],
602 password = request['password'],
603 public_key = request['public_key'],
604 private_key = request['private_key'],
605 user_hrn = request['user_hrn'],
607 email_hash = request['email_hash'],
611 # sends email to user to activate the email
612 theme.template_name = 'activate_user.html'
613 html_content = render_to_string(theme.template, request)
614 theme.template_name = 'activate_user.txt'
615 text_content = render_to_string(theme.template, request)
616 theme.template_name = 'activate_user_email_subject.txt'
617 subject = render_to_string(theme.template, request)
618 subject = subject.replace('\n', '')
619 #sender = 'support@myslice.info'
620 theme.template_name = 'email_default_sender.txt'
621 sender = render_to_string(theme.template, request)
622 sender = sender.replace('\n', '')
623 recipient = [request['email']]
624 #recipient = recipient.append(request['email'])
626 msg = EmailMultiAlternatives(subject, text_content, sender, recipient)
627 msg.attach_alternative(html_content, "text/html")
630 # saves the user to django auth_user table [needed for password reset]
631 user = User.objects.create_user(request['email'], request['email'], request['password'])
633 # Creating a manifold user
634 user_id = manifold_add_user(wsgi_request, request)
636 # Creating a Manifold account on the MySlice platform
637 # Note the JSON representation of public and private keys already includes quotes
639 'user_hrn' : request['user_hrn'],
640 'user_public_key' : request['public_key'],
642 if request['private_key']:
643 account_config['user_private_key'] = request['private_key']
645 user_id = user_detail['user_id'] + 1 # the user_id for the newly created user in local:user
647 # XXX TODO: Require a myslice platform
648 # ALERT: this will disapear with ROUTERV2 of Manifold
649 # We have to consider the case where several registries can be used
650 # Removed hardcoded platform = 5
651 # This platform == 'myslice' is a TMP FIX !!
653 reg_platform_query = Query().get('local:platform') \
654 .filter_by('platform', '==', 'myslice') \
655 .select('platform_id')
656 reg_platform = execute_admin_query(wsgi_request, reg_platform_query)
658 reg_platform_id = reg_platform[0]['platform_id']
660 'platform_id' : reg_platform_id, # XXX ALERT !!
662 'auth_type' : request['auth_type'],
663 'config' : json.dumps(account_config),
665 manifold_add_account(wsgi_request, account_params)
667 print "Failed creating manifold account on platform %s for user: %s" % ('myslice', request['email'])
670 # Send an email: the recipients are the PI of the authority
671 # If No PI is defined for this Authority, send to a default email (different for each theme)
672 recipients = authority_get_pi_emails(wsgi_request, request['authority_hrn'])
674 theme.template_name = 'user_request_email.html'
675 html_content = render_to_string(theme.template, request)
677 theme.template_name = 'user_request_email.txt'
678 text_content = render_to_string(theme.template, request)
680 theme.template_name = 'user_request_email_subject.txt'
681 subject = render_to_string(theme.template, request)
682 subject = subject.replace('\n', '')
684 theme.template_name = 'email_default_sender.txt'
685 sender = render_to_string(theme.template, request)
686 sender = sender.replace('\n', '')
688 msg = EmailMultiAlternatives(subject, text_content, sender, recipients)
689 msg.attach_alternative(html_content, "text/html")
692 print "Failed to send email, please check the mail templates and the SMTP configuration of your server"
694 traceback.print_exc()