portal/registrationview.py

   1 import os.path, re
   2 import json
   3 from random import randint
   4
   5 from django.views.generic       import View
   6 from django.template.loader     import render_to_string
   7 from django.shortcuts           import render
   8 from django.contrib.auth        import get_user_model
   9 from django.contrib.sites.models import Site
  10
  11 from unfold.page                import Page
  12 from unfold.loginrequired       import FreeAccessView
  13 from ui.topmenu                 import topmenu_items_live
  14
  15 from manifoldapi.manifoldapi    import execute_admin_query
  16 from manifold.core.query        import Query
  17
  18 from portal.models              import PendingUser
  19 #from portal.actions             import create_pending_user
  20 # Edelberto - LDAP
  21 from portal.actions             import create_pending_user, ldap_create_user
  22
  23 from myslice.theme import ThemeView
  24
  25 # since we inherit from FreeAccessView we cannot redefine 'dispatch'
  26 # so let's override 'get' and 'post' instead
  27 #
  28 class RegistrationView (FreeAccessView, ThemeView):
  29     template_name = 'registration_view.html'
  30
  31     def post (self, request):
  32         return self.get_or_post (request, 'POST')
  33
  34     def get (self, request):
  35         return self.get_or_post (request, 'GET')
  36
  37     def get_or_post(self, wsgi_request, method):
  38         """
  39         """
  40         errors = []
  41
  42         authorities_query = Query.get('authority').select('name', 'authority_hrn')
  43         authorities = execute_admin_query(wsgi_request, authorities_query)
  44         if authorities is not None:
  45             authorities = sorted(authorities)
  46
  47         # Page rendering
  48         page = Page(wsgi_request)
  49         page.add_js_files  ( [ "js/jquery.validate.js", "js/my_account.register.js" ] )
  50         page.add_css_files ( [ "css/onelab.css", "css/registration.css" ] )
  51         page.add_css_files ( [ "http://code.jquery.com/ui/1.10.3/themes/smoothness/jquery-ui.css" ] )
  52
  53         if method == 'POST':
  54             # The form has been submitted
  55
  56             # get the domain url
  57             current_site = Site.objects.get_current()
  58             current_site = current_site.domain
  59
  60             user_request = {
  61                 'first_name'    : wsgi_request.POST.get('firstname',     ''),
  62                 'last_name'     : wsgi_request.POST.get('lastname',      ''),
  63                 'authority_hrn' : wsgi_request.POST.get('authority_hrn', ''),
  64                 'email'         : wsgi_request.POST.get('email',         '').lower(),
  65                 'password'      : wsgi_request.POST.get('password',      ''),
  66                 'current_site'  : current_site
  67             }
  68
  69             # Construct user_hrn from email (XXX Should use common code)
  70             split_email = user_request['email'].split("@")[0]
  71             split_email = split_email.replace(".", "_")
  72             user_request['user_hrn'] = user_request['authority_hrn'] \
  73                      + '.' + split_email
  74
  75             # Validate input
  76             UserModel = get_user_model()
  77             if (re.search(r'^[\w+\s.@+-]+$', user_request['first_name']) == None):
  78                 errors.append('First Name may contain only letters, numbers, spaces and @/./+/-/_ characters.')
  79             if (re.search(r'^[\w+\s.@+-]+$', user_request['last_name']) == None):
  80                 errors.append('Last Name may contain only letters, numbers, spaces and @/./+/-/_ characters.')
  81             # checking in django_db !!
  82             if PendingUser.objects.filter(email__iexact = user_request['email']):
  83                 errors.append('Email is pending for validation. Please provide a new email address.')
  84             if UserModel._default_manager.filter(email__iexact = user_request['email']):
  85                 errors.append('This email is not usable. Please contact the administrator or try with another email.')
  86             # Does the user exist in Manifold?
  87             user_query  = Query().get('local:user').select('user_id','email')
  88             user_details = execute_admin_query(wsgi_request, user_query)
  89             for user_detail in user_details:
  90                 if user_detail['email'] == user_request['email']:
  91                     errors.append('Email already registered in Manifold. Please provide a new email address.')
  92             # Does the user exist in sfa? [query is very slow!!]
  93             #user_query  = Query().get('user').select('user_hrn','user_email')
  94             # XXX Test based on the user_hrn is quick
  95             user_query  = Query().get('user').select('user_hrn','user_email').filter_by('user_hrn','==',user_request['user_hrn'])
  96             user_details_sfa = execute_admin_query(wsgi_request, user_query)
  97
  98             for user in user_details_sfa:
  99                 if user['user_email'] == user_request['email']:
 100                     errors.append('Email already registered in SFA registry. Please use another email.')
 101                 if user['user_hrn'] == user_request['user_hrn']:
 102                     # add random number if user_hrn already exists in the registry
 103                     user_request['user_hrn'] = user_request['authority_hrn'] \
 104                             + '.' + split_email + str(randint(1,1000000))
 105
 106             # XXX TODO: Factorize with portal/accountview.py
 107             if 'generate' in wsgi_request.POST['question']:
 108                 user_request['auth_type'] = 'managed'
 109
 110                 # XXX Common code, dependency ?
 111                 from Crypto.PublicKey import RSA
 112                 private = RSA.generate(1024)
 113
 114                 # Example: private_key = '-----BEGIN RSA PRIVATE KEY-----\nMIIC...'
 115                 # Example: public_key = 'ssh-rsa AAAAB3...'
 116                 user_request['private_key'] = private.exportKey()
 117                 user_request['public_key']  = private.publickey().exportKey(format='OpenSSH')
 118
 119             else:
 120                 user_request['auth_type'] = 'user'
 121
 122                 up_file        = wsgi_request.FILES['user_public_key']
 123
 124                 file_content   = up_file.read().strip()
 125                 file_name      = up_file.name
 126                 file_extension = os.path.splitext(file_name)[1]
 127
 128                 ALLOWED_EXTENSIONS =  ['.pub','.txt']
 129                 if file_extension not in ALLOWED_EXTENSIONS or not re.search(r'ssh-rsa',file_content):
 130                     errors.append('Please upload a valid RSA public key.')
 131
 132                 user_request['private_key'] = None
 133                 user_request['public_key']  = file_content
 134
 135             if not errors:
 136                 try:
 137                     # verify if is a  LDAP
 138                     mail = user_detail['email']
 139                     login = mail.split('@')[0]
 140                     org = mail.split('@')[1]
 141                     o = org.split('.')[-2]
 142                     dc = org.split('.')[-1]
 143                     # To know if user is a LDAP user - Need to has a 'dc' identifier
 144                     if dc == 'br' or 'eu':
 145                         # LDAP insert directly - but with userEnable = FALSE
 146                         ldap_create_user(wsgi_request, user_request, user_detail)
 147                 except Exception, e:
 148                     print "LDAP: problem em access the LDAP with this credentail"
 149                 create_pending_user(wsgi_request, user_request, user_detail)
 150                 self.template_name = 'user_register_complete.html'
 151
 152                 return render(wsgi_request, self.template, {'theme': self.theme})
 153
 154         else:
 155             user_request = {}
 156
 157         template_env = {
 158           'topmenu_items': topmenu_items_live('Register', page),
 159           'errors': errors,
 160           'authorities': authorities,
 161           'theme': self.theme
 162           }
 163         template_env.update(user_request)
 164         template_env.update(page.prelude_env ())
 165         return render(wsgi_request, self.template,template_env)