parquet.pyx

cdef cppclass PyDecryptionKeyRetriever(CDecryptionKeyRetriever):
    object pyobject

    __init__(object pyobject):
        this.pyobject = pyobject

    c_string GetKey(const c_string& key_metadata) const:
        h = cp.PyGILState_Ensure()
        try:
            if callable(this.pyobject):
                return <c_string> tobytes(this.pyobject(key_metadata))
            elif hasattr(this.pyobject, "get_key"):
                return <c_string> tobytes(this.pyobject.get_key(key_metadata))
            else:
                raise RuntimeError(
                    "KeyRetriever must be callable or implement get_key")
        finally:
            cp.PyGILState_Release(h)