mirror of
https://github.com/bookwyrm-social/bookwyrm.git
synced 2024-06-03 05:49:35 +00:00
99a9dbe5f4
This way, we need not list every value again to create the enum. N.B.: enum values are now accessed as `models.NotificationType.FOO`, instead of `models.Notification.FOO`.
72 lines
2 KiB
Python
72 lines
2 KiB
Python
""" 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 user account"""
|
|
|
|
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 in self.target.also_known_as.all():
|
|
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
|
|
)
|
|
|
|
else:
|
|
raise PermissionDenied()
|