^PSrSSKJr SSKJr SSKJr SSKrSSKrSSKJrJ r J r J r SSK J r SSKJr SS KJr SS KJr SS KJr SS KJr \R*rS rSr"SS5rS\ R2S\4SjrS\ R2S\4SjrS=S\ R2S\ \R:R<S\ \S\ \S\ \S\ \4 Sjjr S>S\ R2S\ \R:RBS\ \S\ \S\ \S\ \4 Sjjr"S>S\ R2S\R:RFS \ \S\ \S\ \4 S!jjr$S\ R2S\R:RBS"\ \S#\ \S$\ \4 S%jr%S\ R2SS&S'\&4S(jr'S\ R2S)\S*\4S+jr(S?S\ R2S,\ "S-S.\ \4/5S\R:RBS/\ \S0\ \S1\ \S2\ \S3\ \ \\4S4\ \S5\ \S6\ \S7\ \4S8jjr)S@S\ R2S\S9\S/\ \S0\ \S1\ \S2\ \S3\ \ \\4S4\ \S6\ \S7\ \S5\ \S\ \4S:jjr*SAS\ R2S\R:RBS;\&4S<jjr+g)Bz.The BigQuery CLI data transfer client library.)absolute_import)division)print_functionN)AnyDict NamedTupleOptional) discovery)utils) bq_api_utils)bq_error) bq_id_utils)bq_processor_utilsauthorization_code version_infoc \rSrSrSrSS\\S\\S\\S\\S\\4 S jjrSS \\ \\ 4S \ \\ 44S jjr SS \\ \\4S \ \\ 44S jjr S\4Sjr SrS\S \ \\44SjrSrg)TransferScheduleArgsz.Arguments to customize data transfer schedule.Nschedule start_timeend_timedisable_auto_schedulingevent_driven_schedulec@XlX lX0lX@lXPlgN)rrrrr)selfrrrrrs +platform/bq/clients/client_data_transfer.py__init__TransferScheduleArgs.__init__sM OM#: !6options_to_copyreturncUR5 0nUR(a URUR5US'U$UR(a0US'U$0US'U(aSU;a[ US5US'UR (aUR USS'UR (a!URUR 5USS'UR(a!URUR5USS'U$)a-Returns a dictionary of schedule options v2. Args: options_to_copy: Existing options to copy from. Returns: A dictionary of schedule options v2 expected by the bigquery.transfers.create and bigquery.transfers.update API methods. Raises: bq_error.BigqueryError: If shedule options conflict. eventDrivenSchedulemanualScheduletimeBasedScheduler startTimeendTime) _validate_schedule_optionsr_process_event_driven_schedulerdictrr_time_or_infitityrrr!optionss rto_schedule_options_v2_payload3TransferScheduleArgs.to_schedule_options_v2_payload,s  ##%G !!'+'J'J  $ $(g#$* N%  % %"$g" N&(g!" 0OC'+ / 0( #$ 37==#$Z0 484J4J OO5 #$[1 262H2H MM3 #$Y/ Nr c[U=(d 05nURbURUR5US'URbURUR5US'URUS'U$)zReturns a dictionary of schedule options. Args: options_to_copy: Existing options to be copied. Returns: A dictionary of schedule options expected by the bigquery.transfers.create and bigquery.transfers.update API methods. r'r(disableAutoScheduling)r+rr,rrr-s rto_schedule_options_payload0TransferScheduleArgs.to_schedule_options_payloadVso?(b)G "!33DOODgk }} 11$--@gi'+'C'CG #$ Nr time_strcU=(d S$)zCReturns None to indicate Inifinity, if time_str is an empty string.N)rr5s rr,&TransferScheduleArgs._time_or_infitityos  tr c[URURUR/5nURSLn[ UR UU/5S:a[R"S5eg)zgValidates schedule options. Raises: bq_error.BigqueryError: If the given schedule options conflict. NzThe provided scheduling options conflict. Please specify one of no_auto_scheduling, time-based schedule or event-driven schedule.) anyrrrrsumrr BigqueryError)ris_time_based_scheduleis_event_driven_schedules rr)/TransferScheduleArgs._validate_schedule_optionsss~ ! 7 $99E  ( ( " $     " " O  r c[R"U5nSU;a[R"S5eU$![an[R"S5UeSnAff=f)aProcesses the event_driven_schedule given in JSON format. Args: event_driven_schedule: The user specified event driven schedule. This should be in JSON format given as a string. Ex: --event_driven_schedule='{"pubsub_subscription":"subscription"}'. Returns: parsed_event_driven_schedule: The parsed event driven schedule. Raises: bq_error.BigqueryError: If there is an error with the given params. z9Event driven schedule should be specified in JSON format.Npubsub_subscriptionz 4       798<<(:B?@% X %#m*<{*J&\\/"" <<8 99  r rrrun_timecL[R"U[RRSS9 [ U5nU(aSU0nOSX#S.0nUR 5R 5R5nURXVS9R5nURS5$)aRStarts manual transfer runs. Args: transfer_client: the transfer client to use. reference: Transfer configuration name for the run. start_time: Start time of the range of transfer runs. end_time: End time of the range of transfer runs. run_time: Specific time for a transfer run. Returns: The list of started transfer runs. start_manual_transfer_runsrhrequestedRunTimerequestedTimeRange)r'r()parentbodyr_) rrkrlrrNrWrXrYstartManualRunsr[rZ) rQrarrrrrconfigs_requestrs rrr~s&!!99 ) y>&  )D JL D$,,.88:JJL/  , ,F , F N N P( f r z3bq_id_utils.ApiClientHelper.TransferConfigReferencer"c8[R"U[RRSS9 UR 5R 5R 5RURS9R5 g![Ra gf=f)z$Returns true if the transfer exists.transfer_existsrhrUTF) rrkrlrrWrXrYrZtransferConfigNamer[r BigqueryNotFoundError)rQras rrrs !!99  ((*::<@@  ) )A gi   ' ' sABBBproject_referencedata_source_idcUS-U-nUR5R5R5RUS9R 5$)Nz/locations/-/dataSources/rU)rWrX dataSourcesrZr[)rQrrdata_source_retrievals r_fetch_data_sourcersL 55F  y{{} s%s&wy r id_fallbacksIDS project_idtarget_dataset display_namerefresh_window_daysparams auth_infoservice_account_namedestination_kms_keynotification_pubsub_topic schedule_argsc [R"U[RRSS9 S[R "US9R -n [XR5n /n0nU SUS'U(aX?S'URS5 U(aURS5 XOS 'U(a'[R"Xo5nURS 5 U(a<[X U S5n[R"UUUU S5nURS 5 U (a4U RU RS 55US 'URS 5 U (aXS'URS5 Ub[ U;aUR[ 5 Ub["U;aUR["5 U(aURS5 U (aSSU 00US'URS5 UR%5R'5R)5R+UURSR-U5UcSOUR[ 5UcSOUR["5USS9R/5 g![0R2aXnU(aGS[5U5;a3[R6"XS9n[0R8"SU<3SS0/5UeSnAgSnAgSnAff=f)aUpdates a transfer config. Args: transfer_client: the transfer client to use. id_fallbacks: IDs to use when they have not been explicitly specified. reference: the TransferConfigReference to update. target_dataset: Optional updated target dataset. display_name: Optional change to the display name. refresh_window_days: Optional update to the refresh window days. Some data sources do not support this. params: Optional parameters to update. auth_info: A dict contains authorization info which can be either an authorization_code or a version_info that the user input if they want to update credentials. service_account_name: The service account that the user could act as and used as the credential to create transfer runs from the transfer config. destination_kms_key: Optional KMS key for encryption. notification_pubsub_topic: The Pub/Sub topic where notifications will be sent after transfer runs associated with this transfer config finish. schedule_args: Optional parameters to customize data transfer schedule. Raises: BigqueryTypeError: if reference is not a TransferConfigReference. bq_error.BigqueryError: required field not given. update_transfer_configrhz projects/)r dataSourceIddestinationDatasetIdz&transfer_config.destination_dataset_idztransfer_config.display_name displayNameztransfer_config.paramsz(transfer_config.data_refresh_window_daysscheduleOptionsV2z!transfer_config.scheduleOptionsV2rz)transfer_config.notification_pubsub_topicNr kms_key_namevalueencryption_configurationz%encryption_configuration.kms_key_name,2)rrV updateMaskauthorizationCode versionInfoserviceAccountNamex__xgafvzNot found: Dataset)rr] Not found: reasonnotFound)rrkrlrbq_client_utilsGetProjectReference projectIdr\rappendrProcessParamsFlagrProcessRefreshWindowDaysFlagr/rZAUTHORIZATION_CODE VERSION_INFOrWrXrYpatchjoinr[r BigqueryInterfaceErrorrNGetDatasetReferencer)rQrrarrrrrrrrrrcurrent_config update_mask update_itemsdata_source_inforGdataset_references rrrsZ!!99 % "))|DNN'33.+,!/!?,~+9'(?@56". %77ML/0)N>,J&BB~& L AB44   2 3 $% :;0I,-BC1Y>)*|y8|$-."560L+,>?((*::<BB   ) )88K(%D9==9K+L%-D9==3N/C gi  ( (.#a&8)==#  * *. 08Z2H"   9~s/BJK2A K--K2 data_sourcec 0n U(aX=S'U(aXMS'O[R"S5eX-S'U(a#[XU5n[R"X^X5n U(a[R "Xm5n O[R"S5eU (a US-U -nOUS-nU (aU R 5U S'U (aXS 'U (a S S U 00U S 'UR5R5R5RUU UcS OUR[5UcS OUR[5US9R5nUS$)aCreate a transfer config corresponding to TransferConfigReference. Args: transfer_client: the transfer client to use. reference: the TransferConfigReference to create. data_source: The data source for the transfer config. target_dataset: The dataset where the new transfer config will exist. display_name: A display name for the transfer config. refresh_window_days: Refresh window days for the transfer config. params: Parameters for the created transfer config. The parameters should be in JSON format given as a string. Ex: --params="{'param':'value'}". The params should be the required values needed for each data source and will vary. auth_info: A dict contains authorization info which can be either an authorization_code or a version_info that the user input if they need credentials. service_account_name: The service account that the user could act as and used as the credential to create transfer runs from the transfer config. notification_pubsub_topic: The Pub/Sub topic where notifications will be sent after transfer runs associated with this transfer config finish. schedule_args: Optional parameters to customize data transfer schedule. destination_kms_key: Optional KMS key for encryption. location: The location where the new transfer config will run. Raises: BigqueryNotFoundError: if a requested item is not found. bq_error.BigqueryError: if a required field isn't provided. Returns: The generated transfer configuration name. rrz A display name must be provided.rzParameters must be provided.z /locations/z /locations/-rrrrrN)rrrrrrV)r r=rrrrr/rWrXrYcreaterZrrr[)rQrarrrrrrrrrrrb create_itemsrrnew_transfer_configs rcreate_transfer_configrPsn\,+9'(".  !C DD!,~)K&BB|L  %77ML  !? @@  & 1F 'F446$%0I,-"560L+,  y{ v'dY]];M-N }}\*1 wy$ V $$r ignore_not_foundc[R"U[RRSS9 UR 5R 5R 5RURS9R5 g![Ra0nU(d[R"SU<3SS0/5UeSnAgSnAff=f)a}Deletes TransferConfigReference reference. Args: transfer_client: the transfer client to use. reference: the TransferConfigReference to delete. ignore_not_found: Whether to ignore "not found" errors. Raises: BigqueryTypeError: if reference is not a TransferConfigReference. bq_error.BigqueryNotFoundError: if reference does not exist and ignore_not_found is False. delete_transfer_configrhrUrrrN) rrkrlrrWrXrYdeleterr[r r)rQrarrGs rrrs$!!99 % ((*::<CC  ) )D gi  ' '   * *& (8Z*@"   sABC&CC)NNNNN)NNN) NNNNNNNNN) NNNNNNNNNN)F),rM __future__rrrrCrttypingrrrr googleapiclientr clientsr rr r rrServicerrrResourcerNr\r`rlrmintrgrr}TransferRunReferencerrrOrrrrrr7r rrs4&% 22%,$   * I(I(X););#i&8&8cIM"# $%) D''D 33DDEDsmD} D  D c] DV"& $ ;''; 33KKL;#;# ;  ; SM ;B#'!% $ :'':**??:3-:# :  :z$''$**BB$ $sm $ sm $N''D (''6%)"&)- *.*.)-/348#B''B 8C= ) B**BBBSMB3-B"#B SMBS#X'B#3-B"#B (}!B"01#BR%)"&)- *.*./348)-"q%''q%q%q%SM q% 3- q% "# q% SMq%S#X'q%#3-q% (}q%01q%"#q%smq%n#''**BBr