File: //lib/google-cloud-sdk/lib/surface/artifacts/generic/__pycache__/download.cpython-312.pyc
�
��� � �j � d Z ddlmZ ddlmZ ddlmZ ddlZddlmZ ddl m
Z
ddl m Z dd l m
Z
dd
l mZ dd lmZ e
j" e
j$ e
j& j( e
j& j* e
j& j, � G d � d
e
j. � � � Zy)zGImplements the command to download generic artifacts from a repository.� )�absolute_import)�division)�unicode_literalsN)�
exceptions)�base)�
download_util)� file_util)�flags)�logc �< � e Zd ZdZddd�Zed� � Zd� Zd� Zd� Z y )
�Downloadz?Download a generic artifact from a generic artifact repository.z
{description}al To download version v0.1.0 of myfile.txt located in a repository in "us-central1" to /path/to/destination/:
$ {command} --location=us-central1 --project=myproject --repository=myrepo --package=mypackage --version=v0.1.0 --destination=/path/to/destination/ --name=myfile.txt
To download version v0.1.0 of myfile.txt located in a repository in "us-central1" to /path/to/destination/ using parallel multipart download with 4 threads:
$ {command} --location=us-central1 --project=myproject --repository=myrepo --package=mypackage --version=v0.1.0 --destination=/path/to/destination/ --name=myfile.txt --parallelism=4
To download version v0.1.0 of myfile.txt in 8000 byte chunks located in a repository in "us-central1" to /path/to/destination/:
$ {command} --location=us-central1 --project=myproject --repository=myrepo --package=mypackage --version=v0.1.0 --destination=/path/to/destination/ --name=myfile.txt --chunk-size=8000
To download all files of version v0.1.0 and package mypackage located in a repository in "us-central1" to /path/to/destination/
while maintaining the folder hierarchy:
$ {command} --location=us-central1 --project=myproject --repository=myrepo --package=mypackage --version=v0.1.0 --destination=/path/to/destination/
)� DESCRIPTION�EXAMPLESc �^ � t j � j | � t j � j | � | j dddd�� | j dddd�� | j d d
dd �� | j d d
d�� | j ddd�� y)z\Set up arguments for this command.
Args:
parser: An argparse.ArgumentParser.
z
--destination� DESTINATIONTz4The path where you want to save the downloaded file.)�metavar�required�helpz --package�ARTIFACTzThe artifact to download.z --version�VERSIONz(The version of the artifact to download.z--name�NAMEz<If specified, the file name within the artifact to download.)r r z
--parallelism� PARALLELISMzLSpecifies the number of threads to use for downloading the file in parallel.N)r
�GetRequiredRepoFlag� AddToParser� GetChunkSize� add_argument)�parsers �)lib/surface/artifacts/generic/download.py�Argsz
Download.ArgsC s� � �
� � � �+�+�F�3� ����$�$�V�,�
���� ��
C� � � ������
(� � � ������
7� � � �����
K� � �
���� �
� � � c � � |j j j � }t j j |j � |_ t j j |j � s"t j d|j z � �t j j |j � s"t j d|j z � �|j rot j j |j � }dj |j |j |j � }| j# ||||� yt% j& |� }|s9t j( dj |j |j � � �| j+ |||� y)z*Run the generic artifact download command.z&Destination directory does not exist: z Destination is not a directory: z{}:{}:{}z*No files found for package: {} version: {}N)�CONCEPTS�
repository�Parse�os�path�
expanduser� destination�exists�
ar_exceptions�DirectoryNotExistError�isdir�PathNotDirectoryError�name�basename�format�package�version�downloadGenericArtifactr �ListGenericFiles�ArtifactRegistryError�batchDownloadFiles)�self�args�repo_ref� file_name�file_id�
list_filess r �Runz Download.Runm sU � � �}�}�'�'�-�-�/�H��w�w�)�)�$�*:�*:�;�D��
�7�7�>�>�$�*�*� +� � 0� 0�
2�T�5E�5E�
E�
� � �7�7�=�=��)�)� *� � /� /�
,�t�/?�/?�
?�
� � �y�y��'�'�"�"�4�9�9�-�i��!�!�$�,�,�� � �d�i�i�H�g�
�"�"�4��7�I�F� �-�-�d�3�j�
��1�1� 8� ?� ?�� � �d�l�l�
�
�
�
� � �d�H�j�9r c
�� � t j j |j |� }t j
|j |j |j |� }d}|j xs |}|j xs d} t j ||j � |dt |� t | � � t j j# dj% |j � � y )Ni 0 � Fz&Successfully downloaded the file to {})r% r& �joinr( r �EscapeFileNameFromIDs�
projectsId� locationsId�repositoriesId�
chunk_size� parallelismr r
� RelativeName�intr �status�Printr0 )
r7 r8 r9 r; r: �
final_path� file_escaped�default_chunk_sizerE rF s
r r3 z Download.downloadGenericArtifact� s� � ���� � �d� .� .� �:�J� �2�2����� � �� � �� �L� )��� � �6�$6�J��"�"�'�a�K������!�!�#��
� �J�� �K��
� �J�J���0�7�7��8H�8H�I�r c � � |D ]� }t j j |j � }|j dd� d j dd� }d|v r^t j j
|� }t j t j j |j |� d�� | j ||||� �� y )N�:r? z%2F�/T)�exist_ok) r% r& r/ r. �rsplit�replace�dirname�makedirsr@ r( r3 )r7 r8 r9 r< �filesr; r: �ds r r6 zDownload.batchDownloadFiles� s� � ������ � ����,�g��.�.��a�(��+�3�3�E�3�?�i� � � � �G�G�O�O�I� &��
� � �B�G�G�L�L��!1�!1�1�5��E�
�"�"�4��7�I�F� r N)
�__name__�
__module__� __qualname__�__doc__�
detailed_help� staticmethodr r= r3 r6 � r r r
r
s> � �
H� %���-�: �'� �'�R:�:�0 Gr r
)r[ �
__future__r r r r% � googlecloudsdk.api_lib.artifactsr r* �googlecloudsdk.callioper �$googlecloudsdk.command_lib.artifactsr r r
�googlecloudsdk.corer �DefaultUniverseOnly�
ReleaseTracks� ReleaseTrack�ALPHA�BETA�GA�Commandr
r^ r r �<module>rk s� �� N� &� � '� � H� (� >� :� 6� #� ������������T� .� .� 3� 3�T�5F�5F�5I�5I��HG�t�|�|� HG�� �HGr