;xj IZUddlZddlZddlZddlmZmZddlmZddlm Z m Z m Z m Z m Z mZmZejdkrddlmZnddlmZeeedd fZ ddlZn#e$rddlZYnwxYwd Zn #e$rd ZYnwxYw ejd krdd lmZndd lmZd Zn #e$rd ZYnwxYwdZdZdZ dZ!dZ"dZ#Gdde$Z%Gdde Z&Gdde Z'Gdde Z(Gdde d Z)GddZ*e*eZ+e*e,d <d!e(d"dfd#Z- d8d$e.dzd%e/d"e0fd&Z1Gd'd(e Z2ed)e2*Z3Gd+d,eZ4Gd-d.e4e e3Z5Gd/d0Z6Gd1d2e5e'Z7Gd3d4e4Z8Gd5d6e5d7Z9dS)9N)ABCabstractmethod)Executor)AnyFinalGenericProtocol TypedDictTypeVarcast) )Buffer)Unionzmemoryview[int]zmemoryview[bytes]TF)r ZstdDecompressori@iiceZdZdZdS)TooManyMembersErrorz:A stream concatenated more members than the caller allows.N)__name__ __module__ __qualname____doc__JC:\Projects\Historical\venv\Lib\site-packages\aiohttp/compression_utils.pyrr6sDDDDrrc0eZdZdedefdZddedefdZdS) ZLibCompressObjProtocoldatareturncdSNrselfr!s rcompressz ZLibCompressObjProtocol.compress;r.modecdSr$rr&r)s rflushzZLibCompressObjProtocol.flush<r(rN.)rrrrbytesr'intr,rrrr r :s?2V2222255#55555555rr ceZdZd dededefdZd dedefdZede fdZ edefd Z edefd Z d S) ZLibDecompressObjProtocol.r! max_lengthr"cdSr$rr&r!r2s r decompressz$ZLibDecompressObjProtocol.decompress@r(rlengthcdSr$rr&r6s rr,zZLibDecompressObjProtocol.flushAr(rcdSr$rr&s reofzZLibDecompressObjProtocol.eofC3rcdSr$rr:s runconsumed_tailz)ZLibDecompressObjProtocol.unconsumed_tailFs(+rcdSr$rr:s r unused_dataz%ZLibDecompressObjProtocol.unused_dataI$'CrNr-) rrrrr/r.r5r,propertyboolr;r>r@rrrr1r1?sKKvK3KKKKK77C7U7777 TX ++++X+ 'U'''X'''rr1ceZdZUeed<eed<eed<eed<eed< ddeded ed ed ed ed zdefdZ dd ed edefdZ dded edede fdZ dd ededede fdZ d S)ZLibBackendProtocol MAX_WBITS Z_FULL_FLUSH Z_SYNC_FLUSH Z_BEST_SPEEDZ_FINISH.levelmethodwbitsmemLevelstrategyzdictNr"cdSr$r)r&rKrLrMrNrOrPs r compressobjzZLibBackendProtocol.compressobjTs #&#rcdSr$r)r&rMrPs r decompressobjz!ZLibBackendProtocol.decompressobj]s$'Crr!cdSr$r)r&r!rKrMs rr'zZLibBackendProtocol.compressarbufsizecdSr$r)r&r!rMrWs rr5zZLibBackendProtocol.decompressdrVr)......)..) rrrr/__annotations__rr rRr1rTr.r'r5rrrrErEMsvNNNMMM"&&&& &  &  &}& !&&&&14((('-( "(((( ?B&)8; AD&):= rrEc.eZdZUeed<eed<eed<dS)CompressObjArgsrMrOrKN)rrrr/rYrrrr[r[is+ JJJMMM JJJJJrr[)totalc*eZdZdefdZedefdZedefdZ edefdZ edefdZ edefdZ edefd Z d ed edefd Zd ed edefd Zded ed edefdZded ed edefdZdedefdZdS)ZLibBackendWrapper _zlib_backendc||_dSr$)r_)r&r_s r__init__zZLibBackendWrapper.__init__ps2?rr"c.t|jddS)Nr undefinedgetattrr_r:s rnamezZLibBackendWrapper.namesst):{CCCrc|jjSr$)r_rFr:s rrFzZLibBackendWrapper.MAX_WBITSws!++rc|jjSr$)r_rGr:s rrGzZLibBackendWrapper.Z_FULL_FLUSH{!..rc|jjSr$)r_rHr:s rrHzZLibBackendWrapper.Z_SYNC_FLUSHrirc|jjSr$)r_rIr:s rrIzZLibBackendWrapper.Z_BEST_SPEEDrirc|jjSr$)r_rJr:s rrJzZLibBackendWrapper.Z_FINISHs!**rargskwargsc&|jj|i|Sr$)r_rRr&rmrns rrRzZLibBackendWrapper.compressobjs-t!-t>v>>>rc&|jj|i|Sr$)r_rTrps rrTz ZLibBackendWrapper.decompressobjs/t!/@@@@rr!c.|jj|g|Ri|Sr$)r_r'r&r!rmrns rr'zZLibBackendWrapper.compresss(*t!*4A$AAA&AAArc.|jj|g|Ri|Sr$)r_r5rss rr5zZLibBackendWrapper.decompresss(,t!,TCDCCCFCCCrattrnamec,t|j|Sr$rd)r&rus r __getattr__zZLibBackendWrapper.__getattr__st)8444rN)rrrrErarBstrrfr/rFrGrHrIrJrr rRr1rTrr.r'r5rwrrrr^r^os@&9@@@@DcDDDXD,3,,,X,/c///X//c///X//c///X/+#+++X+???8O????A3A#A:SAAAABVBCB3B5BBBBDvDcDSDUDDDD5C5C555555rr^ ZLibBackendnew_zlib_backendr"c|t_dSr$)ryr_)rzs rset_zlib_backendr|s 0Krencodingsuppress_deflate_headercb|dkrdtjzS|r tj n tjS)Ngzip)ryrF)r}r~s rencoding_to_moders66K)))%< WK ! ! !+BWWrc\eZdZd dededefdZedefdZ edefdZ dS) MemberDecompressObjProtocol.r!r2r"cdSr$rr4s rr5z&MemberDecompressObjProtocol.decompressr(rcdSr$rr:s rr;zMemberDecompressObjProtocol.eofr<rcdSr$rr:s rr@z'MemberDecompressObjProtocol.unused_datarArNr-) rrrrr/r.r5rBrCr;r@rrrrrsdKKvK3KKKKK TX 'U'''X'''rr_DecompressObjT)boundceZdZdefdedzdedzfdZeefde dede fdZ efde dede fd Z e edefd ZdS) DecompressionBaseHandlerNexecutormax_sync_chunk_sizec"||_||_dS)z&Base class for decompression handlers.N) _executor_max_sync_chunk_size)r&rrs rraz!DecompressionBaseHandler.__init__s "$7!!!rr!r2r"cdSDecompress the given data.Nrr4s rdecompress_syncz(DecompressionBaseHandler.decompress_syncr(rcK|jQt||jkr9tj|j|j||d{VS|||Sr)rlenasyncioget_event_looprun_in_executorrrr4s rr5z#DecompressionBaseHandler.decompresss  % 1D D555 /11AA 4dJ ##D*555rcdS)z7Return True if more output is available by passing b"".Nrr:s rdata_availablez'DecompressionBaseHandler.data_availabler(r)rrrMAX_SYNC_CHUNK_SIZErr/rarZLIB_MAX_LENGTH_UNLIMITEDrr.rr5rBrCrrrrrrs%)*=88T/8!4Z8888.G)))(+) )))^) /H 6 6 6(+ 6  6 6 6 6FFFF^XFFFrrcleZdZUdZeed<eed<dZedzed<e defdZ ded edefd Z dS) ConcatDecompressionHandlerayHandler for a codec whose streams may concatenate independent members. Concatenated gzip/deflate members and multi-frame zstd (https://datatracker.ietf.org/doc/html/rfc8878#section-3.1.1) decode the same way: a decompressor handles one member, then flags eof and leaves the rest of the input in unused_data, so every member after it needs a fresh one. _unlimited _decompressorN_pending_unused_datar"cdS)z*Return a decompressor for the next member.Nrr:s r_new_decompressorz,ConcatDecompressionHandler._new_decompressorr(rfirstr2cXt|jj}|g}t|}d}t}|}d} |t|krJ|jjrH| dz } | t krtdt d||_t}||j kr(||z }|dkrt||d|_ nt||zt|} |j ||| |} | r'|| |t| z }|jjr| t|jjz }n| }t|dzt}|t|kJd|S)z9Decode the members following the one ``first`` came from.rz Compressed stream has more than z membersNr) memoryviewrr@rMEMBER_WINDOW_MINr;MAX_DECOMPRESS_MEMBERSrrrr.rminr5appendMEMBER_WINDOW_MAXjoin) r&rr2 remainingpartsproducedposwindowbudgetmembersendchunks r_decompress_membersz.ConcatDecompressionHandler._decompress_membersst1=>> u::"C NN""!% +1 333-<1<<<&*%;%;%=%="*T_,,#h.Q;;05io0F0FD-cFlC NN33C&11)CG2DfMME ' U###CJJ&!% <C 2 >???VaZ):;;AC NN""Dxxr) rrrrr/rYrrr.rrrrrrrrsOOO"""")-%$,---9?999^9,,C,E,,,,,,rrceZdZddddddefdedzdededzdedzdedzdedzd edzfd Zd e d e fd Z d e d e fdZ ddedzd e fdZ dS)ZLibCompressorNFr}r~rKrMrOrrc||_||_|t||n||_t t j|_i}|j|d<|||d<|||d<|jjdi||_dS)NrMrOrKr) rrr_moder^ryr_rR _compressor) r&r}r~rKrMrOrrrns rrazZLibCompressor.__init__!s"$7!} X'> ? ? ? %7{7P$Q$Q"$*w  !)F:   #F7O94-9CCFCCrr!r"c6|j|Sr$)rr'r%s r compress_synczZLibCompressor.compress_sync<s((...rcK|jduot||jk}|r=tj|j|jj|d{VS||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) rrrget_running_looprrrr'r)r&r!should_use_executors rr'zZLibCompressor.compress?s&  %T 1 6D D55    133CC 0 94 !!$'''rr)cR|j||n |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. )rr,r_rJr+s rr,zZLibCompressor.flush[s1%%$DD$*<*E   rr$)rrrrrxrCr/rrarr.rr'r,rrrrr s $(-  #$(*=DD*D"&DTz D Tz D * DT/D!4ZDDDD6/&/U////(6(e((((8   #*         rrc eZdZeZdddefdedzdededzde dzffd Z de fd Z efd e d e defd Zdde defdZedefdZedefdZxZS)ZLibDecompressorNFr}r~rrct||t|||_t t j|_||_d|_ dS)NrrF) superrarrr^ryr_rr _last_empty)r&r}r~rr __class__s rrazZLibDecompressor.__init__nsj (@STTT%h0GHH $6{7P$Q$Q!3355 rr"cB|j|jS)N)rM)r_rTrr:s rrz"ZLibDecompressor._new_decompressor{s!//dj/AAArr!r2c|j|jt|z}d|_|j|jj|z|}|jjr"|jjr|||}|dk|_|jjr.|j |j j kr| |_|SNr) rr.rr5r>r;r@rrrr_rFrr&r!r2results rrz ZLibDecompressor.decompress_sync~s  $ 0,uT{{:D(,D %#..   . 5z     ! Bd&8&D B--fjAAF"S=   ! :dj43E3O&O&O!%!7!7!9!9D  rrr6ct|dkr|j|n|jS)Nr)rr,r8s rr,zZLibDecompressor.flushs=zz   $ $V , , ,#))++ rcVt|jjp|j p|jduSr$)rCrr>rrr:s rrzZLibDecompressor.data_availables8 #3 4 4 5## 5(4 rc|jjSr$)rr;r:s rr;zZLibDecompressor.eofs!%%r)r)rrrrrrrxrCrr/rar1rrr.rr,rBrr; __classcell__rs@rrrksT*J $(-$(*= ! !* !"& !T/ ! !4Z ! ! ! ! ! !B#<BBBB/H(+ 4  C         X &T&&&X&&&&&rrc~eZdZdefdedzdedzddffd Zefdedede fdZ de fd Z e de fd ZxZS) BrotliDecompressorNrrr"ctstdtj|_d|_t ||dS)z)Decompress data using the Brotli library.zIThe brotli decompression is not available. Please install `Brotli` moduleFrN) HAS_BROTLI RuntimeErrorbrotli Decompressor_objrrrar&rrrs rrazBrotliDecompressor.__init__sb  1 '))   (@STTTTTrr!r2ct|jdrh|tkr.tt|j|}ntt|j||}ng|tkr.tt|j|}n.tt|j||}|dk|_|S)rr5r)hasattrrrr r.r5processrrs rrz"BrotliDecompressor.decompress_syncs 49l + + J666eTY%9%9$%?%?@@eTY%9%9$ %K%KLL666eTY%6%6t%<%<==eTY%6%6tZ%H%HII!S= rct|jdr,tt|jSdS)zFlush the decompressor.r,r)rrr r.r,r:s rr,zBrotliDecompressor.flushs7 49g & & 2ty0011 1srcF|j o|j Sr$)r is_finishedrr:s rrz!BrotliDecompressor.data_availables$9((***C43C/CCr)rrrrrr/rarrr.rr,rBrCrrrs@rrrs %)*= U UT/ U!4Z U  U U U U U U /H(+ $u DDDDXDDDDDrrceZdZeZdefdedzdedzddffd Zd dZ e fde d ede fd Z de fd Zedefd ZxZS)ZSTDDecompressorNrrr"ctstdt||||_dS)NzOThe zstd decompression is not available. Please install `backports.zstd` moduler)HAS_ZSTDrrrarrrs rrazZSTDDecompressor.__init__s^  9  (@STTT!3355rrctSr$rr:s rrz"ZSTDDecompressor._new_decompressors!!!rr!r2c:|tkrtn|}|j|j|z}d|_|j||}|jjr"|jjr|||}|jjr||_|Sr$) rZSTD_MAX_LENGTH_UNLIMITEDrrr5r;r@rr)r&r!r2zstd_max_lengthrs rrz ZSTDDecompressor.decompress_syncs666 & %   $ 0,t3D(,D %#..t_EE   ! Gd&8&D G--foFFF   ! :!%!7!7!9!9D  rcdSrrr:s rr,zZSTDDecompressor.flush ssrcH|jj o |jj p|jduSr$)r needs_inputr;rr:s rrzZSTDDecompressor.data_available s3". . Mt7I7M3M3  &d 2 3r)r"r)rrrrrrrr/rarrrr.rr,rBrCrrrs@rrrs*J%)*= 6 6T/ 6!4Z 6  6 6 6 6 6 6""""/H(+ 6u3333X33333rrr)NF):rsyszlibabcrrconcurrent.futuresrtypingrrrr r r r version_infocollections.abcrrr. bytearray brotlicffir ImportErrorrcompression.zstdrbackports.zstdrrrrrrr ValueErrorrr r1rEr[r^ryrYr|rxrCr/rrrrrrrrrrrrrs ########''''''JJJJJJJJJJJJJJJJJJw&&&&&&& 5)%68KK LF#####  JJJJJ 7""5555555333333HHHHH EEEEE*EEE66666h666 ( ( ( ( ( ( ( ((8iu *5*5*5*5*5*5*5*5Z#5"4T":": :::1':1t1111  $)XXDjX!X XXXX((((((((('+3NOOO F F F F Fs F F FF@@@@@!97?;S@@@FH H H H H H H H V>&>&>&>&>&12KL>&>&>&B-D-D-D-D-D1-D-D-D`535353535312DE5353535353sBAA+ A%"A+$A%%A++A54A59BBB