"BdZdZ ddZ ddZdZdZy) zBCommon command-agnostic utility functions for sql import commands.c|*|jjjjS|jjjj |j S)N) ImportContextBakImportOptionsValueBakTypeValueValuesEnumFULLlookup_by_nameupper) sql_messagesbak_types -lib/googlecloudsdk/api_lib/sql/import_util.py ParseBakTyper sY ""88OOTT  # # 9 9 P P _ _nn Nc |r~d}|s|r'|jjj||}|jd|||jjj||jj|||S|jd|||jjj||jj|S)a(Generates the ImportContext for the given args, for importing from SQL. Args: sql_messages: module, The messages module that should be used. uri: The URI of the bucket to import from; the output of the 'uri' arg. database: The database to import to; the output of the '--database' flag. user: The Postgres user to import as; the output of the '--user' flag. parallel: Whether to use parallel import or not; the output of the '--parallel' flag. threads: The number of threads to use; the output of the '--threads' flag. Only applicable for parallel import. clean: Clean (DROP) database objects before recreating them. Corresponds to the --clean flag on pg_restore. Only applies if --parallel is set. PostgreSQL only. if_exists: Include SQL statement (IF EXISTS) with each DROP statement produced by --clean; Corresponds to the --if-exists flag on pg_restore. Only applies if --parallel is set. PostgreSQL only. Returns: ImportContext, for use in InstancesImportRequest.importContext. N)cleanifExistssql#importContext)parallelthreadspostgresImportOptions)kinduridatabasefileType importUsersqlImportOptions)r)rSqlImportOptionsValuePostgresImportOptionsValueFileTypeValueValuesEnumSQL) r rruserrrr if_existspostgres_import_optionss r SqlImportContextr"s>"  $ $ : : % %  &   % % ++CCGG%33II"9J &    % % ++CCGG%33IIJ &  r c |jd|jj|xsg||||| |||jjj|S)a[Generates the ImportContext for the given args, for importing from CSV. Args: sql_messages: module, The messages module that should be used. uri: The URI of the bucket to import from; the output of the 'uri' arg. database: The database to import into; the output of the '--database' flag. table: The table to import into; the output of the '--table' flag. columns: The CSV columns to import form; the output of the '--columns' flag. user: The Postgres user to import as; the output of the '--user' 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: ImportContext, for use in InstancesImportRequest.importContext. r)columnstablequoteCharacterescapeCharacterfieldsTerminatedBylinesTerminatedBy)rcsvImportOptionsrrrr)rCsvImportOptionsValuerCSV) r rrr%r$rquoteescapefields_terminated_bylines_terminated_bys r CsvImportContextr1_sn>  # # #11GG-Ru 1/ H1 ))AAEE $  r c d} |rZ|rX|rV|jj|jjj|||} |r,|| j_n|jj} |r|| _|| _| | _t|| | _ | | jd| _ | | _ |jd|||jjj| S)aGenerates the ImportContext for the given args, for importing from BAK. Args: sql_messages: module, The messages module that should be used. uri: The URI of the bucket to import from; the output of the `uri` arg. database: The database to import to; the output of the `--database` flag. cert_path: The certificate used for encrypted .bak; the output of the `--cert-path` flag. pvk_path: The private key used for encrypted .bak; the output of the `--pvk-path` flag. pvk_password: The private key password used for encrypted .bak; the output of the `--pvk-password` or `--prompt-for-pvk-password` flag. keep_encrypted: Whether or not to decrypt the imported encrypted BAK file. striped: Whether or not the import is striped. no_recovery: Whether the import executes with NORECOVERY keyword. recovery_only: Whether the import skip download and bring database online. bak_type: Type of the bak file. stop_at: Equivalent to SQL Server STOPAT keyword. stop_at_mark: Equivalent to SQL Server STOPATMARK keyword. Returns: ImportContext, for use in InstancesImportRequest.importContext. N)certPathpvkPath pvkPassword)encryptionOptionsz%Y-%m-%dT%H:%M:%S.%fZr)rrrrbakImportOptions)rrEncryptionOptionsValuer6 keepEncryptedstriped noRecovery recoveryOnlyr bakTypestrftimestopAt stopAtMarkrBAK)r rr cert_pathpvk_path pvk_passwordkeep_encryptedr: no_recovery recovery_onlyr stop_at stop_at_markbak_import_optionss r BakImportContextrKs L8 %33II&44JJ  $  J ;I**8%33IIK !("-$1!+L(C  ' 0 01H I".  # #  ))AAEE) $ ++r c|jj||||}|jd|jjj|S)aGenerates the ImportContext for the given args, for importing from 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: ImportContext, for use in InstancesImportRequest.importContext. )namecertificatePathprivateKeyPathprivateKeyPasswordr)rrtdeImportOptions)rTdeImportOptionsValuerTDE)r certificaterBrCrDtde_import_optionss r TdeImportContextrVsb.$11GG % H'  # # ))AAEE) $ ++r )NNFNFF)NNNNNN)__doc__r r"r1rKrVr r rYsQI   @N" *.)-*ZC+L +r