dZddlmZddlmZddlmZddlZddlmZddlmZddlm Z dd lm Z dd lm Z dd lm Z d Z dd Z ddZdZdZy)z4Utility functions for performing download operation.)absolute_import)division)unicode_literalsN)errors) gzip_util) hash_util) storage_url) symlink_util)tracker_file_util_symctjj|sytj|||||rtj |ytj ||y)a Converts temporary file to final form by decompressing or renaming. Args: source_resource (ObjectResource): May contain encoding metadata. temporary_file_path (str): File path to unzip or rename. final_file_path (str): File path to write final file to. do_not_decompress_flag (bool): User flag that blocks decompression. server_encoding (str|None): Server-reported `content-encoding` of file. Returns: (bool) True if file was decompressed or renamed, and False if file did not exist. FT)ospathexistsrdecompress_gzip_if_necessaryremoverenamesource_resourcetemporary_file_pathfinal_file_pathdo_not_decompress_flagserver_encodings @lib/googlecloudsdk/command_lib/storage/tasks/cp/download_util.py_decompress_or_rename_filer!s\( + , ++O,?,;,B,; = II!" II!?3 c|xr |j}|r |tz}n|}t|||||}|sy|r,tj||t j ||S)aConverts temporary file to final form. This may involve decompressing, renaming, and/or converting symlink placeholders to actual symlinks. Args: source_resource (ObjectResource): May contain encoding metadata. temporary_file_path (str): File path to unzip or rename. final_file_path (str): File path to write final file to. do_not_decompress_flag (bool): User flag that blocks decompression. server_encoding (str|None): Server-reported `content-encoding` of file. convert_symlinks (bool): Whether symlink placeholders should be converted to actual symlinks. Returns: (bool) True if file was decompressed, renamed, and/or converted to a symlink; False if file did not exist. rF)placeholder_path symlink_path) is_symlink$SYMLINK_TEMPORARY_PLACEHOLDER_SUFFIXrr )create_symlink_from_temporary_placeholderrr) rrrrrconvert_symlinks make_symlinkdecompress_or_rename_pathdecompress_or_rename_results rfinalize_downloadr'Cs}4"@o&@&@,"5!E"F!0 :%-/3% ! % ::2II'( $$rc tj|||y#tj$r?t j |t jtj|wxYw)a Confirms hashes match for copied objects. Args: download_path (str): URL of object being validated. source_hash (str): Hash of source object. destination_hash (str): Hash of downloaded object. Raises: HashMismatchError: Hashes are not equal. N) rvalidate_object_hashes_matchrHashMismatchErrorrrr delete_download_tracker_filesr storage_url_from_string) download_path source_hashdestination_hashs r/validate_download_hash_and_delete_corrupt_filesr0ws\  **=++;=  ! ! IIm33++M:<  s AA,c8|jdk(r |r|dyy)z>Returns valid download range bool and reports progress if not.rTF)size)cloud_resourceprogress_callbacks r(return_and_report_if_nothing_to_downloadr5s!A  r)FN)FNF)__doc__ __future__rrrr"googlecloudsdk.command_lib.storagerrrr r r r!rr'r0r5rrr:sW;&' 588:;@'-$! L! 1%h ,r