�

����z
�dZd
dl
mZ
d
dl
mZ
d
dl
mZ
d
dlmZ
d
dlmZ
d
dlm	Z	
d
dl
mZ
d
d	lm
Z

d
d
lmZ
ej ej"j$ej"j&ej"j(�Gd�dej*��Zd
ddj/dd�d�e_y)zAImplements the command to upload GooGet packages to a repository.�)
�absolute_import)
�division)
�unicode_literals)
�transfer)
�apis)
�waiter)
�base)
�flags)
�	resourcesc�&�eZ
dZd
Zed��Zd�Zy)�Upload�2Upload a GooGet package to an artifact repository.c
��t
j�j|�

tjj|�

|jd
ddd��
y)z\Set up arguements for this command.

    Args:
      parser: An argparse.ArgumentPaser.
    z--source�SOURCETz,            The path of a package to upload.)�metavar�required�helpN)r
�GetRepoArgFromBeta�AddToParserr	�
ASYNC_FLAG�add_argument)
�parsers
 �&lib/surface/artifacts/googet/upload.py�ArgszUpload.Args#sN��
����*�*�6�2��O�O����'�
������
0�	�1�c�r�t
jd
d�}|j}d|jd<|
jj
j
�}|j|j���
}tjj|
jd��}|jj||��}|j}tj j#|j$d	�
�}	|
j&r|	St)j*t)j,|j.�
|	d�}
|
S)zRun package import command.�artifactregistry�v1�	multipartzX-Goog-Upload-Protocol)
�parentzapplication/gzip)
�	mime_type)
�uploadz.artifactregistry.projects.locations.operations)
�
collectionzUploading package)r�GetClientInstance�MESSAGES_MODULE�additional_http_headers�CONCEPTS�
repository�Parse�IArtifactregistryProjectsLocationsRepositoriesGoogetArtifactsUploadRequest�RelativeNamerr
�FromFile�source�/projects_locations_repositories_googetArtifacts�	operationr�REGISTRY�ParseRelativeName�name�async_r�WaitFor�CloudOperationPollerNoResources�projects_locations_operations)�self�args�client�messages�repo_ref�requestr"�response�op�op_ref�results           r�Runz
Upload.Run4s
��
�
#�
#�$6��
=�F��%�%�H�?J�F�"�"�#;�<��}�}�'�'�-�-�/�H��`�`��$�$�&�a
�
(�G��_�_�
%�
%�d�k�k�=O�
%�
P�F��E�E�L�L���M
�
 �H�
�	�	�B�
�
�
�
1�
1�
���L�2�
N
�F��{�{�
�m��~�~�
�
0�
0��2�2�
4�
�%�'�f�
�mrN)�__name__�
__module__�__qualname__�__doc__�staticmethodrrA�rrr
r
s��;��1��1� rr
rzX
      *{command}* uploads a GooGet package to the specified artifact repository.
      z�
      To upload the package `my-package.goo` to `my-repo`, run:

        $ {0} my-repo --location=us-central1 --source={1}
    z	{command}zmy-package.goo)�brief�DESCRIPTION�EXAMPLESN)rE�
__future__rrr�apitools.base.pyr�googlecloudsdk.api_lib.utilrr�googlecloudsdk.callioper	�$googlecloudsdk.command_lib.artifactsr
�googlecloudsdk.corer�
ReleaseTracks�ReleaseTrack�ALPHA�BETA�GA�Commandr
�format�
detailed_helprGrr�<module>rY
s��


� 
H
�&��'�%�,�.�(�6�)���
�
�D�%�%�+�+�T�->�->�-C�-C��%�%�(�(�
*�1
�T�\�\�1
�
*�1
�j
B
�
��	��{�,�-�
��r