U dZ @sldZddlZddlmZiZiZiZdZd$ddZddZ d d Z d d Z d dZ ddZ eej ej dddgdz ddlZWnek rYnXeej ej dddgdz ddlZWnek rYnXeej ej ddgdz ddlZWnek rYnXeej ej dddgdz ddlZWnek rBYn&XddZdd Zeeed!d"d#gddS)%zCompression utilities.N) ensure_bytes)registerencoders get_encoder get_decodercompress decompresscs0|t<|t<|r,tfdd|DdS)aRegister new compression method. Arguments: encoder (Callable): Function used to compress text. decoder (Callable): Function used to decompress previously compressed text. content_type (str): The mime type this compression method identifies as. aliases (Sequence[str]): A list of names to associate with this compression method. c3s|]}|fVqdSN).0alias content_typer 5/tmp/pip-unpacked-wheel-hqfrjlvz/kombu/compression.py szregister..N) _encoders _decoders_aliasesupdate)encoderdecoderraliasesr r rrs rcCsttS)z/Return a list of available compression methods.)listrr r r rr!srcCst||}t||fS)zGet encoder by alias name.)rgetrtr r rr&s rcCstt||S)zGet decoder by alias name.)rrrrr r rr,srcCst|\}}|t||fS)zCompress text. Arguments: body (AnyStr): The text to compress. content_type (str): mime-type of compression method to use. )rr)bodyrrr r rr1s rcCs t||S)zDecompress compressed text. Arguments: body (AnyStr): Previously compressed text to uncompress. content_type (str): mime-type of compression method used. )r)rrr r rr<srzapplication/x-gzipgzipzlib)rzapplication/x-bz2bzip2Zbzipzapplication/x-brotlibrotlizapplication/x-lzmalzmaxzcCst}||Sr )zstdZZstdCompressorr)rcr r r zstd_compressjsr%cCst}||Sr )r#ZZstdDecompressorr)rdr r rzstd_decompressnsr'zapplication/zstdr# zstandard)N)__doc__rZkombu.utils.encodingrrrr__all__rrrrrrbz2 ImportErrorr r!r(r#r%r'r r r rsn