#!/usr/bin/env python
# -*- coding=utf-8 -*-
from codernitydb3.database import Database
from codernitydb3.hash_index import HashIndex


class WithXIndex(HashIndex):

    def __init__(self, *args, **kwargs):
        kwargs["key_format"] = "I"
        super(WithXIndex, self).__init__(*args, **kwargs)

    def make_key_value(self, data):
        a_val = data.get("x")
        if a_val is not None:
            return a_val, None
        return None

    def make_key(self, key):
        return key


def main():
    db = Database("/tmp/tut2")
    db.create()
    x_ind = WithXIndex(db.path, "x")
    db.add_index(x_ind)

    for x in range(100):
        db.insert(dict(x=x))

    for y in range(100):
        db.insert(dict(y=y))

    x = db.get("x", 10, with_doc=True)
    print(x)


if __name__ == "__main__":
    main()