forked from mirrors/bookwyrm
168 lines
3.8 KiB
Python
168 lines
3.8 KiB
Python
''' undo wrapper activity '''
|
|
from dataclasses import dataclass
|
|
from typing import List
|
|
from django.apps import apps
|
|
|
|
from .base_activity import ActivityObject, Signature, resolve_remote_id
|
|
from .book import Edition
|
|
|
|
|
|
@dataclass(init=False)
|
|
class Verb(ActivityObject):
|
|
''' generic fields for activities - maybe an unecessary level of
|
|
abstraction but w/e '''
|
|
actor: str
|
|
object: ActivityObject
|
|
|
|
def action(self):
|
|
''' usually we just want to save, this can be overridden as needed '''
|
|
self.object.to_model()
|
|
|
|
|
|
@dataclass(init=False)
|
|
class Create(Verb):
|
|
''' Create activity '''
|
|
to: List
|
|
cc: List
|
|
signature: Signature = None
|
|
type: str = 'Create'
|
|
|
|
|
|
@dataclass(init=False)
|
|
class Delete(Verb):
|
|
''' Create activity '''
|
|
to: List
|
|
cc: List
|
|
type: str = 'Delete'
|
|
|
|
def action(self):
|
|
''' find and delete the activity object '''
|
|
obj = self.object.to_model(save=False, allow_create=False)
|
|
obj.delete()
|
|
|
|
|
|
|
|
@dataclass(init=False)
|
|
class Update(Verb):
|
|
''' Update activity '''
|
|
to: List
|
|
type: str = 'Update'
|
|
|
|
def action(self):
|
|
''' update a model instance from the dataclass '''
|
|
self.object.to_model(allow_create=False)
|
|
|
|
|
|
@dataclass(init=False)
|
|
class Undo(Verb):
|
|
''' Undo an activity '''
|
|
type: str = 'Undo'
|
|
|
|
def action(self):
|
|
''' find and remove the activity object '''
|
|
# this is so hacky but it does make it work....
|
|
# (because you Reject a request and Undo a follow
|
|
model = None
|
|
if self.object.type == 'Follow':
|
|
model = apps.get_model('bookwyrm.UserFollows')
|
|
obj = self.object.to_model(model=model, save=False, allow_create=False)
|
|
obj.delete()
|
|
|
|
|
|
@dataclass(init=False)
|
|
class Follow(Verb):
|
|
''' Follow activity '''
|
|
object: str
|
|
type: str = 'Follow'
|
|
|
|
def action(self):
|
|
''' relationship save '''
|
|
self.to_model()
|
|
|
|
|
|
@dataclass(init=False)
|
|
class Block(Verb):
|
|
''' Block activity '''
|
|
object: str
|
|
type: str = 'Block'
|
|
|
|
def action(self):
|
|
''' relationship save '''
|
|
self.to_model()
|
|
|
|
|
|
@dataclass(init=False)
|
|
class Accept(Verb):
|
|
''' Accept activity '''
|
|
object: Follow
|
|
type: str = 'Accept'
|
|
|
|
def action(self):
|
|
''' find and remove the activity object '''
|
|
obj = self.object.to_model(save=False, allow_create=False)
|
|
obj.accept()
|
|
|
|
|
|
@dataclass(init=False)
|
|
class Reject(Verb):
|
|
''' Reject activity '''
|
|
object: Follow
|
|
type: str = 'Reject'
|
|
|
|
def action(self):
|
|
''' find and remove the activity object '''
|
|
obj = self.object.to_model(save=False, allow_create=False)
|
|
obj.reject()
|
|
|
|
|
|
@dataclass(init=False)
|
|
class Add(Verb):
|
|
'''Add activity '''
|
|
target: str
|
|
object: Edition
|
|
type: str = 'Add'
|
|
notes: str = None
|
|
order: int = 0
|
|
approved: bool = True
|
|
|
|
def action(self):
|
|
''' add obj to collection '''
|
|
target = resolve_remote_id(self.target, refresh=False)
|
|
# we want to related field that isn't the book, this is janky af sorry
|
|
model = [t for t in type(target)._meta.related_objects \
|
|
if t.name != 'edition'][0].related_model
|
|
self.to_model(model=model)
|
|
|
|
|
|
@dataclass(init=False)
|
|
class Remove(Verb):
|
|
'''Remove activity '''
|
|
target: ActivityObject
|
|
type: str = 'Remove'
|
|
|
|
def action(self):
|
|
''' find and remove the activity object '''
|
|
obj = self.object.to_model(save=False, allow_create=False)
|
|
obj.delete()
|
|
|
|
|
|
@dataclass(init=False)
|
|
class Like(Verb):
|
|
''' a user faving an object '''
|
|
object: str
|
|
type: str = 'Like'
|
|
|
|
def action(self):
|
|
''' like '''
|
|
self.to_model()
|
|
|
|
|
|
@dataclass(init=False)
|
|
class Announce(Verb):
|
|
''' boosting a status '''
|
|
object: str
|
|
type: str = 'Announce'
|
|
|
|
def action(self):
|
|
''' boost '''
|
|
self.to_model()
|