8 FdZdZ ddZ ddZdZdZy) zBCommon command-agnostic utility functions for sql export commands.c|*|jjjjS|jjjj |j S)N) ExportContextBakExportOptionsValueBakTypeValueValuesEnumFULLlookup_by_nameupper) sql_messagesbak_types -lib/googlecloudsdk/api_lib/sql/export_util.py ParseBakTyper sY ""88OOTT  # # 9 9 P P _ _nn Nc |rY|jd||xsg||jjj|jj|xsg||S|jjj ||} |jd||xsg||jjj|jj|xsg|| S)a7Generates the ExportContext for the given args, for exporting to SQL. Args: sql_messages: module, The messages module that should be used. uri: The URI of the bucket to export to; the output of the 'uri' arg. database: The list of databases to export from; the output of the '--database' flag. table: The list of tables to export from; the output of the '--table' flag. offload: bool, The export offload flag. parallel: Whether to use parallel export or not. threads: The number of threads to use. Only applicable for parallel export. clean: Whether to include SQL statement (DROP ) required to drop database objects prior to import; corresponds to the clean flag on pg_dump. Only applies to PostgreSQL non-parallel exports. if_exists: Whether to include SQL statement (IF EXISTS) to each drop statement produced by the clean flag; corresponds to the if-exists flag on pg_dump. Only applies to PostgreSQL non-parallel exports. Returns: ExportContext, for use in InstancesExportRequest.exportContext. sql#exportContext)tablesparallelthreads)kinduri databasesoffloadfileTypesqlExportOptions)cleanifExists)rrpostgresExportOptions)rFileTypeValueValuesEnumSQLSqlExportOptionsValuePostgresExportOptionsValue) r rdatabasetablerrrr if_existspostgres_export_optionss r SqlExportContextr$sB  % % .b++CCGG%33II;B7J &    ++,F,F)-G-   % % .b++CCGG%33II;B"9J &  r c |jd||xsg||jjj|jj|||||S)aGenerates the ExportContext for the given args, for exporting to CSV. Args: sql_messages: module, The messages module that should be used. uri: The URI of the bucket to export to; the output of the 'uri' arg. database: The list of databases to export from; the output of the '--database' flag. query: The query string to use to generate the table; the output of the '--query' flag. offload: bool, The export offload flag. quote: character in Hex. The quote character for CSV format; the output of the '--quote' flag. escape: character in Hex. The escape character for CSV format; the output of the '--escape' flag. fields_terminated_by: character in Hex. The fields delimiter character for CSV format; the output of the '--fields-terminated-by' flag. lines_terminated_by: character in Hex. The lines delimiter character for CSV format; the output of the '--lines-terminated-by' flag. Returns: ExportContext, for use in InstancesExportRequest.exportContext. r) selectQueryquoteCharacterescapeCharacterfieldsTerminatedBylinesTerminatedBy)rrrrrcsvExportOptions)rrCSVCsvExportOptionsValue) r rr queryrquoteescapefields_terminated_bylines_terminated_bys r CsvExportContextr3]sj>  # #  B))AAEE#11GG 1/ H1 $ 2 2r c R|jj} ||| _||| _|| _t ||| _||jd| _||jd| _ |jd|||jjj| S)aaGenerates the ExportContext for the given args, for exporting to BAK. Args: sql_messages: module, The messages module that should be used. uri: The URI of the bucket to export to; the output of the 'uri' arg. database: The list of databases to export from; the output of the '--database' flag. stripe_count: How many stripes to perform the export with. striped: Whether the export should be striped. bak_type: Type of bak file that will be exported. SQL Server only. differential_base: Whether the bak file export can be used as differential base for future differential backup. SQL Server only. export_log_start_time: start time of the log export. SQL Server only. export_log_end_time: end time of the log export. SQL Server only. Returns: ExportContext, for use in InstancesExportRequest.exportContext. z%Y-%m-%dT%H:%M:%S.%fZr)rrrrbakExportOptions) rrstriped stripeCountdifferentialBaser bakTypestrftimeexportLogStartTimeexportLogEndTimerBAK) r rr stripe_countr6r differential_baseexport_log_start_timeexport_log_end_timebak_export_optionss r BakExportContextrCs:$11GGI !(%1"(9%+L(C&,A,J,J-)$*=*F*F+'  # #  ))AAEE) $ ++r c|jj||||}|jd|jjj|S)aGenerates the ExportContext for the given args, for exporting to TDE. Args: sql_messages: module, The messages module that should be used. certificate: The certificate name; the output of the `--certificate` flag. cert_path: The certificate path in Google Cloud Storage; the output of the `--cert-path` flag. pvk_path: The private key path in Google Cloud Storage; the output of the `--pvk-path` flag. pvk_password: The password that encrypts the private key; the output of the `--pvk-password` or `--prompt-for-pvk-password` flag. Returns: ExportContext, for use in InstancesImportRequest.exportContext. )namecertificatePathprivateKeyPathprivateKeyPasswordr)rrtdeExportOptions)rTdeExportOptionsValuerTDE)r certificate cert_pathpvk_path pvk_passwordtde_export_optionss r TdeExportContextrQsb.$11GG % H'  # # ))AAEE) $ ++r )NNFFNFF)NNFNNNN)__doc__r r$r3rCrQr r rTsUI     >F#" *.)-*2Z5+p +r