maubot-kickbot/kickbot/db.py

20 lines
634 B
Python

from __future__ import annotations
from mautrix.util.async_db import UpgradeTable, Connection
upgrade_table = UpgradeTable()
@upgrade_table.register(description="Table initialization")
async def upgrade_v1(conn: Connection) -> None:
await conn.execute(
"""CREATE TABLE user_events (
mxid TEXT PRIMARY KEY,
last_message_timestamp BIGINT NOT NULL
)"""
)
@upgrade_table.register(description="Include ignore_inactivity column")
async def upgrade_v2(conn: Connection) -> None:
await conn.execute("ALTER TABLE user_events ADD COLUMN ignore_inactivity INT DEFAULT O")