def __init__(self, path, flags):
if self._env is not None or self._db is not None:
raise RuntimeError("object was already initialized")
# initialize environment
env = lib.sp_env()
if env == ffi.NULL:
raise MemoryError("could not allocate environment")
# set db directory and flags
_flags = ffi.cast("int", flags)
_path = ffi.new("char[]", path)
r = lib.sp_ctl(env, lib.SPDIR, _flags, _path)
if r != 0:
error = lib.sp_error(env)
lib.sp_destroy(env)
raise SophiaError(ffi.string(error))
# open db
db = lib.sp_open(env)
if db == ffi.NULL:
error = lib.sp_error(env)
lib.sp_destroy(env)
raise SophiaError(ffi.string(error))
self._env = env
self._db = db
def get(self, key):
if self._closed:
raise RuntimeError("db is closed")
if not isinstance(key, bytes):
raise TypeError("byte string objects are supported")
_key = ffi.new("char[]", key)
_value = ffi.new("void**")
_value_size = ffi.new("size_t*")
r = lib.sp_get(self._db, _key, ffi.sizeof(_key), _value, _value_size)
if r == -1:
self._raise_error()
elif r == 0:
raise KeyError("%s not found" % key)
value = ffi.string(ffi.cast("char*", _value[0]), _value_size[0])
libc.free(_value[0])
return value
def _raise_error(self):
error = lib.sp_error(self._db)
raise SophiaError(ffi.string(error))