24 lines
413 B
Python
24 lines
413 B
Python
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",
|
|
]
|