#-*- coding:utf-8 -*-
from hashlib import md5
from CodernityDB.hash_index import HashIndex
class ProjectIndex(HashIndex):
    def __init__(self, *args, **kwargs):
        kwargs['key_format'] = '16s'
        super(ProjectIndex, self).__init__(*args, **kwargs)
    def make_key_value(self, data):
        if data['_t'] == 'project':
            return md5(data['name'].encode('utf-8')).digest(), None
    def make_key(self, key):
        return md5(key).digest()
class TaskIndex(HashIndex):
    def __init__(self, *args, **kwargs):
        kwargs['key_format'] = '16s'
        super(TaskIndex, self).__init__(*args, **kwargs)
    def make_key_value(self, data):
        if data['_t'] == 'task':
            return md5(data['project'].encode('utf-8')).digest(), None
    def make_key(self, key):
        return md5(key).digest()