takahe/users/views/settings/migration.py

54 wiersze
1.9 KiB
Python

from django import forms
from django.conf import settings
from django.contrib import messages
from django.contrib.auth.decorators import login_required
from django.http import Http404
from django.shortcuts import redirect
from django.utils.decorators import method_decorator
from django.views.generic import FormView
from users.models import Identity
from users.views.base import IdentityViewMixin
@method_decorator(login_required, name="dispatch")
class MigrateInPage(IdentityViewMixin, FormView):
"""
Lets the identity's profile be migrated in or out.
"""
template_name = "settings/migrate_in.html"
extra_context = {"section": "migrate_in"}
class form_class(forms.Form):
alias = forms.CharField(
help_text="The @account@example.com username you want to move here"
)
def clean_alias(self):
self.alias_identity = Identity.by_handle(
self.cleaned_data["alias"], fetch=True
)
if self.alias_identity is None:
raise forms.ValidationError("Cannot find that account.")
return self.alias_identity.actor_uri
def form_valid(self, form):
self.identity.add_alias(form.cleaned_data["alias"])
messages.info(self.request, f"Alias to {form.alias_identity.handle} added")
return redirect(".")
def get_context_data(self, **kwargs):
context = super().get_context_data(**kwargs)
if not settings.SETUP.ALLOW_USER_MIGRATION:
raise Http404()
# If they asked for an alias deletion, do it here
if "remove_alias" in self.request.GET:
self.identity.remove_alias(self.request.GET["remove_alias"])
context["aliases"] = []
if self.identity.aliases:
context["aliases"] = [
Identity.by_actor_uri(uri) for uri in self.identity.aliases
]
return context