
                             d Z ddlmZ ddlmZ ddlmZ ddlZddlmZ ddlm	Z	 ddlm
Z
 ej                   G d	 d
ej                               Zy)DMigrate data from a source database to Cloud Spanner given a schema.    )absolute_import)division)unicode_literalsN)base)flags)migration_backendc                   N    e Zd ZdZd ej
                  d      iZed        Zd Z	y)SchemaAndDatar   EXAMPLESz        To generate schema and copy data to Cloud Spanner from a source database:

          $ {command} --source=postgresql < ~/cart.pg_dump --target-profile='instance=spanner-instance'
      c                    t        j                         j                  |        t        j                         j                  |        t        j                         j                  |        t        j
                         j                  |        t        j                         j                  |        t        j                         j                  |        t        j                         j                  |        t        j                         j                  |        t        j                         j                  |        t        j                         j                  |        t        j                         j                  |        y)z$Register the flags for this command.N)r   GetSpannerMigrationSourceFlagAddToParserGetSpannerMigrationPrefixFlag&GetSpannerMigrationSkipForeignKeysFlag$GetSpannerMigrationSourceProfileFlagGetSpannerMigrationTargetFlag$GetSpannerMigrationTargetProfileFlag!GetSpannerMigrationWriteLimitFlagGetSpannerMigrationDryRunFlagGetSpannerMigrationLogLevelFlagGetSpannerMigrationProjectFlag'GetSpannerMigrationDataflowTemplateFlag)parsers    .lib/surface/spanner/migrate/schema_and_data.pyArgszSchemaAndData.Args(   s    
'')55f=	'')55f=	002>>vF	..0<<VD	'')55f=	..0<<VD	++-99&A	'')55f=	))+77?	((*66v>	113??G    c                    t        j                         }t        j                  ddi      } |d|j                  |j                  |j
                  |j                  |j                  |j                  |j                  |j                  |j                  |j                  |j                  |      }|j                  | _        | j                  |      S )z Run the schema-and-data command.GCLOUD_HB_PLUGINtrue)
extra_varszschema-and-data)commandsourceprefixskip_foreign_keyssource_profiletargettarget_profilewrite_limitdry_run	log_levelprojectdataflow_templateenv)r	   SpannerMigrationWrapperGetEnvArgsForCommandr#   r$   r%   r&   r'   r(   r)   r*   r+   r,   r-   	exit_code _DefaultOperationResponseHandler)selfargscommand_executorenv_varsresponses        r   RunzSchemaAndData.Run7   s    (@@B 55&/1H!{{{{00**{{**$$..00H ''DN00::r   N)
__name__
__module____qualname____doc__textwrapdedentdetailed_helpstaticmethodr   r8    r   r   r   r      s?    L 
(// 
 - H H;r   r   )r<   
__future__r   r   r   r=   googlecloudsdk.callioper   "googlecloudsdk.command_lib.spannerr   r	   DefaultUniverseOnlyBinaryBackedCommandr   rA   r   r   <module>rG      sG    K &  '  ( 4 @ 0;D,, 0; 0;r   