# -*- coding: utf-8 -*-
# This file is part of Shuup.
#
# Copyright (c) 2012-2017, Shoop Commerce Ltd. All rights reserved.
#
# This source code is licensed under the OSL-3.0 license found in the
# LICENSE file in the root directory of this source tree.
from django.conf import settings
from django.contrib import messages
from django.contrib.auth import REDIRECT_FIELD_NAME
from django.http import HttpResponseNotFound
from django.shortcuts import redirect
from django.utils.http import is_safe_url
from django.utils.translation import ugettext_lazy as _
from django.views.generic import View
from registration.backends.default import views as default_views
from registration.backends.simple import views as simple_views
from shuup import configuration
from shuup.front.apps.registration.forms import CompanyRegistrationForm
from shuup.front.template_helpers import urls
[docs]def activation_complete(request):
messages.success(request, _("Activation successful!"))
if urls.has_url('shuup:customer_edit'):
return redirect('shuup:customer_edit')
else:
return redirect(settings.LOGIN_REDIRECT_URL)
[docs]def registration_complete(request):
if settings.SHUUP_REGISTRATION_REQUIRES_ACTIVATION:
messages.success(
request, _("Registration complete. Please follow the instructions sent to your email address."))
return redirect(settings.LOGIN_REDIRECT_URL)
[docs]class RegistrationViewMixin(object):
template_name = "shuup/registration/register.jinja"
[docs] def get_success_url(self, *args, **kwargs):
url = self.request.POST.get(REDIRECT_FIELD_NAME)
if url and is_safe_url(url, self.request.get_host()):
return url
return ('shuup:registration_complete', (), {})
[docs]class RegistrationNoActivationView(RegistrationViewMixin, simple_views.RegistrationView):
pass
[docs]class RegistrationWithActivationView(RegistrationViewMixin, default_views.RegistrationView):
SEND_ACTIVATION_EMAIL = False
[docs]class RegistrationView(View):
[docs] def dispatch(self, request, *args, **kwargs):
if settings.SHUUP_REGISTRATION_REQUIRES_ACTIVATION:
view_class = RegistrationWithActivationView
else:
view_class = RegistrationNoActivationView
return view_class.as_view()(request, *args, **kwargs)
[docs]class CompanyRegistrationView(RegistrationViewMixin, default_views.RegistrationView):
template_name = "shuup/registration/company_register.jinja"
form_class = CompanyRegistrationForm
SEND_ACTIVATION_EMAIL = False
[docs] def dispatch(self, request, *args, **kwargs):
if not configuration.get(None, "allow_company_registration"):
return HttpResponseNotFound()
return super(CompanyRegistrationView, self).dispatch(request, *args, **kwargs)
[docs]class ActivationView(default_views.ActivationView):
template_name = "shuup/registration/activation_failed.jinja"
[docs] def get_success_url(self, *args, **kwargs):
return ('shuup:registration_activation_complete', (), {})