
    )                        d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm	Z	 ddl
mZ d	d
dZd	ddZ G d de      Z e	j"                  e	j$                  j&                         G d dee	j(                               Z e	j"                  e	j$                  j,                         G d dee	j(                               Zy)z;Command to restart migration jobs for a database migration.    )absolute_import)division)unicode_literals)migration_jobs)resource_args)base)flagszE
        Restart a Database Migration Service migration job.
        zf        To restart a migration job:

          $ {command} MIGRATION_JOB --region=us-central1
        )DESCRIPTIONEXAMPLESa	          To restart a migration job:

          $ {command} MIGRATION_JOB --region=us-central1


        To restart a migration job without running prior configuration verification:

          $ {command} MIGRATION_JOB --region=us-central1 --skip-validation
        c                   &    e Zd ZdZed        Zd Zy)_Restart3Restart a Database Migration Service migration job.c                 0    t        j                  | d       y)Args is called by calliope to gather arguments for this command.

    Args:
      parser: An argparse parser that you can use to add arguments that go on
        the command line after this command. Positional arguments are allowed.
    z
to restartN)r   AddOnlyMigrationJobResourceArgsparsers    8lib/surface/database_migration/migration_jobs/restart.pyArgsz_Restart.Args9   s     11&,G    c                     |j                   j                  j                         }t        j                  | j                               }|j                  |j                         |      S )a*  Restart a Database Migration Service migration job.

    Args:
      args: argparse.Namespace, The arguments that this command was invoked
        with.

    Returns:
      A dict object representing the operations resource describing the restart
      operation if the restart was successful.
    )CONCEPTSmigration_jobParser   MigrationJobsClientReleaseTrackRestartRelativeName)selfargsmigration_job_ref	mj_clients       r   Runz_Restart.RunC   sX     3399;2243D3D3FGI&&( r   N)__name__
__module____qualname____doc__staticmethodr   r#    r   r   r   r   6   s    ;H Hr   r   c                   $    e Zd ZdZeZed        Zy)	RestartGAr   c                     t         j                  |        t        j                  |        t        j                  |        t        j
                  |        yr   N)r   r   mj_flagsAddSkipValidationFlag*AddMigrationJobObjectsConfigFlagForRestartAddRestartFailedObjectsFlagr   s    r   r   zRestartGA.Args]   s:     MM&""6*77?((0r   N)r$   r%   r&   r'   DETAILED_HELP_GAdetailed_helpr(   r   r)   r   r   r+   r+   W   s    ;"-
1 
1r   r+   c                   $    e Zd ZdZeZed        Zy)RestartAlphar   c                 .    t         j                  |        yr-   )r   r   r   s    r   r   zRestartAlpha.Argsq   s     MM&r   N)r$   r%   r&   r'   DETAILED_HELP_ALPHAr3   r(   r   r)   r   r   r5   r5   k   s    ;%- r   r5   N)r'   
__future__r   r   r   )googlecloudsdk.api_lib.database_migrationr   r   googlecloudsdk.callioper   <googlecloudsdk.command_lib.database_migration.migration_jobsr	   r.   r7   r2   objectr   ReleaseTracksr   GACommandr+   ALPHAr5   r)   r   r   <module>rA      s    B &  ' D C ( Z		 		 "v B D%%(()1$,, 1 *1& D%%++,8T\\  -r   