portal/registrationview.py

   1 import os.path, re
   2
   3 from django.core.mail           import send_mail
   4
   5 from django.views.generic       import View
   6 from django.template.loader     import render_to_string
   7 from django.shortcuts           import render
   8
   9 from unfold.page                import Page
  10 from myslice.viewutils          import topmenu_items
  11
  12 from manifold.manifoldapi       import execute_query
  13 from manifold.core.query        import Query
  14
  15 from portal.models              import PendingUser
  16 from portal.actions             import authority_get_pi_emails
  17
  18 # This is a rough porting from views.py
  19 # the former function-based view is now made a class
  20 # we redefine dispatch as it is simple
  21 # and coincidentally works since we do not need LoginRequiredAutoLogoutView
  22 # a second stab should redefine post and get instead
  23 # also this was not thoroughly tested either, might miss some imports
  24 # to be continued...
  25
  26 class RegistrationView (View):
  27
  28     def dispatch (self, request):
  29
  30         errors = []
  31
  32         authorities_query = Query.get('authority').\
  33             filter_by('authority_hrn', 'included', ['ple.inria', 'ple.upmc']).\
  34             select('name', 'authority_hrn')
  35         #authorities_query = Query.get('authority').select('authority_hrn')
  36         authorities = execute_query(request, authorities_query)
  37         # xxx tocheck - if authorities is empty, it's no use anyway
  38         # (users won't be able to validate the form anyway)
  39
  40         page = Page(request)
  41         page.add_js_files  ( [ "js/jquery.validate.js", "js/my_account.register.js" ] )
  42         page.add_css_files ( [ "css/onelab.css", "css/registration.css" ] )
  43
  44         print 'registration view, method',request.method
  45
  46         if request.method == 'POST':
  47             # We shall use a form here
  48
  49             #get_email = PendingUser.objects.get(email)
  50             reg_fname = request.POST.get('firstname', '')
  51             reg_lname = request.POST.get('lastname', '')
  52             #reg_aff = request.POST.get('affiliation','')
  53             reg_auth = request.POST.get('authority_hrn', '')
  54             reg_email = request.POST.get('email','').lower()
  55
  56             #POST value validation
  57             if (re.search(r'^[\w+\s.@+-]+$', reg_fname)==None):
  58                 errors.append('First Name may contain only letters, numbers, spaces and @/./+/-/_ characters.')
  59             if (re.search(r'^[\w+\s.@+-]+$', reg_lname) == None):
  60                 errors.append('Last Name may contain only letters, numbers, spaces and @/./+/-/_ characters.')
  61             # XXX validate authority hrn !!
  62             if PendingUser.objects.filter(email__iexact=reg_email):
  63                 errors.append('Email already registered.Please provide a new email address.')
  64             if 'generate' in request.POST['question']:
  65                 # Generate public and private keys using SFA Library
  66                 from sfa.trust.certificate  import Keypair
  67                 k = Keypair(create=True)
  68                 public_key = k.get_pubkey_string()
  69                 private_key = k.as_pem()
  70                 private_key = ''.join(private_key.split())
  71                 public_key = "ssh-rsa " + public_key
  72                 # Saving to DB
  73                 keypair = '{"user_public_key":"'+ public_key + '", "user_private_key":"'+ private_key + '"}'
  74                 #keypair = re.sub("\r", "", keypair)
  75                 #keypair = re.sub("\n", "\\n", keypair)
  76                 #keypair = keypair.rstrip('\r\n')
  77                 #keypair = ''.join(keypair.split())
  78             else:
  79                 up_file = request.FILES['user_public_key']
  80                 file_content =  up_file.read()
  81                 file_name = up_file.name
  82                 file_extension = os.path.splitext(file_name)[1]
  83                 allowed_extension =  ['.pub','.txt']
  84                 if file_extension in allowed_extension and re.search(r'ssh-rsa',file_content):
  85                     keypair = '{"user_public_key":"'+ file_content +'"}'
  86                     keypair = re.sub("\r", "", keypair)
  87                     keypair = re.sub("\n", "\\n",keypair)
  88                     keypair = ''.join(keypair.split())
  89                     # for sending email
  90                     public_key = file_content
  91                 else:
  92                     errors.append('Please upload a valid RSA public key [.txt or .pub].')
  93
  94             #b = PendingUser(first_name=reg_fname, last_name=reg_lname, affiliation=reg_aff,
  95             #                email=reg_email, password=request.POST['password'], keypair=keypair)
  96             #b.save()
  97             if not errors:
  98                 b = PendingUser(
  99                     first_name=reg_fname,
 100                     last_name=reg_lname,
 101                     #affiliation=reg_aff,
 102                     authority_hrn=reg_auth,
 103                     email=reg_email,
 104                     password=request.POST['password'],
 105                     keypair=keypair
 106                     )
 107                 b.save()
 108
 109                 # Send email
 110                 ctx = {
 111                     'First Name'    : reg_fname,
 112                     'Last Name'     : reg_lname,
 113                     'Authority'     : reg_auth,
 114                     'Email'         : reg_email,
 115                     'Public Key'    : public_key,
 116                     'cc_myself'     : True # form.cleaned_data['cc_myself']
 117                     }
 118                 #not working
 119                 #recipients = authority_get_pi_emails(request,reg_auth)
 120                 recipients = ['devel@myslice.info']
 121                 if ctx['cc_myself']:
 122                     recipients.append(ctx['Email'])
 123
 124                 msg = render_to_string('user_request_email.txt', ctx)
 125                 send_mail("Onelab New User request for %s submitted"%reg_email, msg, reg_email, recipients)
 126
 127                 return render(request, 'user_register_complete.html')
 128
 129         template_env = {
 130           'topmenu_items': topmenu_items('Register', request),
 131           'errors': errors,
 132           'firstname': request.POST.get('firstname', ''),
 133           'lastname': request.POST.get('lastname', ''),
 134           #'affiliation': request.POST.get('affiliation', ''),
 135           'authority_hrn': request.POST.get('authority_hrn', ''),
 136           'email': request.POST.get('email', ''),
 137           'password': request.POST.get('password', ''),
 138           'authorities': authorities,
 139           }
 140         template_env.update(page.prelude_env ())
 141         return render(request, 'registration_view.html',template_env)