Skip to content
auth.py 51 KiB
Newer Older
from __future__ import annotations
Andreas Klöckner's avatar
Andreas Klöckner committed

__copyright__ = "Copyright (C) 2014 Andreas Kloeckner"

__license__ = """
Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the "Software"), to deal
in the Software without restriction, including without limitation the rights
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, and to permit persons to whom the Software is
furnished to do so, subject to the following conditions:

The above copyright notice and this permission notice shall be included in
all copies or substantial portions of the Software.

THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
THE SOFTWARE.
"""
import re
from typing import (
    TYPE_CHECKING,
    Any,
    cast,
Andreas Klöckner's avatar
Andreas Klöckner committed
)

import django.forms as forms
from crispy_forms.helper import FormHelper
Andreas Klöckner's avatar
Andreas Klöckner committed
from crispy_forms.layout import Button, Div, Layout, Submit
from django import http
from django.conf import settings
Andreas Klöckner's avatar
Andreas Klöckner committed
from django.contrib import messages
from django.contrib.auth import (
    REDIRECT_FIELD_NAME,
    get_user_model,
    login as auth_login,
    logout as auth_logout,
Andreas Klöckner's avatar
Andreas Klöckner committed
)
from django.contrib.auth.decorators import login_required, user_passes_test
from django.contrib.auth.forms import AuthenticationForm as AuthenticationFormBase
Dong Zhuang's avatar
Dong Zhuang committed
from django.contrib.auth.validators import ASCIIUsernameValidator
Andreas Klöckner's avatar
Andreas Klöckner committed
from django.core.exceptions import (
    MultipleObjectsReturned,
    ObjectDoesNotExist,
    PermissionDenied,
Andreas Klöckner's avatar
Andreas Klöckner committed
    SuspiciousOperation,
)
from django.http import HttpResponseRedirect
from django.shortcuts import get_object_or_404, redirect, render, resolve_url
from django.template.response import TemplateResponse
Andreas Klöckner's avatar
Andreas Klöckner committed
from django.urls import reverse
from django.utils.http import url_has_allowed_host_and_scheme
from django.utils.translation import gettext_lazy as _
from django.views.decorators.cache import never_cache
from django.views.decorators.csrf import csrf_protect
Andreas Klöckner's avatar
Andreas Klöckner committed
from django.views.decorators.debug import sensitive_post_parameters
from django_select2.forms import ModelSelect2Widget
from djangosaml2.backends import Saml2Backend
from accounts.models import User
Andreas Klöckner's avatar
Andreas Klöckner committed
from course.constants import (
    participation_permission as pperm,
    participation_status,
    user_status,
Andreas Klöckner's avatar
Andreas Klöckner committed
)
from course.models import (
    AuthenticationToken,
    Participation,
    ParticipationRole,
Andreas Klöckner's avatar
Andreas Klöckner committed
)
from course.utils import CoursePageContext, course_view, render_course_page
Andreas Klöckner's avatar
Andreas Klöckner committed
from relate.utils import (
    HTML5DateTimeInput,
    StyledForm,
    StyledModelForm,
    get_site_name,
    string_concat,
Andreas Klöckner's avatar
Andreas Klöckner committed
)
    import datetime
Andreas Klöckner's avatar
Andreas Klöckner committed

    from django.db.models import query
Dong Zhuang's avatar
Dong Zhuang committed

Isuru Fernando's avatar
Isuru Fernando committed

def get_pre_impersonation_user(request):
    is_impersonating = hasattr(
            request, "relate_impersonate_original_user")
    if is_impersonating:
        return request.relate_impersonate_original_user
    return None


def get_impersonable_user_qset(impersonator: User) -> query.QuerySet:
    if impersonator.is_superuser:
        return User.objects.exclude(pk=impersonator.pk)
    my_participations = Participation.objects.filter(
    impersonable_user_qset = User.objects.none()
        # Notice: if a TA is not allowed to view participants'
        # profile in one course, then he/she is not able to impersonate
        # any user, even in courses he/she is allow to view profiles
        # of all users.
        if part.has_permission(pperm.view_participant_masked_profile):
            return User.objects.none()
Andreas Klöckner's avatar
Andreas Klöckner committed
        impersonable_roles = [
            argument
            for perm, argument in part.permissions()
            if perm == pperm.impersonate_role]
        q = (Participation.objects
             .filter(course=part.course,
                     status=participation_status.active,
                     roles__identifier__in=impersonable_roles)
             .select_related("user"))
        # There can be duplicate records. Removing duplicate records is needed
        # only when rendering ImpersonateForm
        impersonable_user_qset = (
            impersonable_user_qset
Andreas Klöckner's avatar
Andreas Klöckner committed
            | User.objects.filter(pk__in=q.values_list("user__pk", flat=True))
    return impersonable_user_qset
    def __init__(self, get_response):
        self.get_response = get_response

    def __call__(self, request):
        if "impersonate_id" in request.session:
            imp_id = request.session["impersonate_id"]
            impersonee = None

            try:
                if imp_id is not None:
                    impersonee = cast(User, get_user_model().objects.get(id=imp_id))
            except ObjectDoesNotExist:
                pass

            may_impersonate = False
            if impersonee is not None:
                if request.user.is_superuser:
                    may_impersonate = True
                    qset = get_impersonable_user_qset(cast(User, request.user))
                    if qset.filter(pk=cast(User, impersonee).pk).count():
                        may_impersonate = True
            if may_impersonate:
                request.relate_impersonate_original_user = request.user
                request.user = impersonee
            else:
                messages.add_message(request, messages.ERROR,
                        _("Error while impersonating."))

        return self.get_response(request)

class UserSearchWidget(ModelSelect2Widget):
    model = User
    search_fields = [
            "username__icontains",
            "email__icontains",
            "first_name__icontains",
            "last_name__icontains",
            ]

    def label_from_instance(self, u):
        if u.first_name and u.last_name:
            return (
                    f"{u.get_full_name()} ({u.username} - {u.email})")
        else:
            # for users with "None" fullname
            return (
                    f"{u.username} ({u.email})")
class ImpersonateForm(StyledForm):
    def __init__(self, *args: Any, **kwargs: Any) -> None:
        super().__init__(*args, **kwargs)
        self.fields["user"] = forms.ModelChoiceField(
ifaint's avatar
ifaint committed
                help_text=_("Select user to impersonate."),
                widget=UserSearchWidget(
                    queryset=qset,
                    attrs={"data-minimum-input-length": 0},
                ),
                label=_("User"))
        self.fields["add_impersonation_header"] = forms.BooleanField(
                required=False,
                initial=True,
                label=_("Add impersonation header"),
                help_text=_("Add impersonation header to every page rendered "
                    "while impersonating, as a reminder that impersonation "
                    "is in progress."))

        self.helper.add_input(Submit("submit", _("Impersonate")))
def impersonate(request: http.HttpRequest) -> http.HttpResponse:
    if not request.user.is_authenticated:
        raise PermissionDenied()

Dong Zhuang's avatar
Dong Zhuang committed
    impersonable_user_qset = get_impersonable_user_qset(cast(User, request.user))
    if not impersonable_user_qset.count():
        raise PermissionDenied()
    if hasattr(request, "relate_impersonate_original_user"):
        messages.add_message(request, messages.ERROR,
                _("Already impersonating someone."))
        return redirect("relate-home")
    # Remove duplicate and sort
    # order_by().distinct() directly on impersonable_user_qset will not work
    qset = (User.objects
            .filter(pk__in=impersonable_user_qset.values_list("pk", flat=True))
            .order_by("last_name", "first_name", "username"))
    if request.method == "POST":
        form = ImpersonateForm(request.POST, impersonable_qset=qset)
        if form.is_valid():
            impersonee = form.cleaned_data["user"]

            request.session["impersonate_id"] = impersonee.id
            request.session["relate_impersonation_header"] = form.cleaned_data[
                    "add_impersonation_header"]

            # Because we'll likely no longer have access to this page.
            return redirect("relate-home")
Andreas Klöckner's avatar
Andreas Klöckner committed
    return render(request, "generic-form.html", {
        "form_description": _("Impersonate user"),
def stop_impersonating(request: http.HttpRequest) -> http.JsonResponse:
Andreas Klöckner's avatar
Andreas Klöckner committed
    if request.method != "POST":
        raise PermissionDenied(_("only AJAX POST is allowed"))

Dong Zhuang's avatar
Dong Zhuang committed
    if not request.user.is_authenticated:
        raise PermissionDenied()

    if "stop_impersonating" not in request.POST:
        raise SuspiciousOperation(_("odd POST parameters"))

Andreas Klöckner's avatar
Andreas Klöckner committed
    if not hasattr(request, "relate_impersonate_original_user"):
        # prevent user without pperm to stop_impersonating
        my_participations = Participation.objects.filter(
            user=request.user,
            status=participation_status.active)

        may_impersonate = False
        for part in my_participations:
            perms = [
                perm
                for perm, argument in part.permissions()
                if perm == pperm.impersonate_role]
            if any(perms):
                may_impersonate = True
                break

        if not may_impersonate:
            raise PermissionDenied(_("may not stop impersonating"))
        messages.add_message(request, messages.ERROR,
                _("Not currently impersonating anyone."))
        return http.JsonResponse({})
    del request.session["impersonate_id"]
    messages.add_message(request, messages.INFO,
            _("No longer impersonating anyone."))
    return http.JsonResponse({"result": "success"})


def impersonation_context_processor(request):
    return {
            "currently_impersonating":
Andreas Klöckner's avatar
Andreas Klöckner committed
            hasattr(request, "relate_impersonate_original_user"),
            "add_impersonation_header":
            request.session.get("relate_impersonation_header", True),
def make_sign_in_key(user: User) -> str:
    # Try to ensure these hashes aren't guessable.
    import hashlib
Andreas Klöckner's avatar
Andreas Klöckner committed
    import random
    from time import time
    m = hashlib.sha1()
    m.update(user.email.encode("utf-8"))
    m.update(hex(random.getrandbits(128)).encode())
    m.update(str(time()).encode("utf-8"))
    return m.hexdigest()


def logout_confirmation_required(
        func=None, redirect_field_name=REDIRECT_FIELD_NAME,
        logout_confirmation_url="relate-logout-confirmation"):
    Decorator for views that checks that no user is logged in.
    If a user is currently logged in, redirect him/her to the logout
    confirmation page.
    """
    actual_decorator = user_passes_test(
Dong Zhuang's avatar
Dong Zhuang committed
        lambda u: u.is_anonymous,
        login_url=logout_confirmation_url,
        redirect_field_name=redirect_field_name
    )
    if func:
        return actual_decorator(func)
    return actual_decorator
    def authenticate(self, request, user_id=None, token=None):
Andreas Klöckner's avatar
Andreas Klöckner committed
        users = get_user_model().objects.filter(
                id=user_id, sign_in_key=token)
Andreas Klöckner's avatar
Andreas Klöckner committed
        assert users.count() <= 1
        if users.count() == 0:
Andreas Klöckner's avatar
Andreas Klöckner committed
        (user,) = users
Andreas Klöckner's avatar
Andreas Klöckner committed
        user.status = user_status.active
        user.sign_in_key = None
        user.save()
Andreas Klöckner's avatar
Andreas Klöckner committed
        return user

    def get_user(self, user_id):
        try:
            return get_user_model().objects.get(pk=user_id)
        except get_user_model().DoesNotExist:
@logout_confirmation_required
Dong Zhuang's avatar
Dong Zhuang committed
def sign_in_choice(request, redirect_field_name=REDIRECT_FIELD_NAME):
    redirect_to = request.POST.get(redirect_field_name,
                                   request.GET.get(redirect_field_name, ""))
Andreas Klöckner's avatar
Andreas Klöckner committed
    next_uri = ""
Dong Zhuang's avatar
Dong Zhuang committed
    if redirect_to:
        next_uri = f"?{redirect_field_name}={redirect_to}"
Dong Zhuang's avatar
Dong Zhuang committed

Andreas Klöckner's avatar
Andreas Klöckner committed
    return render(request, "sign-in-choice.html", {
        "next_uri": next_uri,
        "social_provider_to_logo": {
            "google-oauth2": "google",
            },
        "social_provider_to_human_name": {
            "google-oauth2": "Google",
            },
        })
# {{{ conventional login

class LoginForm(AuthenticationFormBase):
    def __init__(self, *args, **kwargs):
        self.helper = FormHelper()
        self.helper.form_tag = False
        self.helper.label_class = "col-lg-2"
        self.helper.field_class = "col-lg-8"

        self.helper.add_input(Submit("submit", _("Sign in")))
        super().__init__(*args, **kwargs)
@sensitive_post_parameters()
@csrf_protect
@never_cache
@logout_confirmation_required
def sign_in_by_user_pw(request, redirect_field_name=REDIRECT_FIELD_NAME):
    """
    Displays the login form and handles the login action.
    """
    if not settings.RELATE_SIGN_IN_BY_USERNAME_ENABLED:
        messages.add_message(request, messages.ERROR,
                _("Username-based sign-in is not being used"))
        return redirect("relate-sign_in_choice")

    redirect_to = request.POST.get(redirect_field_name,
                                   request.GET.get(redirect_field_name, ""))

    if request.method == "POST":
        form = LoginForm(request, data=request.POST)
        if form.is_valid():

            # Ensure the user-originating redirection url is safe.
            if not url_has_allowed_host_and_scheme(
                    allowed_hosts={request.get_host()},
                    require_https=request.is_secure()):
                redirect_to = resolve_url("relate-home")

            user = form.get_user()

            # Okay, security check complete. Log the user in.
            auth_login(request, user)

            return HttpResponseRedirect(redirect_to)
    else:
        form = LoginForm(request)

Dong Zhuang's avatar
Dong Zhuang committed
    next_uri = ""
    if redirect_to:
        next_uri = f"?{redirect_field_name}={redirect_to}"
Dong Zhuang's avatar
Dong Zhuang committed

        redirect_field_name: redirect_to,
    }

    return TemplateResponse(request, "course/login.html", context)

class SignUpForm(StyledModelForm):
    username = forms.CharField(required=True, max_length=30,
            label=_("Username"),
Dong Zhuang's avatar
Dong Zhuang committed
            validators=[ASCIIUsernameValidator()])
        model = get_user_model()
        fields = ("email",)

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.fields["email"].required = True

        self.helper.add_input(
                Submit("submit", _("Send email")))
@logout_confirmation_required
def sign_up(request):
    if not settings.RELATE_REGISTRATION_ENABLED:
                _("self-registration is not enabled"))
    if request.method == "POST":
        form = SignUpForm(request.POST)
        if form.is_valid():
            if get_user_model().objects.filter(
                    username=form.cleaned_data["username"]).count():
                messages.add_message(request, messages.ERROR,
                        _("A user with that username already exists."))

            else:
                email = form.cleaned_data["email"]
                user = get_user_model()(
                        email=email,
                        username=form.cleaned_data["username"])

                user.set_unusable_password()
Andreas Klöckner's avatar
Andreas Klöckner committed
                user.status = user_status.unconfirmed
                user.sign_in_key = make_sign_in_key(user)
                from relate.utils import render_email_template
                message = render_email_template("course/sign-in-email.txt", {
                    "user": user,
                    "sign_in_uri": request.build_absolute_uri(
                        reverse(
                            "relate-reset_password_stage2",
Andreas Klöckner's avatar
Andreas Klöckner committed
                            args=(user.id, user.sign_in_key,))
                        + "?to_profile=1"),
                    "home_uri": request.build_absolute_uri(
                from django.core.mail import EmailMessage
                msg = EmailMessage(
                        string_concat(f"[{_(get_site_name())}] ",
                        getattr(settings, "NO_REPLY_EMAIL_FROM",
                                settings.ROBOT_EMAIL_FROM),
                from relate.utils import get_outbound_mail_connection
                msg.connection = (
                        get_outbound_mail_connection("no_reply")
                        if hasattr(settings, "NO_REPLY_EMAIL_FROM")
                        else get_outbound_mail_connection("robot"))
                msg.send()

                messages.add_message(request, messages.INFO,
                        _("Email sent. Please check your email and click "
                        "the link."))
        else:
            if ("email" in form.errors
                    and "That email address is already in use."
                    in form.errors["email"]):
                messages.add_message(request, messages.ERROR,
                        _("That email address is already in use. "
                        "Would you like to "
                        "<a href='%s'>reset your password</a> instead?")
                        % reverse(
                            "relate-reset_password"))

    else:
        form = SignUpForm()

    return render(request, "generic-form.html", {
ifaint's avatar
ifaint committed
        "form_description": _("Sign up"),
Dong Zhuang's avatar
Dong Zhuang committed
class ResetPasswordFormByEmail(StyledForm):
    email = forms.EmailField(required=True, label=_("Email"),
                             max_length=User._meta.get_field("email").max_length)

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)

        self.helper.add_input(
                Submit("submit", _("Send email")))
Dong Zhuang's avatar
Dong Zhuang committed
class ResetPasswordFormByInstid(StyledForm):
    instid = forms.CharField(max_length=100,
                              required=True,
                              label=_("Institutional ID"))
Dong Zhuang's avatar
Dong Zhuang committed
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
Dong Zhuang's avatar
Dong Zhuang committed

        self.helper.add_input(
                Submit("submit", _("Send email")))


def masked_email(email):
    # return a masked email address
Dong Zhuang's avatar
Dong Zhuang committed
    return email[:2] + "*" * (len(email[3:at])-1) + email[at-1:]


@logout_confirmation_required
Dong Zhuang's avatar
Dong Zhuang committed
def reset_password(request, field="email"):
    if not settings.RELATE_REGISTRATION_ENABLED:
                _("self-registration is not enabled"))
Dong Zhuang's avatar
Dong Zhuang committed
    # return form class by string of class name
    ResetPasswordForm = globals()["ResetPasswordFormBy" + field.title()]  # noqa
    if request.method == "POST":
        form = ResetPasswordForm(request.POST)
        user = None
        if form.is_valid():
            exist_users_with_same_email = False
Dong Zhuang's avatar
Dong Zhuang committed
            if field == "instid":
                inst_id = form.cleaned_data["instid"]
                try:
                    user = get_user_model().objects.get(
                            institutional_id__iexact=inst_id)
                except ObjectDoesNotExist:
Dong Zhuang's avatar
Dong Zhuang committed

            if field == "email":
                email = form.cleaned_data["email"]
                try:
                    user = get_user_model().objects.get(email__iexact=email)
                except ObjectDoesNotExist:
                    pass
                except MultipleObjectsReturned:
                    exist_users_with_same_email = True
            if exist_users_with_same_email:
                # This is for backward compatibility.
                messages.add_message(request, messages.ERROR,
                        _("Failed to send an email: multiple users were "
                          "unexpectedly using that same "
                          "email address. Please "
Dong Zhuang's avatar
Dong Zhuang committed
                          "contact site staff."))
                if user is None:
                    FIELD_DICT = {  # noqa
                            "email": _("email address"),
                            "instid": _("institutional ID")
                            }
Dong Zhuang's avatar
Dong Zhuang committed
                    messages.add_message(request, messages.ERROR,
                            _("That %(field)s doesn't have an "
                                "associated user account. Are you "
                                "sure you've registered?")
                            % {"field": FIELD_DICT[field]})
Dong Zhuang's avatar
Dong Zhuang committed
                else:
                    if not user.email:
                        messages.add_message(request, messages.ERROR,
                                _("The account with that institution ID "
                                    "doesn't have an associated email."))
                    else:
                        email = user.email
                        user.sign_in_key = make_sign_in_key(user)
                        user.save()

                        from relate.utils import render_email_template
                        message = render_email_template(
                            "course/sign-in-email.txt", {
                                "user": user,
                                "sign_in_uri": request.build_absolute_uri(
                                    reverse(
                                        "relate-reset_password_stage2",
                                        args=(user.id, user.sign_in_key,))),
                                "home_uri": request.build_absolute_uri(
                                    reverse("relate-home"))
                            })
                        from django.core.mail import EmailMessage
                        msg = EmailMessage(
                                string_concat(f"[{_(get_site_name())}] ",
                                              _("Password reset")),
                                message,
                                getattr(settings, "NO_REPLY_EMAIL_FROM",
                                        settings.ROBOT_EMAIL_FROM),
                                [email])

                        from relate.utils import get_outbound_mail_connection
                        msg.connection = (
                                get_outbound_mail_connection("no_reply")
                                if hasattr(settings, "NO_REPLY_EMAIL_FROM")
                                else get_outbound_mail_connection("robot"))
                        msg.send()

                        if field == "instid":
                            messages.add_message(request, messages.INFO,
                                _("The email address associated with that "
                                  "account is %s.")
                                % masked_email(email))
                        messages.add_message(request, messages.INFO,
                                _("Email sent. Please check your email and "
                                  "click the link."))
                        return redirect("relate-home")
    else:
        form = ResetPasswordForm()

Dong Zhuang's avatar
Dong Zhuang committed
    return render(request, "reset-passwd-form.html", {
        "field": field,
Andreas Klöckner's avatar
Andreas Klöckner committed
        "form_description":
            _("Password reset on %(site_name)s")
            % {"site_name": _(get_site_name())},
        "form": form
        })


class ResetPasswordStage2Form(StyledForm):
ifaint's avatar
ifaint committed
    password = forms.CharField(widget=forms.PasswordInput(),
                              label=_("Password"))
    password_repeat = forms.CharField(widget=forms.PasswordInput(),

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)

        self.helper.add_input(
                Submit("submit_user", _("Update")))

    def clean(self):
        password = cleaned_data.get("password")
        password_repeat = cleaned_data.get("password_repeat")
        if password and password != password_repeat:
Andreas Klöckner's avatar
Andreas Klöckner committed
            self.add_error("password_repeat",
                    _("The two password fields didn't match."))
@logout_confirmation_required
def reset_password_stage2(request, user_id, sign_in_key):
    if not settings.RELATE_REGISTRATION_ENABLED:
        raise SuspiciousOperation(
                _("self-registration is not enabled"))
    def check_sign_in_key(user_id, token):
        user = get_user_model().objects.get(id=user_id)
        return user.sign_in_key == token

    try:
        if not check_sign_in_key(user_id=int(user_id), token=sign_in_key):
            messages.add_message(request, messages.ERROR,
                    _("Invalid sign-in token. Perhaps you've used an old token "
                    "email?"))
            raise PermissionDenied(_("invalid sign-in token"))
    except get_user_model().DoesNotExist:
        messages.add_message(request, messages.ERROR, _("Account does not exist."))
        raise PermissionDenied(_("invalid sign-in token"))
    if request.method == "POST":
        form = ResetPasswordStage2Form(request.POST)
        if form.is_valid():
            from django.contrib.auth import authenticate, login
            user = authenticate(user_id=int(user_id), token=sign_in_key)
            if user is None:
Dong Zhuang's avatar
Dong Zhuang committed
                messages.add_message(request, messages.ERROR,
                     _("Invalid sign-in token. Perhaps you've used an old token "
                     "email?"))
                raise PermissionDenied(_("invalid sign-in token"))

            if not user.is_active:
                messages.add_message(request, messages.ERROR,
                        _("Account disabled."))
ifaint's avatar
ifaint committed
                raise PermissionDenied(_("invalid sign-in token"))

            user.set_password(form.cleaned_data["password"])
            user.save()

            login(request, user)

            if (not (user.first_name and user.last_name)
                    or "to_profile" in request.GET):
                messages.add_message(request, messages.INFO,
ifaint's avatar
ifaint committed
                        _("Successfully signed in. "
                        "Please complete your registration information below."))

                return redirect(
                       reverse("relate-user_profile")+"?first_login=1")
            else:
                messages.add_message(request, messages.INFO,
                        _("Successfully signed in."))
    else:
        form = ResetPasswordStage2Form()

    return render(request, "generic-form.html", {
Andreas Klöckner's avatar
Andreas Klöckner committed
        "form_description":
            _("Password reset on %(site_name)s")
            % {"site_name": _(get_site_name())},
# }}}


# {{{ email sign-in flow

class SignInByEmailForm(StyledForm):
    email = forms.EmailField(required=True, label=_("Email"),
            # For now, until we upgrade to a custom user model.
            max_length=User._meta.get_field("email").max_length)

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)

        self.helper.add_input(
                Submit("submit", _("Send sign-in email")))
@logout_confirmation_required
def sign_in_by_email(request):
    if not settings.RELATE_SIGN_IN_BY_EMAIL_ENABLED:
        messages.add_message(request, messages.ERROR,
                _("Email-based sign-in is not being used"))
        return redirect("relate-sign_in_choice")
    if request.method == "POST":
        form = SignInByEmailForm(request.POST)
        if form.is_valid():
            email = form.cleaned_data["email"]
            user, created = get_user_model().objects.get_or_create(
                    email__iexact=email,
                    defaults={"username": email, "email": email})
                user.set_unusable_password()
                user.status = user_status.unconfirmed
Andreas Klöckner's avatar
Andreas Klöckner committed
            user.sign_in_key = make_sign_in_key(user)
            user.save()
            from relate.utils import render_email_template
            message = render_email_template("course/sign-in-email.txt", {
                "user": user,
                "sign_in_uri": request.build_absolute_uri(
                    reverse(
                        "relate-sign_in_stage2_with_token",
Andreas Klöckner's avatar
Andreas Klöckner committed
                        args=(user.id, user.sign_in_key,))),
                "home_uri": request.build_absolute_uri(reverse("relate-home"))
            from django.core.mail import EmailMessage
            msg = EmailMessage(
                    _("Your %(relate_site_name)s sign-in link")
                    % {"relate_site_name": _(get_site_name())},
                    getattr(settings, "NO_REPLY_EMAIL_FROM",
                            settings.ROBOT_EMAIL_FROM),
            from relate.utils import get_outbound_mail_connection
            msg.connection = (
                get_outbound_mail_connection("no_reply")
                if hasattr(settings, "NO_REPLY_EMAIL_FROM")
                else get_outbound_mail_connection("robot"))
            msg.send()
            messages.add_message(request, messages.INFO,
                    _("Email sent. Please check your email and click the link."))
    else:
        form = SignInByEmailForm()

    return render(request, "course/login-by-email.html", {
        "form_description": "",
        "form": form
        })


@logout_confirmation_required
def sign_in_stage2_with_token(request, user_id, sign_in_key):
    if not settings.RELATE_SIGN_IN_BY_EMAIL_ENABLED:
        messages.add_message(request, messages.ERROR,
                _("Email-based sign-in is not being used"))
        return redirect("relate-sign_in_choice")
    from django.contrib.auth import authenticate, login
    user = authenticate(user_id=int(user_id), token=sign_in_key)
    if user is None:
        if not get_user_model().objects.filter(pk=int(user_id)).count():
            messages.add_message(request, messages.ERROR,
                _("Account does not exist."))
        else:
            messages.add_message(request, messages.ERROR,
                    _("Invalid sign-in token. Perhaps you've used an old "
                    "token email?"))
        raise PermissionDenied(_("invalid sign-in token"))
    if not user.is_active:
        messages.add_message(request, messages.ERROR,
                _("Account disabled."))
        raise PermissionDenied(_("invalid sign-in token"))

    login(request, user)

    if not (user.first_name and user.last_name):
        messages.add_message(request, messages.INFO,
ifaint's avatar
ifaint committed
                _("Successfully signed in. "
                "Please complete your registration information below."))
               reverse("relate-user_profile")+"?first_login=1")
    else:
        messages.add_message(request, messages.INFO,
                _("Successfully signed in."))
# {{{ user profile

def is_inst_id_editable_before_validation() -> bool:
    return getattr(
        settings, "RELATE_EDITABLE_INST_ID_BEFORE_VERIFICATION", True)


    institutional_id_confirm = forms.CharField(
            max_length=100,
            label=_("Institutional ID Confirmation"),
            required=False)

    class Meta:
        model = get_user_model()
        fields = ("first_name", "last_name", "email", "institutional_id",
                "editor_mode")

    def __init__(self, *args, **kwargs):
        self.is_inst_id_locked = kwargs.pop("is_inst_id_locked")
        super().__init__(*args, **kwargs)
Dong Zhuang's avatar
Dong Zhuang committed

Dong Zhuang's avatar
Dong Zhuang committed
        if self.instance.name_verified:
            self.fields["first_name"].disabled = True
            self.fields["last_name"].disabled = True

        self.fields["email"].disabled = True

Dong Zhuang's avatar
Dong Zhuang committed
        if self.is_inst_id_locked:
            self.fields["institutional_id"].disabled = True
            self.fields["institutional_id_confirm"].disabled = True
        else:
            self.fields["institutional_id_confirm"].initial = (
                self.instance.institutional_id)
        self.fields["institutional_id"].help_text = (
                _("The unique ID your university or school provided, "
                    "which may be used by some courses to verify "
                    "eligibility to enroll. "
                    "<b>Once %(submitted_or_verified)s, it cannot be "
                    "changed</b>.")
                % {"submitted_or_verified":
Andreas Klöckner's avatar
Andreas Klöckner committed
                   (is_inst_id_editable_before_validation()
                   and _("verified")) or _("submitted")})
        # {{{ build layout

        name_fields_layout = ["last_name", "first_name", "email"]
Dong Zhuang's avatar
Dong Zhuang committed
        fields_layout = [Div(*name_fields_layout, css_class="well")]

        if getattr(settings, "RELATE_SHOW_INST_ID_FORM", True):
            inst_field_group_layout = ["institutional_id"]
            if not self.is_inst_id_locked:
                inst_field_group_layout.append("institutional_id_confirm")
            fields_layout.append(Div(*inst_field_group_layout, css_class="well",
                                     css_id="institutional_id_block"))
Dong Zhuang's avatar
Dong Zhuang committed
            # This is needed for django-crispy-form version < 1.7
            self.fields["institutional_id"].widget = forms.HiddenInput()
Dong Zhuang's avatar
Dong Zhuang committed
        if getattr(settings, "RELATE_SHOW_EDITOR_FORM", True):
            fields_layout.append(Div("editor_mode", css_class="well"))
        else:
            # This is needed for django-crispy-form version < 1.7
            self.fields["editor_mode"].widget = forms.HiddenInput()

        self.helper.layout = Layout(*fields_layout)
        self.helper.add_input(
                Submit("submit_user", _("Update")))
Dong Zhuang's avatar
Dong Zhuang committed
        self.helper.add_input(
                Button("signout", _("Sign out"), css_class="btn btn-danger",
                       onclick=(
                           "window.location.href='{}'".format(reverse("relate-logout")))))
Dong Zhuang's avatar
Dong Zhuang committed
        # }}}
Dong Zhuang's avatar
Dong Zhuang committed

    def clean_institutional_id_confirm(self):
        inst_id_confirmed = self.cleaned_data.get("institutional_id_confirm")
            inst_id = self.cleaned_data.get("institutional_id")
            if inst_id and not inst_id_confirmed:
                raise forms.ValidationError(_("This field is required."))
Dong Zhuang's avatar
Dong Zhuang committed
            if any([inst_id, inst_id_confirmed]) and inst_id != inst_id_confirmed:
                raise forms.ValidationError(_("Inputs do not match."))
        return inst_id_confirmed
Dong Zhuang's avatar
Dong Zhuang committed
@login_required
def user_profile(request):
    user_form = None
        if is_inst_id_editable_before_validation():
            return True if (user.institutional_id
                    and user.institutional_id_verified) else False
        else:
            return True if user.institutional_id else False
Dong Zhuang's avatar
Dong Zhuang committed
    def is_requesting_inst_id():
        return not is_inst_id_locked(request.user) and (
            request.GET.get("first_login")
            or (request.GET.get("set_inst_id")
                and request.GET.get("referer")))

    if request.method == "POST":
        if "submit_user" in request.POST:
            user_form = UserForm(
                    request.POST,
                    instance=request.user,
                    is_inst_id_locked=is_inst_id_locked(request.user),