�

���
���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
lmZ
Gd�de	j�Zy
)z$The gcloud datastore import command.�)
�absolute_import)
�division)
�unicode_literals)
�	admin_api)
�
operations)
�base)
�flags)
�
properties)
�	resourcesc�.�eZ
dZd
Zddi
Zed��Zd�Zy)�Importa�
Import Cloud Datastore entities from Google Cloud Storage.

  Imports entities into Google Cloud Datastore. Existing entities with
  the same key are overwritten. The import occurs in the background and its
  progress can be monitored and managed via the Operation resource that is
  created. If an Import operation is cancelled, it is possible that a subset of
  the data has already been imported to Cloud Datastore. This data will not be
  removed.
  �EXAMPLESaW          To import all data exported to the output URL
          `gs://exampleBucket/exampleExport/exampleExport.overall_export_metadata`, run:

            $ {command} gs://exampleBucket/exampleExport/exampleExport.overall_export_metadata

          To import all data exported to the output URL
          `gs://exampleBucket/exampleExport/exampleExport.overall_export_metadata`
          without waiting for the operation to complete, run:

            $ {command} gs://exampleBucket/exampleExport/exampleExport.overall_export_metadata --async

          To import only the `exampleKind` from the data exported to the output
          URL `gs://exampleBucket/exampleExport/exampleExport.overall_export_metadata`,
          run:

            $ {command} gs://exampleBucket/exampleExport/exampleExport.overall_export_metadata --kinds='exampleKind'
      c
��t
j|�

t
j|�

|jd
d��
tj
j
|�

y)z Register flags for this command.�	input_urla

        Location of the import metadata. Must be a valid Google Cloud Storage
        object. The file extension is 'overall_export_metadata'.

        This location is the 'output_url' field of a previous export, and can
        be found via the 'operations describe' command.
        )
�helpN)r	�AddEntityFilterFlags�
AddLabelsFlag�add_argumentr�
ASYNC_FLAG�AddToParser)
�parsers
 �lib/surface/datastore/import.py�ArgszImport.Args>sO��
���v�&�	�����
������
�	�O�O����'�c�
�tjjjj	d
��
}t
jj|
jd��}tj|dj|j|j�|
j|
j|
j ��}|
j"st%j&|�

|S)NT)
�requiredzstorage.objects)
�
collectionz
gs://{}/{})�kinds�
namespaces�labels)r
�VALUES�core�project�Getr�REGISTRY�Parserrr
�format�bucket�objectrr�operation_labels�async_r�WaitForOperation)�self�argsr#�
input_url_ref�responses     r�Runz
Import.RunNs������$�$�,�,�0�0�$�0�?�G��&�&�,�,����#4�-�
6�M��������M�0�0�-�2F�2F�G��j�j��?�?��$�$�&�H��;�;��!�!�(�+��OrN)�__name__�
__module__�__qualname__�__doc__�
detailed_help�staticmethodrr1�rrr
r
s1����
��-�,�
(��
(�rr
N)r5�
__future__rrr� googlecloudsdk.api_lib.datastorerr�googlecloudsdk.callioper�$googlecloudsdk.command_lib.datastorer	�googlecloudsdk.corer
r�Commandr
r8rr�<module>r?
s7�


�
+�&��'�6�7�(�6�*�)�@

�T�\�\�@

r