(SrSSKJr SSKJr SSKJr SSKrSSKrSSKJrJ r J r SSK J r SSK r SSKJr SS KJr SS KJr SS KJr SS KJr SS KJr SSKJr SS KJr SSKJr SSKJr SSKJr SSKJr "SS\R:5rg)zThe BigQuery CLI copy command.)absolute_import)division)print_functionN)ListOptionalTuple)flags)bigquery_client)client_dataset) client_job) client_table)utils)bigquery_command)bq_cached_client) utils_flags)utils_formatting)bq_error) bq_id_utilsc x^\rSrSrSrSrSrSrSrSr Sr S r S \ S \ R4U4S jjrS \R"S\\R(R*S\\\\ 44SjrS \R"S\\R(R*S\ S\R(R*S\ S\4 SjrS\ S\ S\\4SjrSrU=r$)Copy!z5cp [-n] [,]* z**** NOTE! **** z8Dataset %s not found. Please enter a valid dataset name.z\Warning: This operation is a cross-region copy operation. This may incur additional charges.zWarning: This operation is a cross-region copy operation. This may incur additional charges and take a long time to complete. This command is running in sync mode. It is recommended to use async mode (-sync=false) for cross-region copy operation.z1cp: Proceed with cross-region copy of %s? [y/N]: z7cp: Table %s already exists. Replace the table? [y/N]: z %s, exiting.namefvc>[[U] X5 [R"SSSSUS9 [R"SSSSUS9 [R"S SS S US9 [R "S SS US9 [R"SSSSUS9 [R"SSSSUS9 [R "SSSUS9 [R"SSSUS9 URU5 g)N no_clobberFz#Do not overwrite an existing table.n) short_name flag_valuesforcez1Ignore existing destination tables, don't prompt.f append_tablezAppend to an existing table.adestination_kms_keyz;Cloud KMS key for encryption of the destination table data.)rsnapshotz(Create a table snapshot of source table.srestorezNRestore table snapshot to a live table. Deprecated, please use clone instead.r expirationz?Expiration time, in seconds from now, of the destination table.clonezCreate a clone of source table.)superr__init__r DEFINE_boolean DEFINE_stringDEFINE_integer_ProcessCommandRc)selfrr __class__s $platform/bq/frontend/command_copy.pyr+ Copy.__init__6s $t(  -   ;   &   E    2       I   9r 2clientsource_referencesreturnc"SnSn[U5HzupVUR5n[R"URUS9nUc1[ R "UR[U54-SS0/5eUcUnMoXH:wdMvSn X44$ X44$)aChecks whether all source datasets are from same region. Args: client: Bigquery client source_references: Source reference Returns: true - all source datasets are from the same region. Includes the scenario in which there is only one source dataset false - all source datasets are not from the same region. Raises: bq_error.BigqueryNotFoundError: If unable to compute the dataset region TN apiclient referencereasonnotFoundF) enumerateGetDatasetReferencer GetDatasetRegionr:rBigqueryNotFoundError_DATASET_NOT_FOUNDstr) r0r5r6"all_source_datasets_in_same_regionfirst_source_region_valsource_dataset source_regions r2:_CheckAllSourceDatasetsInSameRegionAndGetFirstSourceRegion?Copy._CheckAllSourceDatasetsInSameRegionAndGetFirstSourceRegionms&*.&-...0n$55$$"m  ,,  # #s>':&< < z "    $+  /-2* - BB#/" . BBr4source_references_strdest_referencedestination_regioncUR5nURX5upxUc[R"URUS9nUc1[ R"UR[U54-SS0/5eU(aXX:Xag[UR[RR(aSUR-OSUR -5 UR"(agS[$R&"UR(U4-5:wa[UR*U4-5 g g![ R a!n [S[U 5-5 Sn A gSn A ff=f) aChecks if it is a Cross Region Copy operation and obtains confirmation. Args: client: Bigquery client source_references: Source reference source_references_str: Source reference string dest_reference: Destination dataset reference destination_region: Destination dataset region Returns: true - it is not a cross-region operation, or user has used force option, or cross-region operation is verified confirmed with user, or Insufficient permissions to query datasets for validation false - if user did not allow cross-region operation, or Dataset does not exist hence operation can't be performed. Raises: bq_error.BigqueryNotFoundError: If unable to compute the dataset region Nr9z^Unable to determine source or destination dataset location, skipping cross-region validation: Tr<r= yF)r?rJr r@r:rBigqueryAccessDeniedErrorprintrCrArB_NOTEbq_flagsSYNCHRONOUS_MODEvalue_SYNC_FLAG_ENABLED_WARNING_CROSS_REGION_WARNINGrfrontend_utilsPromptYN_CONFIRM_CROSS_REGION _NOT_COPYING) r0r5r6rLrMrNdestination_datasetrDrEerrs r2#shouldContinueAfterCrossRegionCheck(Copy.shouldContinueAfterCrossRegionCheckse6)<<>  I I >(  #+<<&&2E !  * *  ! !S)<%=$? ? Z   *1    $ $ * * t... D.. .   zz  n%% ""&;%== D  !6 8 89  =  - -  ' H   s5D..E#EE# source_tables dest_tablec l[RR5nURS5Vs/sHn[R "X4S9PM nnSR SU55n[R "X2S9nUR(aSnSn OUR(aSnSn OSnS n S n U (dxUR(dg[R"URUS 9n U (aAS [R"URU-5:wa[!UR"U4-5 g UR%UUUUU 5(dg Sn UR&(aSn Sn O.UR((aSn Sn OUR*(aSn Sn OSn UU [,R."5U S.n [0R2R4(a[0R2R4U S'UR6(aSUR60U S'UR8(aa[:R:R=[?UR8[@R@"5-55n[RB"U5U S'[DRF"X5U40U D6nUc[!SU<S35 g [0RHR4(dURKU5 g [MU5S:aSOSn[!S U<S!U<S"U <S#U<S$3 5 [RN"[PRR"U55 g s snf)%zCopies one table to another. Examples: bq cp dataset.old_table dataset2.new_table bq cp --destination_kms_key=kms_key dataset.old_table dataset2.new_table ,) id_fallbacks identifierz, c38# UHn[U5v M g7f)N)rC).0srcs r2 #Copy.RunWithArgs..s%L:K3c#hh:Ks WRITE_APPENDT WRITE_EMPTYWRITE_TRUNCATEFNr9rQrcopiedSNAPSHOT snapshottedRESTORErestoredCLONEclonedCOPY)write_dispositionignore_already_existsjob_idoperation_typelocation kmsKeyNameencryption_configurationdestination_expiration_timezTable 'z' already exists, skippingr%Tablez 'z' successfully z to '')*rClientGetsplitbq_client_utilsGetTableReferencejoinr!rrr get_table_regionr:rZr[_CONFIRM_OVERWRITErSr]r`r$r&r)rget_job_id_from_flagsrULOCATIONrWr#r(datetimeutcfromtimestampinttime FormatRfc3339r CopyTablerVPrintJobStartInfolenPrintJobMessagesrformat_job_info)r0rbrcr5rjr6rLrMrxryrN operationr{kwds datetime_utcjob pluralitys r2 RunWithArgsCopy.RunWithArgss $ $ ( ( *F!&&s++C ))vN+!II%L:K%LL$66N (" '"*# '88$$ ~'>'>  ! !^ 4(! d#8"::;  3 3   I }}!ni  ni nin.!6335(  D !**00d:  00*d %& &&77 doo + ,l-;,H,H -d ()   > 59 C { ^ EF  & & , , S!./!3#i -y. J %%&6&F&Fs&KLmsL1)__name__ __module__ __qualname____firstlineno__usagerTrBrYrXr\rr]rCr FlagValuesr+r BigqueryClientrrApiClientHelperTableReferencerboolrrJr`rr__static_attributes__ __classcell__)r1s@r2rr!s9 E% %@ * NP ,535E$4$45n&C,,&Ck99HHI&C T8C= ! &CPE,,Ek99HHIE! E "11@@ E  E EN^Ms^M^M ^M^Mr4r)__doc__ __future__rrrrrtypingrrrabslr rUclientsr r r r rrfrontendrrrZrrrr BigqueryCmdrrr4r2rs^$&% ((#" ,%%, %YM  ' 'YMr4