dZddlmZddlmZddlmZddlmZddlmZddlmZddl m Z dd l m Z dd l mZdd l mZd d e j fdZdZdZdZdZdZy)z1Common utility functions for sql export commands.)absolute_import)division)unicode_literals) export_util) operations)validate)base)flags)log) propertiesTFctjj|tj|d}|r|dz}tj ||tj |||y)a^Adds the base export flags to the parser. Args: parser: The current argparse parser to add these flags to. gz_supported: Boolean, specifies whether gz compression is supported. database_required: Boolean, specifies whether the database flag is required. database_help_text: String, specifies the help text for the database flag. zThe path to the file in Google Cloud Storage where the export will be stored. The URI is in the form gs://bucketName/fileName. If the file already exists, the operation fails.z< If the filename ends with .gz, the contents are compressed.N)r ASYNC_FLAG AddToParserr AddInstanceArgumentAddUriArgumentAddDatabaseList)parser gz_supporteddatabase_requireddatabase_help_text uri_help_texts 1lib/googlecloudsdk/command_lib/sql/export_util.pyAddBaseExportFlagsrse//f%F#&-!&@AMv}- 24EFc|j}|j}tj|j|j j |jdtjjjjid}|j|j|j|j|}|jj|}|j j!d|j"|j}|j$r@|j&j)|j+|j|j,S|j.|j0j2j4k(} | r |j6n |j8} t&j:j=||d t>j@jCd jE|| y ) a|Exports data from a Cloud SQL instance. Args: args: argparse.Namespace, The arguments that this command was invoked with. client: SqlClient instance, with sql_client and sql_messages props, for use in generating messages and making API calls. export_context: ExportContext; format-specific export metadata. Returns: A dict representing the export operation resource, if '--async' is used, or else None. Raises: HttpException: An HTTP error response was received while executing API request. ToolException: An error other than HTTP error occurred while executing the command. projectz sql.instances)params collection) exportContext)instancerinstancesExportRequestzsql.operations) operationr)rr"zExporting Cloud SQL instancez%Exported [{instance}] to [{bucket}]. )r bucketN)# sql_client sql_messagesrValidateInstanceNamer resource_parserParser VALUEScorer GetOrFailSqlInstancesExportRequestInstancesExportRequest instancesExportCreatenameasync_rGetSqlOperationsGetRequestr"fileType ExportContextFileTypeValueValuesEnumTDE cert_pathuriOperationsV1Beta4WaitForOperationr statuswriteformat) argsclientexport_contextr$r% instance_refexport_requestresult_operation operation_ref is_tde_export export_dests rRunExportCommandrI9s&  *$$,  .''-- mm**//77AA B .",  99$$"")@@&A(:).  ))00@((// %%""0$-  [[  $ $,,!))]5L5L - N OO   # # ; ; ? ?@ #0TXX+ // M0NP**;BBKC12 rc btj|j|j|j|j |j |j|j|j|j }|j rtjjdt|||S)aExports data from a Cloud SQL instance to a MySQL dump file. Args: args: argparse.Namespace, The arguments that this command was invoked with. client: SqlClient instance, with sql_client and sql_messages props, for use in generating messages and making API calls. Returns: A dict object representing the operations resource describing the export operation if the export was successful. )offloadparallelthreadsclean if_existspServerless exports cost extra. See the pricing page for more information: https://cloud.google.com/sql/pricing. )rSqlExportContextr%r:databasetablerKrLrMrNrOr r=r>rIr@rAsql_export_contexts rRunSqlExportCommandrVws#33  hh mm jjll}}ll JJ  \\JJ{ $(: ;;rc btj|j|j|j|j |j |j|j|j|j }|j rtjjdt|||S)aExports data from a Cloud SQL instance to a CSV file. Args: args: argparse.Namespace, The arguments that this command was invoked with. client: SqlClient instance, with sql_client and sql_messages props, for use in generating messages and making API calls. Returns: A dict object representing the operations resource describing the export operation if the export was successful. )rKquoteescapefields_terminated_bylines_terminated_byrP)rCsvExportContextr%r:rRqueryrKrXrYrZr[r r=r>rI)r@rAcsv_export_contexts rRunCsvExportCommandr_s#33  hh mm jjll JJ [[4422 4 \\JJ{ $(: ;;rc  tj|j|j|j|j |j |j|j|j|j }t|||S)aExport data from a Cloud SQL instance to a SQL Server BAK file. Args: args: argparse.Namespace, The arguments that this command was invoked with. client: SqlClient instance, with sql_client and sql_messages props, for use in generating messages and making API calls. Returns: A dict object representing the operations resource describing the export operation if the export was successful. ) rBakExportContextr%r:rR stripe_countstripedbak_typedifferential_baseexport_log_start_timeexport_log_end_timerIrTs rRunBakExportCommandrhsp#33  hh mm  ll mm       $(: ;;rctj|j|j|j|j |j }t|||S)aExport TDE certificate from a Cloud SQL instance. Args: args: argparse.Namespace, The arguments that this command was invoked with. client: SqlClient instance, with sql_client and sql_messages props, for use in generating messages and making API calls. Returns: A dict object representing the operations resource describing the export operation if the export was successful. )rTdeExportContextr% certificater9pvk_path pvk_passwordrI)r@rAtde_export_contexts rRunTdeExportCommandrosP#33   nn mm   $(: ;;rN)__doc__ __future__rrrgooglecloudsdk.api_lib.sqlrrrgooglecloudsdk.callioper googlecloudsdk.command_lib.sqlr googlecloudsdk.corer r &DEFAULT_DATABASE_LIST_EXPORT_HELP_TEXTrrIrVr_rhrorrrxsY8&'21/(0#* CC G6;|<<<:<4