home

Menu
  • ripgrep search

datasette-copy-to-memory/datasette_copy_to_memory/__init__.py

from datasette import hookimpl
from datasette.utils import sqlite3
 
 
@hookimpl
def startup(datasette):
    plugin_config = datasette.plugin_config("datasette-copy-to-memory") or {}
    databases = plugin_config.get("databases")
    replace = plugin_config.get("replace")
 
    async def inner():
        for db in datasette.databases.values():
            if databases:
                if db.name not in databases:
                    continue
            if db.path:
                memory_name = "{}_memory".format(db.name)
                if replace:
                    memory_name = db.name
                    datasette.remove_database(db.name)
                else:
                    memory_name = "{}_memory".format(db.name)
                memory_db = datasette.add_memory_database(memory_name)
                await memory_db.execute("select 1 + 1")
                # Use a different in-memory database to co-ordinate the VACUUM INTO
                tmp = sqlite3.connect(":memory:", uri=True)
                tmp.execute("ATTACH DATABASE ? AS _copy_from", [db.path])
                tmp.execute(
                    "VACUUM _copy_from INTO ?",
                    ["file:{}?mode=memory&cache=shared".format(memory_name)],
                )
 
    return inner
 
Powered by Datasette