g{'ddlZddlZddlmZmZmZmZmZm Z m Z m Z m Z m Z mZmZmZmZmZmZddlmZGddej*ZdZGddej0ZGdd ej4ZGd d ej8Zd Zd ZddddZ djCee _"ddddZ#ddddZ$djCee$_"dddddZ%ddddZ&dddZ'y)N)IpcReadOptionsIpcWriteOptions ReadStats WriteStatsMessage MessageReaderRecordBatchReader_ReadPandasMixinMetadataVersion read_messageread_record_batch read_schema read_tensor write_tensorget_record_batch_sizeget_tensor_sizeceZdZdZddddZy)RecordBatchStreamReadera Reader for the Arrow streaming binary format. Parameters ---------- source : bytes/buffer-like, pyarrow.NativeFile, or file-like Python object Either an in-memory buffer, or a readable file object. If you want to use memory map use MemoryMappedFile as source. options : pyarrow.ipc.IpcReadOptions Options for IPC deserialization. If None, default values will be used. memory_pool : MemoryPool, default None If None, default memory pool is used. Noptions memory_poolcBt|}|j|||y)Nr _ensure_default_ipc_read_options_open)selfsourcerrs @/var/www/openai/venv/lib/python3.12/site-packages/pyarrow/ipc.py__init__z RecordBatchStreamReader.__init__2s27; 67 D__name__ __module__ __qualname____doc__rr rrr"s +/DEr raNParameters ---------- sink : str, pyarrow.NativeFile, or file-like Python object Either a file path, or a writable file object. schema : pyarrow.Schema The Arrow schema for data to be written to the file. use_legacy_format : bool, default None Deprecated in favor of setting options. Cannot be provided with options. If None, False will be used unless this default is overridden by setting the environment variable ARROW_PRE_0_15_IPC_FORMAT=1 options : pyarrow.ipc.IpcWriteOptions Options for IPC serialization. If None, default values will be used: the legacy format will not be used unless overridden by setting the environment variable ARROW_PRE_0_15_IPC_FORMAT=1, and the V5 metadata version will be used unless overridden by setting the environment variable ARROW_PRE_1_0_METADATA_VERSION=1.c<eZdZdjeZddddZy)RecordBatchStreamWriterz0Writer for the Arrow streaming binary format {}Nuse_legacy_formatrcDt||}|j|||yNr_get_legacy_format_defaultrrsinkschemar*rs rrz RecordBatchStreamWriter.__init__S!,->H 4 1r r"r#r$format_ipc_writer_class_docr%rr&r rr(r(Ns# f "# ;?2r r(c eZdZdZdddddZy)RecordBatchFileReadera Class for reading Arrow record batch data from the Arrow binary file format Parameters ---------- source : bytes/buffer-like, pyarrow.NativeFile, or file-like Python object Either an in-memory buffer, or a readable file object. If you want to use memory map use MemoryMappedFile as source. footer_offset : int, default None If the file is embedded in some larger file, this is the byte offset to the very end of the file data options : pyarrow.ipc.IpcReadOptions Options for IPC serialization. If None, default values will be used. memory_pool : MemoryPool, default None If None, default memory pool is used. NrcDt|}|j||||y)N footer_offsetrrr)rrr;rrs rrzRecordBatchFileReader.__init__ks&27; 6"   =r Nr!r&r rr8r8Xs$=d!=r r8c<eZdZdjeZddddZy)RecordBatchFileWriterz1Writer to create the Arrow binary file format {}Nr)cDt||}|j|||yr,r.r0s rrzRecordBatchFileWriter.__init__xr3r r4r&r rr>r>rs# f "# ;?2r r>c| | td|r5t|ts#tdj t ||St j}|2tttjjdd}tttjjddrt j}t||S)Nz8Can provide at most one of options and use_legacy_formatz expected IpcWriteOptions, got {}ARROW_PRE_0_15_IPC_FORMAT0ARROW_PRE_1_0_METADATA_VERSION)r*metadata_version) ValueError isinstancer TypeErrorr5typer V5boolintosenvirongetV4)r*rrDs rr/r/}s$)< FH H '?3>#VDM24 4&)) RZZ^^$?EF G  C ?E FG*-- ->,< >>r c|r3t|ts#tdjt ||xs tS)Nzexpected IpcReadOptions, got {})rFrrGr5rHr-s rrrs=z'>: - 4 4T'] C    &n&&r r)c t||||SNr))r(r1r2r*rs r new_streamrTs "45F+2 44r zCreate an Arrow columnar IPC stream writer instance {} Returns ------- writer : RecordBatchStreamWriter A writer for the given sink rct|||S)a Create reader for Arrow streaming format. Parameters ---------- source : bytes/buffer-like, pyarrow.NativeFile, or file-like Python object Either an in-memory buffer, or a readable file object. options : pyarrow.ipc.IpcReadOptions Options for IPC serialization. If None, default values will be used. memory_pool : MemoryPool, default None If None, default memory pool is used. Returns ------- reader : RecordBatchStreamReader A reader for the given source r)r)rrrs r open_streamrVs& #67/: <rSs rnew_filerXs v3D)0 22r zCreate an Arrow columnar IPC file writer instance {} Returns ------- writer : RecordBatchFileWriter A writer for the given sink c t||||S)a Create reader for Arrow file format. Parameters ---------- source : bytes/buffer-like, pyarrow.NativeFile, or file-like Python object Either an in-memory buffer, or a readable file object. footer_offset : int, default None If the file is embedded in some larger file, this is the byte offset to the very end of the file data. options : pyarrow.ipc.IpcReadOptions Options for IPC serialization. If None, default values will be used. memory_pool : MemoryPool, default None If None, default memory pool is used. Returns ------- reader : RecordBatchFileReader A reader for the given source r:)r8)rr;rrs r open_filerZs, !m[ 22r nthreadspreserve_indexc8tjj|||}tj}tj||j 5}|j |ddd|jS#1swY|jSxYw)a Serialize a pandas DataFrame into a buffer protocol compatible object. Parameters ---------- df : pandas.DataFrame nthreads : int, default None Number of threads to use for conversion to Arrow, default all CPUs. preserve_index : bool, default None The default of None will store the index as a column, except for RangeIndex which is stored as metadata only. If True, always preserve the pandas index data as a column. If False, no index information is saved and the result will have a default RangeIndex. Returns ------- buf : buffer An object compatible with the buffer protocol. r[N)pa RecordBatch from_pandasBufferOutputStreamr(r2 write_batchgetvalue)dfr\r]batchr1writers rserialize_pandasrhsz( NN & &rH6D ' FE "D # #D%,, 765! 8 ==? 8 ==?s BBT use_threadsctj|}tj|5}|j}dddj |S#1swYxYw)aDeserialize a buffer protocol compatible object into a pandas DataFrame. Parameters ---------- buf : buffer An object compatible with the buffer protocol. use_threads : bool, default True Whether to parallelize the conversion using multiple threads. Returns ------- df : pandas.DataFrame The buffer deserialized as pandas DataFrame Nri)r_ BufferReaderrread_all to_pandas)bufrj buffer_readerreadertables rdeserialize_pandasrs sMOOC(M # #M 2f! 3 ??{? 33 3 2s AAr<)(rLpyarrowr_ pyarrow.librrrrrrr r r r r rrrrrlib_RecordBatchStreamReaderrr6_RecordBatchStreamWriterr(_RecordBatchFileReaderr8_RecordBatchFileWriterr>r/rrTr5r%rVrXrZrhrsr&r rr{s( AAAAAEc::E*).2c::2=C66=42C662>('374  F ! $(T<.15d2  F ! 2Tt26&*$8,04r