3 from random import randint
4 from hashlib import md5
6 from django.views.generic import View
7 from django.template.loader import render_to_string
8 from django.shortcuts import render
9 from django.contrib.auth import get_user_model
10 from django.contrib.sites.models import Site
12 from unfold.page import Page
13 from unfold.loginrequired import FreeAccessView
14 from ui.topmenu import topmenu_items_live
16 from manifoldapi.manifoldapi import execute_admin_query
17 from manifold.core.query import Query
19 from portal.models import PendingUser
20 from django.contrib.auth.models import User #Pedro
21 #from portal.actions import create_pending_user
23 from portal.actions import create_pending_user, ldap_create_user
25 from myslice.theme import ThemeView
27 # since we inherit from FreeAccessView we cannot redefine 'dispatch'
28 # so let's override 'get' and 'post' instead
30 class RegistrationView (FreeAccessView, ThemeView):
31 template_name = 'registration_view.html'
33 def post (self, request):
34 return self.get_or_post (request, 'POST')
36 def get (self, request):
37 return self.get_or_post (request, 'GET')
39 def get_or_post(self, wsgi_request, method):
44 authorities_query = Query.get('authority').select('name','authority_hrn')
45 authorities = execute_admin_query(wsgi_request, authorities_query)
46 if authorities is not None:
47 authorities = sorted(authorities)
50 page = Page(wsgi_request)
51 page.add_js_files ( [ "js/jquery.validate.js", "js/my_account.register.js", "js/jquery.qtip.min.js","js/jquery-ui.js" ] )
52 page.add_css_files ( [ "css/onelab.css", "css/registration.css", "css/jquery.qtip.min.css" ] )
53 page.add_css_files ( [ "https://code.jquery.com/ui/1.10.3/themes/smoothness/jquery-ui.css" ] )
57 # The form has been submitted
60 current_site = Site.objects.get_current()
61 current_site = current_site.domain
63 #authorities_query = Query.get('authority').select('name', 'authority_hrn')
64 #authorities = execute_admin_query(wsgi_request, authorities_query)
66 for authority in authorities:
67 if authority['name'] == wsgi_request.POST.get('org_name', ''):
68 authority_hrn = authority['authority_hrn']
70 # Handle the case when the template uses only hrn and not name
71 if authority_hrn is None:
72 authority_hrn = wsgi_request.POST.get('org_name', '')
74 post_email = wsgi_request.POST.get('email','').lower()
75 salt = randint(1,100000)
76 email_hash = md5(str(salt)+post_email).hexdigest()
77 #email_hash = md5(post_email).digest().encode('base64')[:-1]
79 'first_name' : wsgi_request.POST.get('firstname', ''),
80 'last_name' : wsgi_request.POST.get('lastname', ''),
81 'organization' : wsgi_request.POST.get('org_name', ''),
82 'authority_hrn' : authority_hrn,
84 'username' : wsgi_request.POST.get('username','').lower(),
85 'password' : wsgi_request.POST.get('password', ''),
86 'current_site' : current_site,
87 'email_hash' : email_hash,
89 'validation_link': 'https://' + current_site + '/portal/email_activation/'+ email_hash
92 # Construct user_hrn from email (XXX Should use common code)
93 # split_email = user_request['email'].split("@")[0]
94 # split_email = split_email.replace(".", "_")
95 # user_request['user_hrn'] = user_request['authority_hrn'] \
98 username = user_request['username']
100 if user_request['authority_hrn'] == "fibre" :
101 user_request['username'] = user_request['username'] + "@" + "" # to be defined
103 split_authority = user_request['authority_hrn'].split(".")[1]
104 user_request['username'] = user_request['username'] + '@' + split_authority
105 split_authority = user_request['authority_hrn'].split(".")[0]
107 user_request['user_hrn'] = split_authority + '.' + user_request['username']
110 UserModel = get_user_model()
111 if (re.search(r'^[\w+\s.@+-]+$', user_request['first_name']) == None):
112 errors.append('First name may contain only letters, numbers, spaces and @/./+/-/_ characters.')
113 if (re.search(r'^[\w+\s.@+-]+$', user_request['last_name']) == None):
114 errors.append('Last name may contain only letters, numbers, spaces and @/./+/-/_ characters.')
115 if (re.search(r'^[\w,]+$' , username) == None):
116 errors.append('Username may contain only letters,numbers and -/_ characters.')
117 # checking in django_db !!
118 if PendingUser.objects.filter(email__iexact = user_request['email']):
119 errors.append('Email is pending for validation. Please provide a new email address.')
120 # if UserModel._default_manager.filter(email__iexact = user_request['email']):
121 # errors.append('This email is not usable. Please contact the administrator or try with another email.')
122 if User.objects.filter(username__iexact = user_request['username']):
123 errors.append('This username is already in use, try another one')
124 # Does the user exist in Manifold?
125 user_query = Query().get('local:user').select('user_id','email')
126 user_details = execute_admin_query(wsgi_request, user_query)
127 for user_detail in user_details:
128 if user_detail['email'] == user_request['email']:
129 errors.append('Email already registered in Manifold. Please provide a new email address.')
130 # Does the user exist in sfa? [query is very slow!!]
131 #user_query = Query().get('user').select('user_hrn','user_email')
132 # XXX Test based on the user_hrn is quick
133 user_query = Query().get('user').select('user_hrn','user_email').filter_by('user_hrn','==',user_request['user_hrn'])
134 user_details_sfa = execute_admin_query(wsgi_request, user_query)
136 # for user in user_details_sfa:
137 # if user['user_email'] == user_request['email']:
138 # errors.append('Email already registered in SFA registry. Please use another email.')
139 # if user['user_hrn'] == user_request['user_hrn']:
140 # # add random number if user_hrn already exists in the registry
141 # user_request['user_hrn'] = user_request['authority_hrn'] \
142 # + '.' + split_email + str(randint(1,1000000))
144 # XXX TODO: Factorize with portal/accountview.py
145 # XXX TODO: Factorize with portal/registrationview.py
146 # XXX TODO: Factorize with portal/joinview.py
147 if 'generate' in wsgi_request.POST['question']:
148 user_request['auth_type'] = 'managed'
150 # XXX Common code, dependency ?
151 from Crypto.PublicKey import RSA
152 private = RSA.generate(1024)
154 # Example: private_key = '-----BEGIN RSA PRIVATE KEY-----\nMIIC...'
155 # Example: public_key = 'ssh-rsa AAAAB3...'
156 user_request['private_key'] = private.exportKey()
157 user_request['public_key'] = private.publickey().exportKey(format='OpenSSH')
160 user_request['auth_type'] = 'user'
162 up_file = wsgi_request.FILES['user_public_key']
164 file_content = up_file.read().strip()
165 file_name = up_file.name
166 file_extension = os.path.splitext(file_name)[1]
168 ALLOWED_EXTENSIONS = ['.pub','.txt']
169 if file_extension not in ALLOWED_EXTENSIONS or not re.search(r'ssh-rsa',file_content):
170 errors.append('Please upload a valid RSA public key.')
171 # user_request['private_key'] can't be Null because all db fields are set as NOT NULL
172 user_request['private_key'] = ""
173 user_request['public_key'] = file_content
178 # verify if is a LDAP
179 mail = user_detail['email']
180 login = mail.split('@')[0]
181 org = mail.split('@')[1]
182 o = org.split('.')[-2]
183 dc = org.split('.')[-1]
184 # To know if user is a LDAP user - Need to has a 'dc' identifier
185 if dc == 'br' or 'eu':
186 # LDAP insert directly - but with userEnable = FALSE
187 ldap_create_user(wsgi_request, user_request, user_detail)
190 print "LDAP: problem em access the LDAP with this credentail"
192 create_pending_user(wsgi_request, user_request, user_detail)
193 self.template_name = 'user_register_complete.html'
195 return render(wsgi_request, self.template, {'theme': self.theme})
199 ## this is coming from onelab website onelab.eu
201 'first_name': wsgi_request.GET.get('first_name', ''),
202 'last_name': wsgi_request.GET.get('last_name', ''),
203 'email': wsgi_request.GET.get('email', ''),
207 'topmenu_items': topmenu_items_live('Register', page),
209 'authorities': authorities,
212 template_env.update(user_request)
213 template_env.update(reg_form)
214 template_env.update(page.prelude_env ())
215 return render(wsgi_request, self.template,template_env)