@@ -105,50 +105,8 @@ def update_credentials_with_service_account(override_values={}):
105105 return transfer_config
106106
107107
108- def schedule_backfill (override_values = {}):
109- # [START bigquerydatatransfer_schedule_backfill]
110- import datetime
111-
112- from google .cloud import bigquery_datatransfer
113-
114- transfer_client = bigquery_datatransfer .DataTransferServiceClient ()
115-
116- transfer_config_name = "projects/1234/locations/us/transferConfigs/abcd"
117- # [END bigquerydatatransfer_schedule_backfill]
118- # To facilitate testing, we replace values with alternatives
119- # provided by the testing harness.
120- transfer_config_name = override_values .get (
121- "transfer_config_name" , transfer_config_name
122- )
123- # [START bigquerydatatransfer_schedule_backfill]
124- now = datetime .datetime .now (datetime .timezone .utc )
125- start_time = now - datetime .timedelta (days = 5 )
126- end_time = now - datetime .timedelta (days = 2 )
127-
128- # Some data sources, such as scheduled_query only support daily run.
129- # Truncate start_time and end_time to midnight time (00:00AM UTC).
130- start_time = datetime .datetime (
131- start_time .year , start_time .month , start_time .day , tzinfo = datetime .timezone .utc
132- )
133- end_time = datetime .datetime (
134- end_time .year , end_time .month , end_time .day , tzinfo = datetime .timezone .utc
135- )
136-
137- response = transfer_client .schedule_transfer_runs (
138- parent = transfer_config_name ,
139- start_time = start_time ,
140- end_time = end_time ,
141- )
142-
143- print ("Started transfer runs:" )
144- for run in response .runs :
145- print (f"backfill: { run .run_time } run: { run .name } " )
146- # [END bigquerydatatransfer_schedule_backfill]
147- return response .runs
148-
149-
150108def schedule_backfill_manual_transfer (override_values = {}):
151- # [START bigquerydatatransfer_start_manual_transfer ]
109+ # [START bigquerydatatransfer_schedule_backfill ]
152110 import datetime
153111
154112 from google .cloud .bigquery_datatransfer_v1 import (
@@ -161,13 +119,13 @@ def schedule_backfill_manual_transfer(override_values={}):
161119
162120 # Replace with your transfer configuration name
163121 transfer_config_name = "projects/1234/locations/us/transferConfigs/abcd"
164- # [END bigquerydatatransfer_start_manual_transfer ]
122+ # [END bigquerydatatransfer_schedule_backfill ]
165123 # To facilitate testing, we replace values with alternatives
166124 # provided by the testing harness.
167125 transfer_config_name = override_values .get (
168126 "transfer_config_name" , transfer_config_name
169127 )
170- # [START bigquerydatatransfer_start_manual_transfer ]
128+ # [START bigquerydatatransfer_schedule_backfill ]
171129 now = datetime .datetime .now (datetime .timezone .utc )
172130 start_time = now - datetime .timedelta (days = 5 )
173131 end_time = now - datetime .timedelta (days = 2 )
@@ -199,7 +157,7 @@ def schedule_backfill_manual_transfer(override_values={}):
199157 print ("Started manual transfer runs:" )
200158 for run in response .runs :
201159 print (f"backfill: { run .run_time } run: { run .name } " )
202- # [END bigquerydatatransfer_start_manual_transfer ]
160+ # [END bigquerydatatransfer_schedule_backfill ]
203161 return response .runs
204162
205163
0 commit comments