Function write_encrypted_parquet

python/pyarrow/tests/parquet/test_encryption.py:182–194

(path, table, encryption_config,
                            kms_connection_config, crypto_factory)

def write_encrypted_parquet(path, table, encryption_config,
                            kms_connection_config, crypto_factory):
    if encryption_config.internal_key_material:
        file_encryption_properties = crypto_factory.file_encryption_properties(
            kms_connection_config, encryption_config)
    else:
        file_encryption_properties = crypto_factory.file_encryption_properties(
            kms_connection_config, encryption_config, path)
    assert file_encryption_properties is not None
    with pq.ParquetWriter(
            path, table.schema,
            encryption_properties=file_encryption_properties) as writer:
        writer.write_table(table)


def read_encrypted_parquet(path, decryption_config,