68 lines
1.9 KiB
Python
68 lines
1.9 KiB
Python
import aiosqlite
|
|
import logging
|
|
|
|
migrations = [
|
|
"""
|
|
CREATE TABLE deleted_items (
|
|
id INTEGER PRIMARY KEY,
|
|
timestamp INTEGER NOT NULL,
|
|
item TEXT NOT NULL
|
|
);
|
|
""",
|
|
"""
|
|
CREATE INDEX deleted_items_timestamps ON deleted_items(timestamp);
|
|
""",
|
|
"""
|
|
CREATE TABLE reminders (
|
|
id INTEGER PRIMARY KEY,
|
|
remind_timestamp INTEGER NOT NULL,
|
|
created_timestamp INTEGER NOT NULL,
|
|
reminder TEXT NOT NULL,
|
|
expired INTEGER NOT NULL,
|
|
extra TEXT NOT NULL
|
|
);
|
|
""",
|
|
"""
|
|
CREATE TABLE telephone_config (
|
|
id TEXT PRIMARY KEY,
|
|
guild_id INTEGER NOT NULL,
|
|
channel_id INTEGER NOT NULL UNIQUE,
|
|
webhook TEXT
|
|
);
|
|
""",
|
|
"""
|
|
CREATE TABLE calls (
|
|
from_id TEXT NOT NULL REFERENCES telephone_config(id) UNIQUE,
|
|
to_id TEXT NOT NULL REFERENCES telephone_config(id),
|
|
start_time INTEGER NOT NULL
|
|
);
|
|
"""
|
|
]
|
|
|
|
async def execute_fetchone(self, sql, params=None):
|
|
if params == None: params = ()
|
|
return await self._execute(self._fetchone, sql, params)
|
|
|
|
def _fetchone(self, sql, params):
|
|
cursor = self._conn.execute(sql, params)
|
|
return cursor.fetchone()
|
|
|
|
async def init(db_path):
|
|
db = await aiosqlite.connect(db_path)
|
|
await db.execute("PRAGMA foreign_keys = ON")
|
|
|
|
db.row_factory = aiosqlite.Row
|
|
aiosqlite.Connection._fetchone = _fetchone
|
|
aiosqlite.Connection.execute_fetchone = execute_fetchone
|
|
|
|
version = (await (await db.execute("PRAGMA user_version")).fetchone())[0]
|
|
for i in range(version, len(migrations)):
|
|
await db.executescript(migrations[i])
|
|
# Normally interpolating like this would be a terrible idea because of SQL injection.
|
|
# However, in this case there is not an obvious alternative (the parameter-based way apparently doesn't work)
|
|
# and i + 1 will always be an integer anyway
|
|
await db.execute(f"PRAGMA user_version = {i + 1}")
|
|
await db.commit()
|
|
logging.info(f"Migrated DB to schema {i + 1}")
|
|
|
|
return db |