Nvim persistence service

nvim_persistence_service.py

import sqlite3, msgpack

class NvimPersistenceService(object):
    def __init__(self, vim):
        self.connection = sqlite3.connect('.nvim-persisted.db')
        with self.connection as c:
            c.execute('''
            CREATE TABLE IF NOT EXISTS nvim
            (
                key TEXT PRIMARY KEY,
                value BLOB
            )
            ''')

        put_fn = '\n'.join(['fun! Put(key, value)',
                            '  return rpcrequest(%d, "put", a:key, a:value)',
                            'endfun']) % vim.channel_id
        get_fn = '\n'.join(['fun! Get(key)',
                            '  return rpcrequest(%d, "get", a:key)',
                            'endfun']) % vim.channel_id
        del_fn = '\n'.join(['fun! Del(key)',
                            '  return rpcrequest(%d, "del", a:key)',
                            'endfun']) % vim.channel_id
        vim.command(put_fn)
        vim.command(get_fn)
        vim.command(del_fn)
 
 
    def put(self, key, value):
        stmt = 'REPLACE INTO nvim (key, value) VALUES (?, ?)'
        with self.connection as c:
            c.execute(stmt, (key, buffer(msgpack.packb(value))))
 

    def get(self, key):
        stmt = 'SELECT value FROM nvim WHERE key = ?'
        with self.connection as c:
            for row in c.execute(stmt, (key,)):
                return msgpack.unpackb(row[0])

 
    def delete(self, key):
        stmt = 'DELETE FROM nvim WHERE key = ?'
        with self.connection as c:
            c.execute(stmt, (key,))

 
    def on_plugin_teardown(self):
        self.connection.close()