Newer
Older
from __future__ import annotations
__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 django.forms as forms
from crispy_forms.helper import FormHelper
from django.contrib import messages
from django.contrib.auth import (
REDIRECT_FIELD_NAME,
get_user_model,
login as auth_login,
logout as auth_logout,
)
from django.contrib.auth.decorators import login_required, user_passes_test
from django.contrib.auth.forms import AuthenticationForm as AuthenticationFormBase
from django.contrib.auth.validators import ASCIIUsernameValidator
MultipleObjectsReturned,
ObjectDoesNotExist,
PermissionDenied,
from django.shortcuts import get_object_or_404, redirect, render, resolve_url
from django.template.response import TemplateResponse
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
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
participation_permission as pperm,
participation_status,
user_status,
AuthenticationToken,
Participation,
ParticipationRole,
from course.utils import CoursePageContext, course_view, render_course_page
HTML5DateTimeInput,
StyledForm,
StyledModelForm,
get_site_name,
string_concat,
if TYPE_CHECKING:
# {{{ impersonation
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:
return User.objects.exclude(pk=impersonator.pk)
Andreas Klöckner
committed
my_participations = Participation.objects.filter(
Dong Zhuang
committed
user=impersonator,
status=participation_status.active)
impersonable_user_qset = User.objects.none()
Andreas Klöckner
committed
for part in my_participations:
# 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):
Dong Zhuang
committed
argument
for perm, argument in part.permissions()
if perm == pperm.impersonate_role]
Dong Zhuang
committed
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
| User.objects.filter(pk__in=q.values_list("user__pk", flat=True))
class ImpersonateMiddleware:
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
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():
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 (
class ImpersonateForm(StyledForm):
def __init__(self, *args: Any, **kwargs: Any) -> None:
Dong Zhuang
committed
qset = kwargs.pop("impersonable_qset")
super().__init__(*args, **kwargs)
self.fields["user"] = forms.ModelChoiceField(
Dong Zhuang
committed
queryset=qset,
widget=UserSearchWidget(
queryset=qset,
attrs={"data-minimum-input-length": 0},
),
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:
Dong Zhuang
committed
if not request.user.is_authenticated:
raise PermissionDenied()
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."))
# 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":
Dong Zhuang
committed
form = ImpersonateForm(request.POST, impersonable_qset=qset)
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")
Dong Zhuang
committed
form = ImpersonateForm(impersonable_qset=qset)
"form_description": _("Impersonate user"),
def stop_impersonating(request: http.HttpRequest) -> http.JsonResponse:
raise PermissionDenied(_("only AJAX POST is allowed"))
if not request.user.is_authenticated:
raise PermissionDenied()
if "stop_impersonating" not in request.POST:
raise SuspiciousOperation(_("odd POST parameters"))
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"))
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):
"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
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"))
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(
login_url=logout_confirmation_url,
redirect_field_name=redirect_field_name
)
if func:
return actual_decorator(func)
return actual_decorator
class EmailedTokenBackend:
Andreas Klöckner
committed
def authenticate(self, request, user_id=None, token=None):
users = get_user_model().objects.filter(
id=user_id, sign_in_key=token)
assert users.count() <= 1
if users.count() == 0:
user.status = user_status.active
user.sign_in_key = None
user.save()
def get_user(self, user_id):
try:
return get_user_model().objects.get(pk=user_id)
except get_user_model().DoesNotExist:
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, ""))
next_uri = f"?{redirect_field_name}={redirect_to}"
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
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(
url=redirect_to,
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)
next_uri = f"?{redirect_field_name}={redirect_to}"
context = {
redirect_field_name: redirect_to,
"next_uri": next_uri,
}
return TemplateResponse(request, "course/login.html", context)
class SignUpForm(StyledModelForm):
username = forms.CharField(required=True, max_length=30,
fields = ("email",)
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.fields["email"].required = True
if not settings.RELATE_REGISTRATION_ENABLED:
raise SuspiciousOperation(
_("self-registration is not enabled"))
if request.method == "POST":
form = SignUpForm(request.POST)
if form.is_valid():
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"]
email=email,
username=form.cleaned_data["username"])
user.set_unusable_password()
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",
+ "?to_profile=1"),
"home_uri": request.build_absolute_uri(
reverse("relate-home"))
from django.core.mail import EmailMessage
msg = EmailMessage(
string_concat(f"[{_(get_site_name())}] ",
_("Verify your email")),
getattr(settings, "NO_REPLY_EMAIL_FROM",
from relate.utils import get_outbound_mail_connection
get_outbound_mail_connection("no_reply")
if hasattr(settings, "NO_REPLY_EMAIL_FROM")
else get_outbound_mail_connection("robot"))
messages.add_message(request, messages.INFO,
_("Email sent. Please check your email and click "
"the link."))
return redirect("relate-home")
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", {
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)
class ResetPasswordFormByInstid(StyledForm):
instid = forms.CharField(max_length=100,
required=True,
label=_("Institutional ID"))
super().__init__(*args, **kwargs)
self.helper.add_input(
Submit("submit", _("Send email")))
def masked_email(email):
# return a masked email address
at = email.find("@")
return email[:2] + "*" * (len(email[3:at])-1) + email[at-1:]
if not settings.RELATE_REGISTRATION_ENABLED:
raise SuspiciousOperation(
_("self-registration is not enabled"))
ResetPasswordForm = globals()["ResetPasswordFormBy" + field.title()] # noqa
if request.method == "POST":
form = ResetPasswordForm(request.POST)
if field == "instid":
inst_id = form.cleaned_data["instid"]
try:
user = get_user_model().objects.get(
institutional_id__iexact=inst_id)
except ObjectDoesNotExist:
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 "
else:
if user is None:
FIELD_DICT = { # noqa
"email": _("email address"),
"instid": _("institutional ID")
}
_("That %(field)s doesn't have an "
"associated user account. Are you "
"sure you've registered?")
% {"field": FIELD_DICT[field]})
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."))
else:
form = ResetPasswordForm()
return render(request, "reset-passwd-form.html", {
"field": field,
"form_description":
_("Password reset on %(site_name)s")
% {"site_name": _(get_site_name())},
"form": form
})
class ResetPasswordStage2Form(StyledForm):
password = forms.CharField(widget=forms.PasswordInput(),
label=_("Password"))
password_repeat = forms.CharField(widget=forms.PasswordInput(),
label=_("Password confirmation"))
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
cleaned_data = super().clean()
password = cleaned_data.get("password")
password_repeat = cleaned_data.get("password_repeat")
if password and password != password_repeat:
_("The two password fields didn't match."))
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:
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,
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,
_("Successfully signed in. "
"Please complete your registration information below."))
reverse("relate-user_profile")+"?first_login=1")
else:
messages.add_message(request, messages.INFO,
return redirect("relate-home")
else:
form = ResetPasswordStage2Form()
return render(request, "generic-form.html", {
"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)
super().__init__(*args, **kwargs)
Submit("submit", _("Send sign-in email")))
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(
defaults={"username": email, "email": email})
user.status = user_status.unconfirmed
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",
"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",
from relate.utils import get_outbound_mail_connection
get_outbound_mail_connection("no_reply")
if hasattr(settings, "NO_REPLY_EMAIL_FROM")
else get_outbound_mail_connection("robot"))
messages.add_message(request, messages.INFO,
_("Email sent. Please check your email and click the link."))
return redirect("relate-home")
else:
form = SignInByEmailForm()
return render(request, "course/login-by-email.html", {
"form_description": "",
"form": form
})
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 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,
_("Successfully signed in. "
"Please complete your registration information below."))
Andreas Klöckner
committed
return redirect(
reverse("relate-user_profile")+"?first_login=1")
else:
messages.add_message(request, messages.INFO,
return redirect("relate-home")
def is_inst_id_editable_before_validation() -> bool:
return getattr(
settings, "RELATE_EDITABLE_INST_ID_BEFORE_VERIFICATION", True)
class UserForm(StyledModelForm):
institutional_id_confirm = forms.CharField(
max_length=100,
label=_("Institutional ID Confirmation"),
required=False)
fields = ("first_name", "last_name", "email", "institutional_id",
self.is_inst_id_locked = kwargs.pop("is_inst_id_locked")
super().__init__(*args, **kwargs)
if self.instance.name_verified:
self.fields["first_name"].disabled = True
self.fields["last_name"].disabled = True
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 "
% {"submitted_or_verified":
(is_inst_id_editable_before_validation()
and _("verified")) or _("submitted")})
# {{{ build layout
name_fields_layout = ["last_name", "first_name", "email"]
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"))
else:
# This is needed for django-crispy-form version < 1.7
self.fields["institutional_id"].widget = forms.HiddenInput()
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)
Andreas Klöckner
committed
self.helper.add_input(
Button("signout", _("Sign out"), css_class="btn btn-danger",
onclick=(
"window.location.href='{}'".format(reverse("relate-logout")))))
def clean_institutional_id_confirm(self):
inst_id_confirmed = self.cleaned_data.get("institutional_id_confirm")
if not self.is_inst_id_locked:
inst_id = self.cleaned_data.get("institutional_id")
if inst_id and not inst_id_confirmed:
raise forms.ValidationError(_("This field is required."))
if any([inst_id, inst_id_confirmed]) and inst_id != inst_id_confirmed:
raise forms.ValidationError(_("Inputs do not match."))
return inst_id_confirmed
Andreas Klöckner
committed
def is_inst_id_locked(user):
if is_inst_id_editable_before_validation():
return True if (user.institutional_id
and user.institutional_id_verified) else False
return True if user.institutional_id else False
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),