3 from random import randint
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
11 from unfold.page import Page
12 from unfold.loginrequired import FreeAccessView
13 from ui.topmenu import topmenu_items_live
15 from manifoldapi.manifoldapi import execute_admin_query
16 from manifold.core.query import Query
18 from portal.models import PendingUser
19 #from portal.actions import create_pending_user
21 from portal.actions import create_pending_user, ldap_create_user
23 from myslice.theme import ThemeView
25 # since we inherit from FreeAccessView we cannot redefine 'dispatch'
26 # so let's override 'get' and 'post' instead
28 class RegistrationView (FreeAccessView, ThemeView):
29 template_name = 'registration_view.html'
31 def post (self, request):
32 return self.get_or_post (request, 'POST')
34 def get (self, request):
35 return self.get_or_post (request, 'GET')
37 def get_or_post(self, wsgi_request, method):
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)
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" ] )
54 # The form has been submitted
57 current_site = Site.objects.get_current()
58 current_site = current_site.domain
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
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'] \
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)
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))
106 # XXX TODO: Factorize with portal/accountview.py
107 if 'generate' in wsgi_request.POST['question']:
108 user_request['auth_type'] = 'managed'
110 # XXX Common code, dependency ?
111 from Crypto.PublicKey import RSA
112 private = RSA.generate(1024)
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')
120 user_request['auth_type'] = 'user'
122 up_file = wsgi_request.FILES['user_public_key']
124 file_content = up_file.read().strip()
125 file_name = up_file.name
126 file_extension = os.path.splitext(file_name)[1]
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.')
132 user_request['private_key'] = None
133 user_request['public_key'] = file_content
138 # verify if is a LDAP
139 mail = user_detail['email']
140 login = mail.split('@')[0]
141 org = mail.split('@')[1]
142 o = org.split('.')[-2]
143 dc = org.split('.')[-1]
144 # To know if user is a LDAP user - Need to has a 'dc' identifier
145 if dc == 'br' or 'eu':
146 # LDAP insert directly - but with userEnable = FALSE
147 ldap_create_user(wsgi_request, user_request, user_detail)
150 print "LDAP: problem em access the LDAP with this credentail"
152 create_pending_user(wsgi_request, user_request, user_detail)
153 self.template_name = 'user_register_complete.html'
155 return render(wsgi_request, self.template, {'theme': self.theme})
161 'topmenu_items': topmenu_items_live('Register', page),
163 'authorities': authorities,
166 template_env.update(user_request)
167 template_env.update(page.prelude_env ())
168 return render(wsgi_request, self.template,template_env)