File: //lib/google-cloud-sdk/lib/surface/artifacts/attachments/__pycache__/download.cpython-312.pyc
�
��� � �. � d Z ddlmZ ddlmZ ddlmZ ddlZddlmZ ddl m
Z
ddl m Z dd l m
Z
dd
l mZ dd l mZ dd lmZ dd
lmZ e
j( e
j* e
j, j. � G d� de
j0 � � � Zy)zAImplements the command to download attachments from a repository.� )�absolute_import)�division)�unicode_literalsN)�
exceptions)�base)�attachment_util)�
download_util)�flags)�requests)�log)�unquotec �@ � e Zd ZdZdZddd�Zed� � Zd� Zd� Z d � Z
y
) �Downloadz;Download an Artifact Registry attachment from a repository.�v1z
{description}a� To download the attachment `my-attachment` to `/path/to/destination/`:
$ {command} my-attachment --destination=/path/to/destination/
To download the attachment `my-attachment` in 8000 byte chunks to `/path/to/destination/`:
$ {command} my-attachment --destination=/path/to/destination/ --chunk-size=8000
To download the attachment `my-attachment` using parallel multipart download with 4 threads:
$ {command} my-attachment --destination=/path/to/destination/ --parallelism=4
For Docker-format repositories only: to download the attachment stored in the OCI version `projects/my-project/locations/us/repositories/my-repo/packages/my-package/versions/sha256:123` to `/path/to/destination/`:
$ {command} --oci-version-name=projects/my-project/locations/us/repositories/my-repo/packages/my-package/versions/sha256:123 --destination=/path/to/destination/
For Docker-format repositories only: to download the attachment stored in the OCI version with URI `us-docker.pkg.dev/my-project/my-repo/my-package@sha256:123` to `/path/to/destination/`:
$ {command} --oci-version-name=us-docker.pkg.dev/my-project/my-repo/my-package@sha256:123 --destination=/path/to/destination/
)� DESCRIPTION�EXAMPLESc � � t j � j | � t j � j | � | j dddd�� | j dddd �� | j d
d d �
� y)z\Set up arguments for this command.
Args:
parser: An argparse.ArgumentParser.
z--oci-version-name�OCI_VERSION_NAMEFzVFor Docker-format repositories only. The version name of the OCI artifact to download.)�metavar�required�helpz
--destination� DESTINATIONTz<Path where you want to save the downloaded attachment files.z
--parallelism� PARALLELISMzXSpecifies the number of threads to use for downloading the attachment files in parallel.)r r N)r
�GetOptionalAttachmentFlag� AddToParser� GetChunkSize� add_argument)�parsers �-lib/surface/artifacts/attachments/download.py�Argsz
Download.ArgsC s� � �
�#�#�%�1�1�&�9� ����$�$�V�,�
��� �"��
%� � � ���� ��
K� � � ���� �
-� � � c �� � t j j |j � |_ t j j |j � s"t j d|j z � �t j j |j � s"t j d|j z � �t j |� }| j ||j � y)z%Runs the attachment download command.z&Destination directory does not exist: z Destination is not a directory: N)
�os�path�
expanduser� destination�exists�
ar_exceptions�DirectoryNotExistError�isdir�PathNotDirectoryErrorr �GetAttachmentToDownload�download_files�files)�self�args�
attachments r �Runz Download.Rund s� � � �w�w�)�)�$�*:�*:�;�D��
�7�7�>�>�$�*�*� +� � 0� 0�
2�T�5E�5E�
E�
� � �7�7�=�=��)�)� *� � /� /�
,�t�/?�/?�
?�
� � !�8�8��>�J�����j� .� .�/r! c �6 � d}|j xs |}|j xs d}|D ]� }t j j |� } t |j
dd� d � }| j ||� } t j j |j | � }
t j |
|| dt |� t |� �� t j j! dj# |j � � �� y # t $ r |}Y ��w xY w)Ni 0 � �:F)� parallelismz&Successfully downloaded the file to {})�
chunk_sizer6 r# r$ �basenamer
�rsplit�
IndexError�
get_file_name�joinr& r r �intr �status�Print�format) r/ r0 r. �default_chunk_sizer7 r6 �file�file_id�default_file_name� file_name�
final_paths r r- zDownload.download_filest s� � �(��� � �6�$6�J��"�"�'�a�K��� ��� � ��&�g�$� #�G�N�N�3��$:�1�$=� >�� �$�$�T�+<�=�i��7�7�<�<�� 0� 0�)�<�j�� � �
�
�
�
�
�j�/��+�&�
�
�j�j���
2�
9�
9�$�:J�:J�
K��' �� � $� #��$�s � D
�
D�Dc �8 � t j � }t j � }|j |�� }|j j |� }|j �8|j j D ] }|j dk( s�|j c S |S )N)�namez)artifactregistry.googleapis.com/file_name)
r � GetClient� GetMessages�<ArtifactregistryProjectsLocationsRepositoriesFilesGetRequest�%projects_locations_repositories_files�Get� annotations�additionalProperties�key�value)r/ rB rD �client�messages�request�resp�es r r; zDownload.get_file_name� s� � �
�
�
�
!�F��#�#�%�H��M�M�� N�
� �
� 7� 7� ;� ;�G� D�D� ���#�� � �4�4�!� �5�5�?� ?����.� 5� � r! N) �__name__�
__module__� __qualname__�__doc__� api_version�
detailed_help� staticmethodr r2 r- r; � r! r r r ! sB � � D��+� %���-�6 � � � �@0� �6 r! r )rZ �
__future__r r r r# � googlecloudsdk.api_lib.artifactsr r( �googlecloudsdk.callioper �$googlecloudsdk.command_lib.artifactsr r r
r �googlecloudsdk.corer �six.moves.urllib.parser
�DefaultUniverseOnly�
ReleaseTracks� ReleaseTrack�GA�Commandr r^ r! r �<module>rj sx �� H� &� � '� � H� (� @� >� 6� 9� #� *� �������D�%�%�(�(�)�z �t�|�|� z � *� �z r!