Applying deny register config

Zambom
1 parent c3996a51
Showing 2 changed files with 16 additions and 4 deletions Show diff stats
users/templates/users/login.html
users/views.py
@@ -59,9 +59,11 @@
                             <div class="col-md-6 col-xs-6 col-sm-6 col-lg-6 text-center">
                                 <button type="submite" class="btn btn-success btn-raised btn-block" form="form-login" style="position: initial;"> {% trans 'Log in' %} </button>
                             </div>
-                            <div class="col-md-6 col-xs-6 col-sm-6 col-lg-6 text-center">
-                                <a class="btn btn-default btn-raised btn-block" href="{% url 'users:signup' %}" formaction="#" style="position: initial;">{% trans 'Sign Up' %}</a>
-                            </div>
+                            {% if not deny_register %}
+                                <div class="col-md-6 col-xs-6 col-sm-6 col-lg-6 text-center">
+                                    <a class="btn btn-default btn-raised btn-block" href="{% url 'users:signup' %}" formaction="#" style="position: initial;">{% trans 'Sign Up' %}</a>
+                                </div>
+                            {% endif %}
                         </div>
                     </div>
                     <div class="row">
@@ -304,6 +304,14 @@ class RegisterUser(generic.edit.CreateView):
 		return super(RegisterUser, self).form_valid(form)
+	def dispatch(self, request, *args, **kwargs):
+		security = Security.objects.get(id = 1)
+
+		if security.allow_register:
+			return redirect(reverse_lazy('users:login'))
+
+		return super(RegisterUser, self).dispatch(request, *args, **kwargs)
+
 class ForgotPassword(generic.FormView):
 	template_name = "users/forgot_password.html"
 	success_url = reverse_lazy('users:login')
@@ -401,12 +409,14 @@ class PasswordResetConfirmView(generic.FormView):
 def login(request):
 	context = {}
 	context['title'] = _('Log In')
+	security = Security.objects.get(id = 1)
+
+	context['deny_register'] = security.allow_register
 	if request.POST:
 		username = request.POST['email']
 		password = request.POST['password']
 		user = authenticate(username=username, password=password)
-		security = Security.objects.get(id = 1)
 		if user is not None:
 			if not security.maintence or user.is_staff: