&jNS>ddlZddlZddlZddlmZmZGddeZeZGddej Z e ej Z e j Ze jZe jZdZe ejejdfd Zd Zd Zd Zd ZdZGddeZGddeZdS)N)ffilibceZdZdZdS)errorz Raised whenever an error is encountered with compressing or decompressing data using brotlicffi. .. versionadded:: 0.5.1 N)__name__ __module__ __qualname____doc__P/usr/local/lib/hermes-agent/venv/lib/python3.11/site-packages/brotlicffi/_api.pyrr s  Dr rc<eZdZdZejZejZej Z dS)BrotliEncoderModezP Compression modes for the Brotli encoder. .. versionadded:: 0.5.0 N) rr r r rBROTLI_MODE_GENERICGENERICBROTLI_MODE_TEXTTEXTBROTLI_MODE_FONTFONTr r rrrs6%G  D  DDDr rctt}||}||S)z{ Decompress a complete Brotli-compressed string. :param data: A bytestring containing Brotli-compressed data. ) Decompressor decompressfinish)datads rrrSs0 A <<  DHHJJJ Kr ct||||}||tj}tj|jtjksJtj|jtjksJ|S)a Compress a string using Brotli. .. versionchanged:: 0.5.0 Added ``mode``, ``quality``, `lgwin``, ``lgblock``, and ``dictionary`` parameters. :param data: A bytestring containing the data to compress. :type data: ``bytes`` :param mode: The encoder mode. :type mode: :class:`BrotliEncoderMode` or ``int`` :param quality: Controls the compression-speed vs compression-density tradeoffs. The higher the quality, the slower the compression. The range of this value is 0 to 11. :type quality: ``int`` :param lgwin: The base-2 logarithm of the sliding window size. The range of this value is 10 to 24. :type lgwin: ``int`` :param lgblock: The base-2 logarithm of the maximum input block size. The range of this value is 16 to 24. If set to 0, the value will be set based on ``quality``. :type lgblock: ``int`` :returns: The compressed bytestring. :rtype: ``bytes`` )modequalitylgwinlgblock) Compressor _compressrBROTLI_OPERATION_FINISHBrotliEncoderIsFinished_encoder BROTLI_TRUEBrotliEncoderHasMoreOutput BROTLI_FALSE)rrrr r! compressorcompressed_datas rcompressr,_sP  J !**41LMMO  &z': ; ;s N N N N &z':;;s?OOOOO r cf t|}dS#t$rtd|zwxYw)z* Validate that the mode is valid. z%s is not a valid encoder modeN)r ValueErrorrvals r_validate_moder1sG<$$ <<<4s:;;;` if it is invalid. zUnexpected parameter!zError setting parameter %s: %dN) rBrotliEncoderSetParameterBROTLI_PARAM_MODEr1BROTLI_PARAM_QUALITYr5BROTLI_PARAM_LGWINr9BROTLI_PARAM_LGBLOCKr< RuntimeErrorr'r)encoder parameterparameter_namer0rcs r_set_parameterrHs &w 3 ? ?BC)))s c. . .# c, , , c. . .#2333  S_ ,/D D   r cZeZdZdZdZdZeejej dfdZ dZ dZ e Z dZdZdS) r"a An object that allows for streaming compression of data using the Brotli compression algorithm. .. versionadded:: 0.5.0 :param mode: The encoder mode. :type mode: :class:`BrotliEncoderMode` or ``int`` :param quality: Controls the compression-speed vs compression-density tradeoffs. The higher the quality, the slower the compression. The range of this value is 0 to 11. :type quality: ``int`` :param lgwin: The base-2 logarithm of the sliding window size. The range of this value is 10 to 24. :type lgwin: ``int`` :param lgblock: The base-2 logarithm of the maximum input block size. The range of this value is 16 to 24. If set to 0, the value will be set based on ``quality``. :type lgblock: ``int`` :param dictionary: A pre-set dictionary for LZ77. Please use this with caution: if a dictionary is used for compression, the same dictionary **must** be used for decompression! :type dictionary: ``bytes`` Nrctj|_tjt jt jt j}|stdt j|tj }t|tj d|t|tj d|t|tj d|t|tjd|||_dS)Nz"Unable to allocate Brotli encoder!rrr r!) threadingRLocklockrBrotliEncoderCreateInstancerNULLrCgcBrotliEncoderDestroyInstancerHr?r@rArBr&)selfrrr r!encs r__init__zCompressor.__init__s O%% - Hch   ECDD DfS#:;; sC164@@@sC4iIIIsC2GUCCCsC4iIII r c n|jdstd tt jt |t |dz zdz}tjd}||d<tjd|d}tjd |}tjdt |}tjd|}tjd |} tj |j ||| ||tj } |j n#|j wxYw| tjkrtd |drJ||dz } tj|| d d S) z This private method compresses some data in a given mode. This is used because almost all of the code uses the exact same setup. It wouldn't have to, but it doesn't hurt at all. Fblocking6Concurrently sharing Compressor objects is not allowedi(size_t *r uint8_t [] uint8_t **z#Error encountered compressing data.N)rMacquirerintmathceillenrnewrBrotliEncoderCompressStreamr&rOreleaser'buffer) rRr operationoriginal_output_size available_out output_bufferptr_to_output_buffer input_size input_bufferptr_to_input_bufferrGsize_of_outputs rr#zCompressor._compresss y  % 00 JHJJ J $' #d))s4yyA~6>??$$  GJ//M3M! GL-2BCCM#&7<#G#G SYY77J7<66L"%', "E"E 0 #$B I      DI        =>> >a=   - a0@@z-88;;s DEE#cB||tjS)a> Incrementally compress more data. :param data: A bytestring containing data to compress. :returns: A bytestring containing some compressed data. May return the empty bytestring if not enough data has been inserted into the compressor to create the output yet. )r#rBROTLI_OPERATION_PROCESS)rRrs rr,zCompressor.compress8s~~dC$@AAAr c6|jdstd |dtjg}t j|jtjkrZ| |dtjt j|jtjkZ|j n#|j wxYwd |S)z Flush the compressor. This will emit the remaining output data, but will not destroy the compressor. It can be used, for example, to ensure that given chunks of content will decompress immediately. FrVrXr ) rMr]rr#rBROTLI_OPERATION_FLUSHr(r&r'appendrdjoinrRchunkss rflushzCompressor.flushEs y  % 00 JHJJ J nnS#*DEEFF24=AAO$$ dnnS#2LMMNNN24=AAO$$ I      DI      xxs B"C((Dc|jdstd g}tj|jtjkrZ||dtj tj|jtjkZ|j n#|j wxYwd |S)z Finish the compressor. This will emit the remaining output data and transition the compressor to a completed state. The compressor cannot be used again after this point, and must be replaced. FrVrXr ) rMr]rrr%r&r)rsr#r$rdrtrus rrzCompressor.finishXs y  % 00 JHJJ J F/ >>$%% dnnS#2MNNOOO/ >>$%% I      DI      xxs BC C$)rr r r _dictionary_dictionary_size DEFAULT_MODErBROTLI_DEFAULT_QUALITYBROTLI_DEFAULT_WINDOWrTr#r,processrwrr r rr"r"s8K#30 ,(<(<(>!! I      DI       s AB B&)r )N)rr r r ryrzrrT staticmethodrrrr~rwrrrr r rrrks K    &&\&     4NNN`G   "   !!!!!r r)r_enumrK _brotlicffirr ExceptionrErrorIntEnumrBROTLI_DEFAULT_MODEr{r MODE_GENERICr MODE_TEXTr MODE_FONTrr|r}r,r1r5r9r<rHobjectr"rr r rrs !!!!!!!!     I             (! !899 !(   "   "    /, 3333l<<<   PPP      8P P P P P P P P fppppp6pppppr