o RD©i™ ã@svddlmZddlZddlZddlZddlmZddlmZm Z dZ dZ ee ƒZ dddd„Zdddd„Zddd„ZdS)é)Ú annotationsN)Ú getLogger)ÚIOÚ Generatori@éTÚ raw_data_fdrÚ add_bracketÚboolÚreturnÚbytescCs°t ttj¡}t ¡}|r| d¡| t¡}|rM| |  |¡¡|j dkr?|j }t ttj¡}| |  |¡¡|j dks'| t¡}| |  ¡¡|s|rT| d¡|  ¡S)aDecompresses raw data from file like object with zlib. Args: raw_data_fd: File descriptor object. add_bracket: Whether, or not to add brackets around the output. (Default value = True) Returns: A byte array of the decompressed file. ó[óó]) ÚzlibÚ decompressobjÚ MAGIC_NUMBERÚ MAX_WBITSÚioÚBytesIOÚwriteÚreadÚ CHUNK_SIZEÚ decompressÚ unused_dataÚflushÚgetvalue)rrÚobjÚwriterÚdr©rúY/var/www/Datamplify/venv/lib/python3.10/site-packages/snowflake/connector/gzip_decoder.pyÚdecompress_raw_datas$     ý ù r!cCs\t ¡}|r | d¡tjdgtjtjd}| |j| ¡dd¡|r*| d¡| ¡S)aDExperimental: Decompresses raw data from file like object with zcat. Otherwise same as decompress_raw_data. Args: raw_data_fd: File descriptor object. add_bracket: Whether, or not to add brackets around the output. (Default value = True) Returns: A byte array of the decompressed file. r Úzcat)ÚstdinÚstdout)Úinputrr) rrrÚ subprocessÚPopenÚPIPEÚ communicaterr)rrrÚprrr Údecompress_raw_data_by_zcat,s   r+úGenerator[str]ccsšt ttj¡}dV| t¡}|rA| |¡ d¡V|jdkr:|j}t ttj¡}| |¡ d¡V|jdks!| t¡}|s|  ¡ d¡dVdS)zÇDecompresses a raw data in file like object and yields a Unicode string. Args: raw_data_fd: File descriptor object. Yields: A string of the decompressed file in chunks. ú[zutf-8r ú]N) rrrrrrrÚdecoderr)rrrrrrr Ú%decompress_raw_data_to_unicode_stream@s€    ý úr0)T)rrrr r r )rrr r,)Ú __future__rrr&rÚloggingrÚtypingrrrrÚ__name__Úloggerr!r+r0rrrr Ús