存储.py

import struct
import hashlib
from typing import MutableMapping

import brotli
from rimo_storage import 超dict

from 类 import 阵


def dump(o: 阵) -> bytes:
    c = [*zip(*o)] or ([], [])
    n = len(c[0])
    吸1 = [x.encode('utf8') for x in c[1]]
    字符串长度 = [len(s) for s in 吸1]
    文hint = ''.join([f'{x}s' for x in 字符串长度])
    nz = struct.pack('i', n)
    内容 = struct.pack(f'{n}h{n}e{文hint}', *字符串长度, *c[0], *吸1)
    z = nz+内容
    return z


def load(b: bytes) -> 阵:
    n = struct.unpack('i', b[:4])[0]
    字符串长度 = struct.unpack(f'{n}h', b[4:4+n*2])
    吸0 = struct.unpack(f'{n}e', b[4+n*2:4+n*4])
    文hint = ''.join([f'{x}s' for x in 字符串长度])
    吸1 = struct.unpack(文hint, b[4+n*4:])
    吸1 = [x.decode('utf8') for x in 吸1]
    return [*zip(吸0, 吸1)]


empty = dump([])
c = lambda x: brotli.compress(x, quality=4)
d = lambda x: brotli.decompress(x) if x != b'' else empty


def 索引空间(path) -> MutableMapping[str, 阵]:
    return 超dict(path, compress=(c, d), serialize=(dump, load))


class 融合之门(MutableMapping):
    def __init__(self, path):
        self.d = 超dict(path, compress=(c, d))

    def __getitem__(self, k):
        真k = hashlib.sha224(k.encode('utf8')).hexdigest()[:5]
        for kk, vv in self.d[真k]:
            if kk == k:
                return vv
        else:
            raise KeyError(k)

    def __setitem__(self, k, v):
        真k = hashlib.sha224(k.encode('utf8')).hexdigest()[:5]
        if 真k not in self.d:
            self.d[真k] = [(k, v)]
        else:
            原 = self.d[真k]
            for i, (kk, vv) in enumerate(原):
                if kk == k:
                    原[i] = kk, v
                    break
            else:
                原.append((k, v))
            self.d[真k] = 原

    __iter__=__len__=__delitem__=lambda: 0/0