matrix-appservice-kakaotalk/matrix_appservice_kakaotalk/db/__init__.py

from mautrix.util.async_db import Database

from .message import Message
from .portal import Portal
from .puppet import Puppet
from .upgrade import upgrade_table
from .user import User


def init(db: Database) -> None:
    for table in (Portal, Message, User, Puppet):
        table.db = db


__all__ = [
    "upgrade_table",
    "init",
    "Message",
    "Portal",
    "ThreadType",
    "Puppet",
    "User",
]