U be@sdZddlZddlZeeZiZdZdZ ddZ ddZ d d Z d d Z d dZddZe dfddZe dee dedS)z?Implements the compression layer of the ``smart_open`` library.NdisableZinfer_from_extensioncCsttgtS)z}Return the list of supported compression types available to open. See compression paratemeter to smart_open.open(). )NO_COMPRESSIONINFER_FROM_EXTENSIONget_supported_extensionsrrX/var/www/html/services/stratfitenv/lib/python3.8/site-packages/smart_open/compression.pyget_supported_compression_typessrcCs ttS)zLReturn the list of file extensions for which we have registered compressors.)sorted_COMPRESSOR_REGISTRYkeysrrrrr!srcCsD|r|ddkstd||}|tkr8td||t|<dS)a)Register a callback for transparently decompressing files with a specific extension. Parameters ---------- ext: str The extension. Must include the leading period, e.g. ``.gz``. callback: callable The callback. It must accept two position arguments, file_obj and mode. This function will be called when ``smart_open`` is opening a file with the specified extension. Examples -------- Instruct smart_open to use the `lzma` module whenever opening a file with a .xz extension (see README.rst for the complete example showing I/O): >>> def _handle_xz(file_obj, mode): ... import lzma ... return lzma.LZMAFile(filename=file_obj, mode=mode, format=lzma.FORMAT_XZ) >>> >>> register_compressor('.xz', _handle_xz) r.z,ext must be a string starting with ., not %rz.overriding existing compression handler for %rN) ValueErrorlowerr loggerwarning)extcallbackrrrregister_compressor&s   rcs|jfdd}||_dS)aHEnsure that closing the `outer` stream closes the `inner` stream as well. Use this when your compression library's `close` method does not automatically close the underlying filestream. See https://github.com/RaRe-Technologies/smart_open/issues/630 for an explanation why that is a problem for smart_open. cs(z W5r"d}|XdS)Nclose)argsfpinnerZ outer_closerr close_bothQs   ztweak_close..close_bothNr)outerrrrrr tweak_closeGs rcCs$ddlm}|||}t|||S)Nr)BZ2File)bz2rr)file_objmoderresultrrr _handle_bz2]s   r"cCs$ddl}|j||d}t|||S)Nr)fileobjr )gzipGzipFiler)rr r$r!rrr _handle_gzipds r&c Cs|tkr |S|tkr`z|p|j}Wn(ttfk rNtd||YSXtj |\}}|t kr~| dr~t d|z t |}Wntk r|YSX|||SdS)a Wrap `file_obj` with an appropriate [de]compression mechanism based on its file extension. If the filename extension isn't recognized, simply return the original `file_obj` unchanged. `file_obj` must either be a filehandle object, or a class which behaves like one. If `filename` is specified, it will be used to extract the extension. If not, the `file_obj.name` attribute is used as the filename. zRunable to transparently decompress %r because it seems to lack a string-like .name+z3transparent (de)compression unsupported for mode %rN)rrnamerAttributeError TypeErrorrrospathsplitextr endswithr KeyError)rr compressionfilename_rrrrcompression_wrapperks&     r3z.bz2z.gz)__doc__loggingos.pathr+ getLogger__name__rr rrrrrrr"r&r3rrrrs !'