from mautrix.util.async_db import Database
import sqlite3
import uuid

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


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


# TODO should this be in mautrix-python?
sqlite3.register_adapter(uuid.UUID, lambda u: str(u))
sqlite3.register_converter("UUID", lambda b: uuid.UUID(b))

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