Skip to content

Instantly share code, notes, and snippets.

@lispyclouds

lispyclouds/pod-bb-python-sqlite.py

Created May 5, 2020 12:27
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save lispyclouds/1b5125b1d7a8aba2741f86aade7e8359 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save lispyclouds/1b5125b1d7a8aba2741f86aade7e8359 to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. lispyclouds created this gist May 5, 2020.
    58 changes: 58 additions & 0 deletions pod-bb-python-sqlite.py
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,58 @@
    import json
    import sqlite3
    import sys

    from bcoding import bencode, bdecode


    def read():
    return dict(bdecode(sys.stdin.buffer))


    def write(obj):
    sys.stdout.buffer.write(bencode(obj))
    sys.stdout.flush()


    def debug(msg):
    with open("debug.log", "a") as f:
    f.write(str(msg) + "\n")


    def main():
    while True:
    msg = read()

    op = msg["op"]

    if op == "describe":
    write(
    {
    "format": "json",
    "vars": [{"namespace": "pod.sqlite", "name": "execute!"}],
    }
    )
    elif op == "invoke":
    var = msg["var"]
    id = msg["id"]
    args = json.loads(msg["args"])

    conn = sqlite3.connect("bb.db")
    c = conn.cursor()

    result = None

    if var == "pod.sqlite/execute!":
    try:
    result = c.execute(*args)
    except Exception as e:
    debug(e)

    write({"value": json.dumps(result.fetchall()), "id": id, "status": ["done"]})

    conn.commit()
    conn.close()


    if __name__ == "__main__":
    main()