import markdown_it from django import forms from django.conf import settings from django.contrib.auth.forms import AuthenticationForm from django.contrib.auth.password_validation import validate_password from django.contrib.auth.views import LoginView, LogoutView from django.shortcuts import get_object_or_404, render from django.utils.safestring import mark_safe from django.utils.translation import gettext_lazy as _ from django.views.generic import FormView from core.models import Config from users.models import Invite, PasswordReset, User class Login(LoginView): class form_class(AuthenticationForm): error_messages = { "invalid_login": _("No account was found with that email and password."), "inactive": _("This account is inactive."), } template_name = "auth/login.html" class Logout(LogoutView): pass class Signup(FormView): template_name = "auth/signup.html" class form_class(forms.Form): email = forms.EmailField( help_text="We will send a link to this email to set your password and create your account", ) def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) # Add the invite field if it's enabled if Config.system.signup_invite_only: self.fields["invite_code"] = forms.CharField( help_text="Your invite code from one of our admins" ) # Add the policies if they're defined policies = [] if Config.system.policy_rules: policies.append("Server Rules") if Config.system.policy_terms: policies.append("Terms of Service") if Config.system.policy_privacy: policies.append("Privacy Policy") if policies: links = "" for i, policy in enumerate(policies): if i == 0: links += policy elif i == len(policies) - 1: if len(policies) > 2: links += ", and " else: links += " and " links += policy else: links += ", " links += policy self.fields["policy"] = forms.BooleanField( label="Policies", help_text=f"Have you read the {links}, and agree to them?", widget=forms.Select( choices=[(False, "I do not agree"), (True, "I agree")] ), ) def clean_email(self): email = self.cleaned_data.get("email").lower() if not email: return if User.objects.filter(email=email).exists(): raise forms.ValidationError("This email already has an account") return email def clean_invite_code(self): invite_code = self.cleaned_data["invite_code"].lower().strip() invite = Invite.objects.filter(token=invite_code).first() if not invite: raise forms.ValidationError("That is not a valid invite code") if invite.email and invite.email != self.cleaned_data.get("email"): raise forms.ValidationError( "That is not a valid invite code for this email address" ) return invite_code def clean(self): if not Config.system.signup_allowed: raise forms.ValidationError("Not accepting new users at this time") def form_valid(self, form): user = User.objects.create(email=form.cleaned_data["email"]) # Auto-promote the user to admin if that setting is set if settings.AUTO_ADMIN_EMAIL and user.email == settings.AUTO_ADMIN_EMAIL: user.admin = True user.save() PasswordReset.create_for_user(user) if "invite_code" in form.cleaned_data: Invite.objects.filter(token=form.cleaned_data["invite_code"]).delete() return render( self.request, "auth/signup_success.html", {"email": user.email}, ) def get_context_data(self, **kwargs): context = super().get_context_data(**kwargs) if Config.system.signup_text: context["signup_text"] = mark_safe( markdown_it.MarkdownIt().render(Config.system.signup_text) ) return context class TriggerReset(FormView): template_name = "auth/trigger_reset.html" class form_class(forms.Form): email = forms.EmailField( help_text="We will send a reset link to this email", ) def clean_email(self): email = self.cleaned_data.get("email").lower() if not email: return if not User.objects.filter(email=email).exists(): raise forms.ValidationError("This email does not have an account") return email def form_valid(self, form): PasswordReset.create_for_user( User.objects.get(email=form.cleaned_data["email"]) ) return render( self.request, "auth/trigger_reset_success.html", {"email": form.cleaned_data["email"]}, ) class PerformReset(FormView): template_name = "auth/perform_reset.html" class form_class(forms.Form): password = forms.CharField( widget=forms.PasswordInput, help_text="Must be at least 8 characters, and contain both letters and numbers.", ) repeat_password = forms.CharField( widget=forms.PasswordInput, ) def clean_password(self): password = self.cleaned_data["password"] validate_password(password) return password def clean_repeat_password(self): if self.cleaned_data.get("password") != self.cleaned_data.get( "repeat_password" ): raise forms.ValidationError("Passwords do not match") return self.cleaned_data.get("repeat_password") def dispatch(self, request, token): self.reset = get_object_or_404(PasswordReset, token=token) return super().dispatch(request) def form_valid(self, form): self.reset.user.set_password(form.cleaned_data["password"]) self.reset.user.save() self.reset.delete() return render( self.request, "auth/perform_reset_success.html", {"email": self.reset.user.email}, ) def get_context_data(self, *args, **kwargs): context = super().get_context_data(*args, **kwargs) context["reset"] = self.reset return context