@@ -105,50 +105,8 @@ def update_credentials_with_service_account(override_values={}):
105
105
return transfer_config
106
106
107
107
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
-
150
108
def schedule_backfill_manual_transfer (override_values = {}):
151
- # [START bigquerydatatransfer_start_manual_transfer ]
109
+ # [START bigquerydatatransfer_schedule_backfill ]
152
110
import datetime
153
111
154
112
from google .cloud .bigquery_datatransfer_v1 import (
@@ -161,13 +119,13 @@ def schedule_backfill_manual_transfer(override_values={}):
161
119
162
120
# Replace with your transfer configuration name
163
121
transfer_config_name = "projects/1234/locations/us/transferConfigs/abcd"
164
- # [END bigquerydatatransfer_start_manual_transfer ]
122
+ # [END bigquerydatatransfer_schedule_backfill ]
165
123
# To facilitate testing, we replace values with alternatives
166
124
# provided by the testing harness.
167
125
transfer_config_name = override_values .get (
168
126
"transfer_config_name" , transfer_config_name
169
127
)
170
- # [START bigquerydatatransfer_start_manual_transfer ]
128
+ # [START bigquerydatatransfer_schedule_backfill ]
171
129
now = datetime .datetime .now (datetime .timezone .utc )
172
130
start_time = now - datetime .timedelta (days = 5 )
173
131
end_time = now - datetime .timedelta (days = 2 )
@@ -199,7 +157,7 @@ def schedule_backfill_manual_transfer(override_values={}):
199
157
print ("Started manual transfer runs:" )
200
158
for run in response .runs :
201
159
print (f"backfill: { run .run_time } run: { run .name } " )
202
- # [END bigquerydatatransfer_start_manual_transfer ]
160
+ # [END bigquerydatatransfer_schedule_backfill ]
203
161
return response .runs
204
162
205
163
0 commit comments