U /e @s8UdZddlmZddlZddlmZddlmZddlm Z ddl m Z ddl m Z dd lmZdd lmZddlZdd lmZmZmZdeed iZd ed<eded<dZeeZee ddlZej ej!d ed<W5QRXeeVddl"Z"ze" e#dWne$k r*edYnXe"j e"j!d ed<dZW5QRXeeTddl%Z%ee%j&edkrzedddl'Z%e%j(j e e%j(j!ddd ed<dZW5QRXeeHddl)Z)ee)j&edkredddZ*ddZ+e*e+d ed<W5QRXd d!Z,e,d"d#Z-d$d$d%efd&d'Z.d(d)Z!dS)*zY Record known compressors Includes utilities for determining whether or not to compress ) annotationsN)Callable)suppress)partial)randint)Literal)parse)identity)ensure_memoryviewnbytes no_default)compress decompresszfdict[str | None | Literal[False], dict[Literal[('compress', 'decompress')], Callable[[bytes], bytes]]] compressionsFzlibzNeed snappy >= 0.5.3snappyz0.23.1zNeed lz4 >= 0.23.1T)Zreturn_bytearraylz4z0.9.0zNeed zstandard >= 0.9.0cCs(tjtjdtjdd}||S)Nzdistributed.comm.zstd.levelzdistributed.comm.zstd.threads)levelthreads) zstandardZZstdCompressordaskconfiggetr )dataZzstd_compressorrD/tmp/pip-unpacked-wheel-g426oqom/distributed/protocol/compression.py zstd_compressYs   rcCst}||S)N)rZZstdDecompressorr)rZzstd_decompressorrrrzstd_decompress`srZzstdc CsHtjd}|dkrtS|tkr$|Std|dttt tfdS)Ndistributed.comm.compressionautozcCsddt|d|DS)z8Decompress frames according to information in the headercSs g|]\}}t|d|qS)r)r).0cframerrr szdecompress..r:)zip)headerframesrrrrs r)/__doc__ __future__rloggingcollections.abcr contextlibr functoolsrrandomrtypingrZpackaging.versionr parse_versionZtlzr rZdistributed.utilsr r r r__annotations__r! getLogger__name__logger ImportErrorrr rrr, TypeErrorr __version__Z lz4.blockblockrrrr(r6r>rrrrsl                  $ -