Save and load binary np array into a BLOB database

np_duckdb_blob.py

import binascii
import duckdb
import io
import numpy as np


def makeblob(arr):
    f = io.BytesIO()
    np.save(f, arr)
    f.seek(0)
    blob = f.read()
    return blob.hex()


def readblob(blob):
    f = io.BytesIO()
    f.write(blob)
    f.seek(0)
    data = np.load(f)
    return binascii.unhexlify(data)


duckdb_conn = duckdb.connect(
    database="duck_database.db"
)

duckdb_conn.execute("""CREATE TABLE IF NOT EXISTS binstates (
    blobby BLOB,
)
""")

array_obj = np.array([1, 2, 3, 4, 5])
query = f"""INSERT INTO binstates VALUES ('{blob_obj}')"""
duckdb_conn.execute(query)

result = duckdb_conn.execute("SELECT * FROM binstates;").fetchall()
blob_rec = binascii.unhexlify(result[0][0])

blob_obj_rec = readblob(blob_rec)

# blob_obj == blob_obj_rec