o m=8j=@sUddlZddlZddlZddlmZmZddlmZddlm Z m Z m Z m Z m Z ejdkr5ddlmZnddlmZeeedd fZzzddlZWn eyWddlZYnwd ZWn eyfd ZYnwzejd krtdd lmZndd lmZd ZWn eyd ZYnwdZdZdZGddde ZGddde Z Gddde Z!Gddde d dZ"GdddZ#e#eZ$e#e%d<de!ddfddZ&  d.d e'dBd!e(de)fd"d#Z*Gd$d%d%eZ+Gd&d'd'Z,Gd(d)d)e+Z-Gd*d+d+e+Z.Gd,d-d-e+Z/dS)/N)ABCabstractmethod)Executor)AnyFinalProtocol TypedDictcast) )Buffer)Unionzmemoryview[int]zmemoryview[bytes]TF)r )ZstdDecompressoric@s2eZdZdedefddZd dedefddZd S) ZLibCompressObjProtocoldatareturncCdSNselfrrr`/home/nk/hobo-godmode/plappi-mvp/.venv/lib/python3.10/site-packages/aiohttp/compression_utils.pycompress,z ZLibCompressObjProtocol.compress.modecCrrrrrrrrflush-rzZLibCompressObjProtocol.flushN.)__name__ __module__ __qualname__r bytesrintrrrrrr+src@sneZdZddededefddZddedefdd Zede fd d Z edefd d Z edefddZ dS)ZLibDecompressObjProtocol.r max_lengthrcCrrrrrr&rrr decompress1rz$ZLibDecompressObjProtocol.decompresslengthcCrrrrr)rrrr2rzZLibDecompressObjProtocol.flushcCrrrrrrreof4zZLibDecompressObjProtocol.eofcCrrrr+rrrunconsumed_tail7r-z)ZLibDecompressObjProtocol.unconsumed_tailcCrrrr+rrr unused_data:r-z%ZLibDecompressObjProtocol.unused_dataNr) r r!r"r r$r#r(rpropertyboolr,r.r/rrrrr%0sr%c@seZdZUeed<eed<eed<eed<eed<      ddeded ed ed ed ed BdefddZ dd ed edefddZ dded edede fddZ dd ededede fddZ d S)ZLibBackendProtocol MAX_WBITS Z_FULL_FLUSH Z_SYNC_FLUSH Z_BEST_SPEEDZ_FINISH.levelmethodwbitsmemLevelstrategyzdictNrcCrrr)rr8r9r:r;r<r=rrr compressobjEszZLibBackendProtocol.compressobjcCrrr)rr:r=rrr decompressobjNz!ZLibBackendProtocol.decompressobjrcCrrr)rrr8r:rrrrRr@zZLibBackendProtocol.compressbufsizecCrrr)rrr:rArrrr(Ur@zZLibBackendProtocol.decompress)......)..) r r!r"r$__annotations__r rr>r%r?r#rr(rrrrr2>sn     r2c@s&eZdZUeed<eed<eed<dS)CompressObjArgsr:r<r8N)r r!r"r$rBrrrrrCZs  rC)totalc@seZdZdefddZedefddZedefddZ edefd d Z edefd d Z edefd dZ edefddZ dededefddZdededefddZdedededefddZdedededefddZdedefddZdS) ZLibBackendWrapper _zlib_backendcCs ||_dSr)rF)rrFrrr__init__a zZLibBackendWrapper.__init__rcCst|jddS)Nr undefinedgetattrrFr+rrrnamedszZLibBackendWrapper.namecC|jjSr)rFr3r+rrrr3hzZLibBackendWrapper.MAX_WBITScCrMr)rFr4r+rrrr4lrNzZLibBackendWrapper.Z_FULL_FLUSHcCrMr)rFr5r+rrrr5prNzZLibBackendWrapper.Z_SYNC_FLUSHcCrMr)rFr6r+rrrr6trNzZLibBackendWrapper.Z_BEST_SPEEDcCrMr)rFr7r+rrrr7xrNzZLibBackendWrapper.Z_FINISHargskwargscO|jj|i|Sr)rFr>rrOrPrrrr>|zZLibBackendWrapper.compressobjcOrQr)rFr?rRrrrr?rSz ZLibBackendWrapper.decompressobjrcO|jj|g|Ri|Sr)rFrrrrOrPrrrrzZLibBackendWrapper.compresscOrTr)rFr(rUrrrr(rVzZLibBackendWrapper.decompressattrnamecCs t|j|SrrJ)rrWrrr __getattr__ zZLibBackendWrapper.__getattr__N)r r!r"r2rGr0strrLr$r3r4r5r6r7rrr>r%r?r r#rr(rXrrrrrE`s&rE ZLibBackendnew_zlib_backendrcCs |t_dSr)r[rF)r\rrrset_zlib_backendrHr]encodingsuppress_deflate_headercCs$|dkr dtjS|rtj StjS)Ngzip)r[r3)r^r_rrrencoding_to_modes rbc@szeZdZdefdedBdedBfddZeefde dede fd d Z efde dede fd d Z e edefd dZdS)DecompressionBaseHandlerNexecutormax_sync_chunk_sizecCs||_||_dS)z&Base class for decompression handlers.N) _executor_max_sync_chunk_sizerrdrerrrrGs z!DecompressionBaseHandler.__init__rr&rcCdSDecompress the given data.Nrr'rrrdecompress_syncrz(DecompressionBaseHandler.decompress_synccsD|jdurt||jkrt|j|j||IdHS|||Srj)rglenasyncioget_event_looprun_in_executorrfrlr'rrrr(s   z#DecompressionBaseHandler.decompresscCri)z7Return True if more output is available by passing b"".Nrr+rrrdata_availablerz'DecompressionBaseHandler.data_available)r r!r"MAX_SYNC_CHUNK_SIZErr$rGrZLIB_MAX_LENGTH_UNLIMITEDr r#rlr(r0r1rqrrrrrcs8     rcc@seZdZddddddefdedBdededBdedBdedBdedBd edBfd d Zd e d e fddZ d e d e fddZ ddedBd e fddZ dS)ZLibCompressorNFr^r_r8r:r<rdrec Csv||_||_|durt||n||_ttj|_i}|j|d<|dur'||d<|dur/||d<|jjdi||_dS)Nr:r<r8r) rfrgrb_moderEr[rFr> _compressor) rr^r_r8r:r<rdrerPrrrrGs    zZLibCompressor.__init__rrcCs |j|Sr)rvrrrrr compress_syncrYzZLibCompressor.compress_synccsF|jduo t||jk}|rt|j|jj|IdHS||S)aCompress the data and returned the compressed bytes. Note that flush() must be called after the last call to compress() If the data size is large than the max_sync_chunk_size, the compression will be done in the executor. Otherwise, the compression will be done in the event loop. **WARNING: This method is NOT cancellation-safe when used with flush().** If this operation is cancelled, the compressor state may be corrupted. The connection MUST be closed after cancellation to avoid data corruption in subsequent compress operations. For cancellation-safe compression (e.g., WebSocket), the caller MUST wrap compress() + flush() + send operations in a shield and lock to ensure atomicity. N) rgrmrnget_running_looprprfrvrrw)rrshould_use_executorrrrrs    zZLibCompressor.compressrcCs|j|dur |S|jjS)a Flush the compressor synchronously. **WARNING: This method is NOT cancellation-safe when called after compress().** The flush() operation accesses shared compressor state. If compress() was cancelled, calling flush() may result in corrupted data. The connection MUST be closed after compress() cancellation. For cancellation-safe compression (e.g., WebSocket), the caller MUST wrap compress() + flush() + send operations in a shield and lock to ensure atomicity. N)rvrrFr7rrrrrs  zZLibCompressor.flushr)r r!r"rrrZr1r$rrGr r#rwrrrrrrrts4 rtc seZdZdddefdedBdededBdedBffdd Ze fd e d ed e fd d Z dded e fddZ ed efddZed efddZZS)ZLibDecompressorNFr^r_rdrecsJtj||dt|||_ttj|_|jj|jd|_d|_ d|_ dS)Nrdrer:F) superrGrbrurEr[rFr? _decompressor _last_empty_pending_unused_data)rr^r_rdre __class__rrrG s    zZLibDecompressor.__init__rr&rcCs|jdur|jt|}d|_|j|jj||}|dk|_|jjr^|jjr^|jj}|jj |j d|_|t krF|t |8}|dkrF||_n|j||}|dk|_||7}|jjr^|jjs'|jjrr|j |jj krr|jj |j d|_|S)Nr|r)rr#r~r(r.rr,r/rFr?rursrmr3)rrr&resultunusedchunkrrrrls,     z ZLibDecompressor.decompress_syncrr)cCs|dkr |j|S|jSNr)r~rr*rrrr?s  zZLibDecompressor.flushcCst|jjp|j p|jduSr)r1r~r.rrr+rrrrqFs zZLibDecompressor.data_availablecCrMr)r~r,r+rrrr,NrNzZLibDecompressor.eof)r)r r!r"rrrZr1rr$rGrsr r#rlrr0rqr, __classcell__rrrrrz s6 $rzcreZdZdefdedBdedBddffdd Zefdedede fd d Z de fd d Z e de fd dZZS)BrotliDecompressorNrdrercs0tstdt|_d|_tj||ddS)z)Decompress data using the Brotli library.zIThe brotli decompression is not available. Please install `Brotli` moduleFr{N) HAS_BROTLI RuntimeErrorbrotli Decompressor_objrr}rGrhrrrrGWs zBrotliDecompressor.__init__rr&cCs|t|jdr|tkrtt|j|}n#tt|j||}n|tkr-tt|j|}n tt|j||}|dk|_|S)rkr(r)hasattrrrsr r#r(processr)rrr&rrrrrlfs  z"BrotliDecompressor.decompress_synccCs t|jdrtt|jSdS)zFlush the decompressor.rr)rrr r#rr+rrrrxs zBrotliDecompressor.flushcCs|j o |j Sr)r is_finishedrr+rrrrq~sz!BrotliDecompressor.data_availabler r!r"rrrr$rGrsr r#rlrr0r1rqrrrrrrSs* rcr)ZSTDDecompressorNrdrercs.tstdt|_d|_tj||ddS)NzOThe zstd decompression is not available. Please install `backports.zstd` moduler{)HAS_ZSTDrrrrr}rGrhrrrrGszZSTDDecompressor.__init__rr&cCs|tkrtn|}|jdur|j|}d|_|j||}|jjrO|jjrO|jj}t|_|tkr>|t|8}|dkr>||_n||j||7}|jjrO|jjs$|jjrWt|_|Sr) rsZSTD_MAX_LENGTH_UNLIMITEDrrr(r,r/rrm)rrr&zstd_max_lengthrr/rrrrls*    z ZSTDDecompressor.decompress_synccCri)Nrrr+rrrrr-zZSTDDecompressor.flushcCs|jj o |jj p|jduSr)r needs_inputr,rr+rrrrqszZSTDDecompressor.data_availablerrrrrrs* &r)NF)0rnsyszlibabcrrconcurrent.futuresrtypingrrrrr version_infocollections.abcr r r# bytearray brotlicffir ImportErrorrcompression.zstdrbackports.zstdrrrrsrrr%r2rCrEr[rBr]rZr1r$rbrcrtrzrrrrrrsh           -  #KG0