lightning/tests/plugins/dblog.py

#!/usr/bin/env python3
"""This plugin is used to check that db_write calls are working correctly.
"""
from pyln.client import Plugin, RpcError
import sqlite3

plugin = Plugin()
plugin.sqlite_pre_init_cmds = []
plugin.initted = False


@plugin.init()
def init(configuration, options, plugin):
    if not plugin.get_option('dblog-file'):
        raise RpcError("No dblog-file specified")
    plugin.conn = sqlite3.connect(plugin.get_option('dblog-file'),
                                  isolation_level=None)
    plugin.log("replaying pre-init data:")
    plugin.conn.execute("PRAGMA foreign_keys = ON;")

    print(plugin.sqlite_pre_init_cmds)

    plugin.conn.execute("BEGIN TRANSACTION;")

    for c in plugin.sqlite_pre_init_cmds:
        plugin.conn.execute(c)
        plugin.log("{}".format(c))

    plugin.conn.execute("COMMIT;")

    plugin.initted = True
    plugin.log("initialized {}".format(configuration))


@plugin.hook('db_write')
def db_write(plugin, writes, **kwargs):
    if not plugin.initted:
        plugin.log("deferring {} commands".format(len(writes)))
        plugin.sqlite_pre_init_cmds += writes
    else:
        print(writes)
        plugin.conn.execute("BEGIN TRANSACTION;")

        for c in writes:
            plugin.conn.execute(c)
            plugin.log("{}".format(c))

        plugin.conn.execute("COMMIT;")

    return {"result": "continue"}


plugin.add_option('dblog-file', None, 'The db file to create.')
plugin.run()