Create standalone_next_batch table automatically
This commit is contained in:
parent
484749c08c
commit
e9e325a871
2 changed files with 22 additions and 2 deletions
|
@ -49,7 +49,7 @@ from ..plugin_base import Plugin
|
|||
from ..plugin_server import PluginWebApp, PrefixResource
|
||||
from ..server import AccessLogger
|
||||
from .config import Config
|
||||
from .database import NextBatch
|
||||
from .database import NextBatch, upgrade_table
|
||||
from .loader import FileSystemLoader
|
||||
|
||||
crypto_import_error = None
|
||||
|
@ -124,6 +124,7 @@ db = Database.create(
|
|||
db_args=config.get("database_opts", {}),
|
||||
ignore_foreign_tables=True,
|
||||
log=logging.getLogger("maubot.db"),
|
||||
upgrade_table=upgrade_table,
|
||||
)
|
||||
|
||||
user_id = config["user.credentials.id"]
|
||||
|
|
|
@ -15,11 +15,30 @@
|
|||
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||
from __future__ import annotations
|
||||
|
||||
import logging
|
||||
|
||||
from attr import dataclass
|
||||
|
||||
from mautrix.client import SyncStore
|
||||
from mautrix.types import FilterID, SyncToken, UserID
|
||||
from mautrix.util.async_db import Database
|
||||
from mautrix.util.async_db import Connection, Database, UpgradeTable
|
||||
|
||||
upgrade_table = UpgradeTable(
|
||||
version_table_name="standalone_version", log=logging.getLogger("maubot.db")
|
||||
)
|
||||
|
||||
|
||||
@upgrade_table.register(description="Initial revision")
|
||||
async def upgrade_v1(conn: Connection) -> None:
|
||||
await conn.execute(
|
||||
"""CREATE TABLE IF NOT EXISTS standalone_next_batch (
|
||||
user_id TEXT NOT NULL,
|
||||
next_batch TEXT,
|
||||
filter_id TEXT,
|
||||
PRIMARY KEY (user_id)
|
||||
)"""
|
||||
)
|
||||
|
||||
|
||||
find_q = "SELECT next_batch, filter_id FROM standalone_next_batch WHERE user_id=$1"
|
||||
insert_q = "INSERT INTO standalone_next_batch (user_id, next_batch, filter_id) VALUES ($1, $2, $3)"
|
||||
|
|
Loading…
Reference in a new issue