moviewyrm/bookwyrm/models/__init__.py

''' bring all the models into the app namespace '''
import inspect
import sys

from .book import Book, Work, Edition
from .author import Author
from .connector import Connector

from .shelf import Shelf, ShelfBook

from .status import Status, GeneratedNote, Review, Comment, Quotation
from .status import Favorite, Boost, Notification, ReadThrough
from .attachment import Image

from .tag import Tag, UserTag

from .user import User, KeyPair
from .relationship import UserFollows, UserFollowRequest, UserBlocks
from .federated_server import FederatedServer

from .import_job import ImportJob, ImportItem

from .site import SiteSettings, SiteInvite, PasswordReset

cls_members = inspect.getmembers(sys.modules[__name__], inspect.isclass)
activity_models = {c[1].activity_serializer.__name__: c[1] \
    for c in cls_members if hasattr(c[1], 'activity_serializer')}

def to_activity(activity_json):
    ''' link up models and activities '''
    activity_type = activity_json.get('type')
    return activity_models[activity_type].to_activity(activity_json)