bookwyrm/bookwyrm/activitypub/verbs.py

''' undo wrapper activity '''
from dataclasses import dataclass
from typing import List

from .base_activity import ActivityObject, Signature
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 '''
        obj = self.object.to_model(save=False, allow_create=False)
        obj.delete()


@dataclass(init=False)
class Follow(Verb):
    ''' Follow activity '''
    object: str
    type: str = 'Follow'


@dataclass(init=False)
class Block(Verb):
    ''' Block activity '''
    object: str
    type: str = 'Block'


@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: ActivityObject
    type: str = 'Add'


@dataclass(init=False)
class AddBook(Add):
    '''Add activity that's aware of the book obj '''
    object: Edition
    type: str = 'Add'


@dataclass(init=False)
class AddListItem(AddBook):
    '''Add activity that's aware of the book obj '''
    notes: str = None
    order: int = 0
    approved: bool = True


@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()