$dZddlmZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd l m Z  dd Zdd Zd Zd ZdZdZdZy)z1Common utility functions for sql import commands.) import_util) operations)validate)base)flags)log) properties) console_ioctjj|tj|d}|r|dz}tj ||j ||rtj|dyy)aAdds the base flags for importing data. 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. filetype: String, description of the file type being imported. gz_supported: Boolean, if True then .gz compressed files are supported. user_supported: Boolean, if True then a Postgres user can be specified. Path to the {filetype} file in Google Cloud Storage from which the import is made. The URI is in the form `gs://bucketName/fileName`.0 Compressed gzip files (.gz) are also supported.filetype*PostgreSQL user for this import operation.N)r ASYNC_FLAG AddToParserrAddInstanceArgumentAddUriArgumentformatAddUserparserr gz_supporteduser_supported uri_help_texts 1lib/googlecloudsdk/command_lib/sql/import_util.pyAddBaseImportFlagsrsw//f%F#$ !:Mv}33X3FG MM&FGctjj|tj|d}|r|dz}tj ||j ||rtj|dyy)aAdds the base flags for importing data for bak import. 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. filetype: String, description of the file type being imported. gz_supported: Boolean, if True then .gz compressed files are supported. user_supported: Boolean, if True then a Postgres user can be specified. r r rrN)rrrrrAddBakImportUriArgumentrrrs rAddBakImportFlagsr!9sw//f%F#$ !:M (<(r@|j@jC|jE| j&| jFSd} |r|jrd} t@jHjK|| | |rF|jr:tLjNjQdj/|j0ytLjNjQdj/||y)a|Imports data into 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. import_context: ImportContext; format-specific import metadata. Returns: A dict representing the import 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 collectionz6Bring database [{database}] online with recovery-only.)databaseT)messagedefault cancel_on_noz3Data from [{uri}] will be imported to [{instance}].)uriinstance) importContext)r+r#instancesImportRequestzsql.operations) operationr#)r#r.z&Importing data into Cloud SQL instancezBring database onlinez7Bring database [{database}] online with recovery-only. z1Imported data from [{bucket}] into [{instance}]. )r+bucketN)) sql_client sql_messagesfileType ImportContextFileTypeValueValuesEnumBAKTDE cert_pathr*rValidateInstanceNamer+ ValidateURI recovery_onlyresource_parserParser VALUEScorer# GetOrFailr PromptContinuerr&SqlInstancesImportRequestInstancesImportRequest instancesImportCreatenameasync_rGetSqlOperationsGetRequestr.OperationsV1Beta4WaitForOperationrstatuswrite) argsclientimport_contextr0r1 is_bak_import is_tde_import import_source instance_refimport_requestresult_operation operation_refr's rRunImportCommandrXTs&  *$$,  # # ; ; ? ?@   # # ; ; ? ?@ %2$..txx-  . (:(:;''-- mm**//77AA B ., t)) D K K L ELL M   99$$"")@@&A:. ))00@((// %%""0-  [[  $ $,,!))]5L5L -   5't))%G //-t))JJBII]] J   JJ<CC!- D  rc tj|j|j|j|j |j |j|j|j}t|||S)aImports data from a SQL dump file into 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 representing the import operation resource, if '--async' is used, or else None. )parallelthreadsclean if_exists) rSqlImportContextr1r*r&userrZr[r\r]rXrNrOsql_import_contexts rRunSqlImportCommandrbsa#33  hh mm ii}}ll JJ  $(: ;;rc  tj|j|j|j|j |j |j|j|j|j|j }t|||S)aImports data from a CSV file into 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 representing the import operation resource, if '--async' is used, or else None. ) rCsvImportContextr1r*r&tablecolumnsr_quoteescapefields_terminated_bylines_terminated_byrX)rNrOcsv_import_contexts rRunCsvImportCommandrlss#33  hh mm jj ll ii jj kk    $(: ;;rcbtj|j|j|j|j |j |j|j|j|j|j|j|j|j }t|||S)aImports data from a BAK file into 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 representing the import operation resource, if '--async' is used, or else None. )rBakImportContextr1r*r&r7pvk_path pvk_passwordkeep_encryptedstriped no_recoveryr:bak_typestop_at stop_at_markrXr`s rRunBakImportCommandrws#33  hh mm nn mm   ll   mm ll  $(: ;;rctj|j|j|j|j |j }t|||S)aImports data from a TDE certificate file into 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 representing the import operation resource, if '--async' is used, or else None. )rTdeImportContextr1 certificater7rorprX)rNrOtde_import_contexts rRunTdeImportCommandr|sP#33   nn mm   $(: ;;rN)TT)__doc__googlecloudsdk.api_lib.sqlrrrgooglecloudsdk.calliopergooglecloudsdk.command_lib.sqlrgooglecloudsdk.corerr googlecloudsdk.core.consoler rr!rXrbrlrwr|rrrsP831/(0#*29=H:H6hV<2<6<<