Source code for dl_utils.data.lmdb

# -*- coding: utf-8 -*-
# @Time    : 4/20/23
# @Author  : Yaojie Shen
# @Project : Deep-Learning-Utils
# @File    : lmdb.py

__all__ = ["JsonLmdb"]

import json

from lmdbm import Lmdb




class JsonLmdb(Lmdb):
    """
    https://pypi.org/project/lmdbm/
    """

    def _pre_key(self, value):
        return value.encode("utf-8")

    def _post_key(self, value):
        return value.decode("utf-8")

    def _pre_value(self, value):
        return json.dumps(value).encode("utf-8")

    def _post_value(self, value):
        return json.loads(value.decode("utf-8"))