
                            d Z ddl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  ej                  ej                   j"                        ej$                   G d	 d
e	j&                  ej(                                      Zy)zCCommand to rollback conversion workspaces for a database migration.    N)Optional)resource_args)base)command_mixin)flags)datamigration_v1_messagesc                       e Zd ZdZdddZedej                  ddfd       Zd	ej                  de
ej                     fd
Zy)Rollbackz;Rollback a Database Migration Service conversion workspace.zr
        Rollback a Database Migration Service conversion workspace to the last
        committed snapshot.
      z        To rollback a conversion workspace to the last committed snapshot:

            $ {command} my-conversion-workspace --region=us-central1
      )DESCRIPTIONEXAMPLESparserreturnNc                 Z    t        j                  | d       t        j                  |        y)zArgs 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 rollbackN)r   !AddConversionWorkspaceResourceArgcw_flagsAddNoAsyncFlag)r   s    @lib/surface/database_migration/conversion_workspaces/rollback.pyArgszRollback.Args,   s"     33FMJF#    argsc                     |j                   j                  j                         }| j                  j                  j                  |j                               }| j                  ||d|j                  d            S )a4  Rollback a Database Migration Service conversion workspace.

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

    Returns:
      A dict object representing the operations resource describing the rollback
      operation if the rollback was successful.
    )namezRolled backno_async)conversion_workspace_refresult_operationoperation_namesync)	CONCEPTSconversion_workspaceParseclient
operationsr
   RelativeNameHandleOperationResultIsKnownAndSpecified)selfr   r   r   s       r   RunzRollback.Run7   sz      $}}AAGGI{{--66%224 7  %%!9)$%%j1	 &  r   )__name__
__module____qualname____doc__detailed_helpstaticmethodargparseArgumentParserr   	Namespacer   messages	Operationr'    r   r   r
   r
      se     D


- $8** $t $ $h(( Xh6H6H-I r   r
   )r+   r.   typingr   )googlecloudsdk.api_lib.database_migrationr   googlecloudsdk.callioper   Cgooglecloudsdk.command_lib.database_migration.conversion_workspacesr   r   r   6googlecloudsdk.generated_clients.apis.datamigration.v1r   r1   ReleaseTracksReleaseTrackGADefaultUniverseOnly ConversionWorkspacesCommandMixinCommandr
   r3   r   r   <module>r?      sn    J   C ( ] a h D%%(().}==t|| .  *.r   