bookwyrm/bookwyrm/models/move.py

""" move an object including migrating a user account """
from django.core.exceptions import PermissionDenied
from django.db import models

from bookwyrm import activitypub
from .activitypub_mixin import ActivityMixin
from .base_model import BookWyrmModel
from . import fields
from .notification import Notification, NotificationType


class Move(ActivityMixin, BookWyrmModel):
    """migrating an activitypub object"""

    user = fields.ForeignKey(
        "User", on_delete=models.PROTECT, activitypub_field="actor"
    )

    object = fields.CharField(
        max_length=255,
        blank=False,
        null=False,
        activitypub_field="object",
    )

    origin = fields.CharField(
        max_length=255,
        blank=True,
        null=True,
        default="",
        activitypub_field="origin",
    )

    activity_serializer = activitypub.Move


class MoveUser(Move):
    """migrating an activitypub user account"""

    target = fields.ForeignKey(
        "User",
        on_delete=models.PROTECT,
        related_name="move_target",
        activitypub_field="target",
    )

    def save(self, *args, **kwargs):
        """update user info and broadcast it"""

        # only allow if the source is listed in the target's alsoKnownAs
        if self.user not in self.target.also_known_as.all():
            raise PermissionDenied()

        self.user.also_known_as.add(self.target.id)
        self.user.update_active_date()
        self.user.moved_to = self.target.remote_id
        self.user.save(update_fields=["moved_to"])

        if self.user.local:
            kwargs["broadcast"] = True  # Only broadcast if we are initiating the Move

        super().save(*args, **kwargs)

        for follower in self.user.followers.all():
            if follower.local:
                Notification.notify(
                    follower, self.user, notification_type=NotificationType.MOVE
                )