portal/registrationview.py

   1 import os.path
   2 import re
   3 import json
   4 from random     import randint
   5 from hashlib    import md5
   6
   7 from django.views.generic       import View
   8 from django.template.loader     import render_to_string
   9
  10 from django.shortcuts           import render
  11 from django.contrib.auth        import get_user_model
  12 from django.contrib.sites.models import Site
  13
  14 from unfold.page                import Page
  15 from unfold.loginrequired       import FreeAccessView
  16 from ui.topmenu                 import topmenu_items_live
  17
  18 from manifoldapi.manifoldapi    import execute_admin_query
  19 from manifold.core.query        import Query
  20
  21 from portal.models              import PendingUser
  22 from portal.actions             import create_pending_user
  23
  24 from myslice.theme import ThemeView
  25 from myslice.settings import logger
  26
  27 import activity.user
  28
  29 # since we inherit from FreeAccessView we cannot redefine 'dispatch'
  30 # so let's override 'get' and 'post' instead
  31 #
  32 class RegistrationView (FreeAccessView, ThemeView):
  33     template_name = 'registration_view.html'
  34
  35     def post (self, request):
  36         return self.get_or_post (request, 'POST')
  37
  38     def get (self, request):
  39         return self.get_or_post (request, 'GET')
  40
  41     def get_or_post(self, wsgi_request, method):
  42         """
  43         """
  44         errors = []
  45         authority_hrn = None
  46         # REGISTRY ONLY TO BE REMOVED WITH MANIFOLD-V2
  47         authorities_query = Query.get('authority').select('name', 'authority_hrn')
  48         authorities = execute_admin_query(wsgi_request, authorities_query)
  49         logger.info("RegistrationView authorities = {}".format(authorities))
  50         if authorities is not None:
  51             # Remove the root authority from the list
  52             matching = [s for s in authorities if "." in s['authority_hrn']]
  53             authorities = sorted(matching, key=lambda k: k['authority_hrn'])
  54             authorities = sorted(matching, key=lambda k: k['name'])
  55
  56         logger.debug("############ BREAKPOINT 1 #################")
  57         # Page rendering
  58         page = Page(wsgi_request)
  59
  60         page.add_css_files ( [ "https://code.jquery.com/ui/1.11.2/themes/smoothness/jquery-ui.css" ] )
  61
  62         page.add_js_files  ( [ "js/jquery.validate.js", "js/my_account.register.js", "js/jquery.qtip.min.js","js/jquery-ui.js","js/jquery-ui-combobox.js" ] )
  63
  64         page.add_css_files ( [ "css/onelab.css", "css/registration.css", "css/jquery.qtip.min.css", "css/jquery.ui.combobox.css" ] )
  65         page.expose_js_metadata()
  66         logger.debug("############ BREAKPOINT 2 #################")
  67         if method == 'POST':
  68             reg_form = {}
  69             # The form has been submitted
  70
  71             # get the domain url
  72             current_site = Site.objects.get_current()
  73             current_site = current_site.domain
  74
  75             logger.debug("############ BREAKPOINT 3 #################")
  76             post_email = wsgi_request.POST.get('email','').lower()
  77             salt = randint(1,100000)
  78             email_hash = md5(str(salt)+post_email).hexdigest()
  79             #email_hash = md5(post_email).digest().encode('base64')[:-1]
  80             user_request = {
  81                 'first_name'    : wsgi_request.POST.get('firstname',     ''),
  82                 'last_name'     : wsgi_request.POST.get('lastname',      ''),
  83                 'authority_hrn' : wsgi_request.POST.get('org_name', ''),
  84                 'email'         : post_email,
  85                 'password'      : wsgi_request.POST.get('password',      ''),
  86                 'current_site'  : current_site,
  87                 'email_hash'    : email_hash,
  88                 'pi'            : '',
  89                 'validation_link': current_site + '/portal/email_activation/'+ email_hash
  90             }
  91
  92             logger.debug("############ BREAKPOINT 4 #################")
  93             auth = wsgi_request.POST.get('org_name', None)
  94             if auth is None or auth == "":
  95                 errors.append('Organization required: please select one or request its addition')
  96             else:
  97
  98                 logger.debug("############ BREAKPOINT 5 #################")
  99
 100                 # Construct user_hrn from email (XXX Should use common code)
 101                 split_email = user_request['email'].split("@")[0]
 102                 split_email = split_email.replace(".", "_")
 103                 # Replace + by _ => more convenient for testing and validate with a real email
 104                 split_email = split_email.replace("+", "_")
 105                 user_request['user_hrn'] = user_request['authority_hrn'] \
 106                          + '.' + split_email
 107
 108                 # Validate input
 109                 UserModel = get_user_model()
 110                 if (re.search(r'^[\w+\s.@+-]+$', user_request['first_name']) == None):
 111                     errors.append('First name may contain only letters, numbers, spaces and @/./+/-/_ characters.')
 112                 if (re.search(r'^[\w+\s.@+-]+$', user_request['last_name']) == None):
 113                     errors.append('Last name may contain only letters, numbers, spaces and @/./+/-/_ characters.')
 114                 # Does the user exist in Manifold?
 115                 user_query  = Query().get('local:user').select('user_id','email')
 116                 user_details = execute_admin_query(wsgi_request, user_query)
 117                 for user_detail in user_details:
 118                     if user_detail['email'] == user_request['email']:
 119                         errors.append('Email already registered. <a href="/">Login</a> with your existing account. <a href="/portal/pass_reset/">Forgot your password?</a>')
 120
 121                 # Does the user exist in sfa? [query is very slow!!]
 122                 #user_query  = Query().get('user').select('user_hrn','user_email')
 123                 # XXX Test based on the user_hrn is quick
 124
 125                 # REGISTRY ONLY TO BE REMOVED WITH MANIFOLD-V2
 126                 user_query  = Query().get('myslice:user').select('user_hrn','user_email').filter_by('user_hrn','==',user_request['user_hrn'])
 127                 user_details_sfa = execute_admin_query(wsgi_request, user_query)
 128
 129                 for user in user_details_sfa:
 130                     if user['user_email'] == user_request['email']:
 131                         errors.append('Email already registered in OneLab registry. <a href="/contact">Contact OneLab support</a> or use another email.')
 132                     if user['user_hrn'] == user_request['user_hrn']:
 133                         # add random number if user_hrn already exists in the registry
 134                         user_request['user_hrn'] = user_request['authority_hrn'] \
 135                                 + '.' + split_email + str(randint(1,1000000))
 136
 137                 # checking in django unfold db portal application pending users
 138                 # sqlite3 /var/unfold/unfold.sqlite3
 139                 # select email from portal_pendinguser;
 140                 if PendingUser.objects.filter(email__iexact = user_request['email']):
 141                     errors.append('Account pending for validation. Please wait till your account is validated or contact OneLab support.')
 142
 143                 # checking in django_db !!
 144                 # sqlite3 /var/unfold/unfold.sqlite3
 145                 # select email from auth_user;
 146                 if UserModel._default_manager.filter(email__iexact = user_request['email']):
 147                     errors.append('<a href="/contact">Contact support</a> or try with another email.')
 148
 149                 # XXX TODO: Factorize with portal/accountview.py
 150                 # XXX TODO: Factorize with portal/registrationview.py
 151                 # XXX TODO: Factorize with portal/joinview.py
 152                 if 'generate' in wsgi_request.POST['question']:
 153                     user_request['auth_type'] = 'managed'
 154
 155                     # XXX Common code, dependency ?
 156                     from Crypto.PublicKey import RSA
 157                     private = RSA.generate(1024)
 158
 159                     # Example: private_key = '-----BEGIN RSA PRIVATE KEY-----\nMIIC...'
 160                     # Example: public_key = 'ssh-rsa AAAAB3...'
 161                     user_request['private_key'] = private.exportKey()
 162                     user_request['public_key']  = private.publickey().exportKey(format='OpenSSH')
 163
 164                 else:
 165                     user_request['auth_type'] = 'user'
 166
 167                     up_file        = wsgi_request.FILES['user_public_key']
 168
 169                     file_content   = up_file.read().strip()
 170                     file_name      = up_file.name
 171                     file_extension = os.path.splitext(file_name)[1]
 172
 173                     ALLOWED_EXTENSIONS =  ['.pub','.txt']
 174                     if file_extension not in ALLOWED_EXTENSIONS or not re.search(r'ssh-rsa',file_content):
 175                         errors.append('Please upload a valid RSA public key.')
 176                     # user_request['private_key'] can't be Null because all db fields are set as NOT NULL
 177                     user_request['private_key'] = ""
 178                     user_request['public_key']  = file_content
 179
 180                 if not errors:
 181                     create_pending_user(wsgi_request, user_request, user_detail)
 182                     self.template_name = 'user_register_complete.html'
 183                     # log user activity
 184                     activity.user.registered(self.request)
 185                     return render(wsgi_request, self.template, {'theme': self.theme})
 186
 187         else:
 188             logger.debug("############ BREAKPOINT A #################")
 189             user_request = {}
 190             ## this is coming from onelab website onelab.eu
 191             reg_form = {
 192                 'first_name':  wsgi_request.GET.get('first_name', ''),
 193                 'last_name': wsgi_request.GET.get('last_name', ''),
 194                 'email': wsgi_request.GET.get('email', ''),
 195                 }
 196             # log user activity
 197             activity.user.signup(self.request)
 198             logger.debug("############ BREAKPOINT B #################")
 199
 200         template_env = {
 201           #'topmenu_items': topmenu_items_live('Register', page),
 202           'errors': errors,
 203           'authorities': authorities,
 204           'theme': self.theme
 205           }
 206         template_env.update(user_request)
 207         template_env.update(reg_form)
 208         template_env.update(page.prelude_env ())
 209         logger.debug("############ BREAKPOINT C #################")
 210         return render(wsgi_request, self.template,template_env)