|
13 | 13 | # limitations under the License. |
14 | 14 |
|
15 | 15 |
|
16 | | -def update_config(override_values={}): |
17 | | - # [START bigquerydatatransfer_update_config] |
18 | | - from google.cloud import bigquery_datatransfer |
19 | | - from google.protobuf import field_mask_pb2 |
20 | | - |
21 | | - transfer_client = bigquery_datatransfer.DataTransferServiceClient() |
22 | | - |
23 | | - transfer_config_name = "projects/1234/locations/us/transferConfigs/abcd" |
24 | | - new_display_name = "My Transfer Config" |
25 | | - # [END bigquerydatatransfer_update_config] |
26 | | - # To facilitate testing, we replace values with alternatives |
27 | | - # provided by the testing harness. |
28 | | - new_display_name = override_values.get("new_display_name", new_display_name) |
29 | | - transfer_config_name = override_values.get( |
30 | | - "transfer_config_name", transfer_config_name |
31 | | - ) |
32 | | - # [START bigquerydatatransfer_update_config] |
33 | | - |
34 | | - transfer_config = bigquery_datatransfer.TransferConfig(name=transfer_config_name) |
35 | | - transfer_config.display_name = new_display_name |
36 | | - |
37 | | - transfer_config = transfer_client.update_transfer_config( |
38 | | - { |
39 | | - "transfer_config": transfer_config, |
40 | | - "update_mask": field_mask_pb2.FieldMask(paths=["display_name"]), |
41 | | - } |
42 | | - ) |
43 | | - |
44 | | - print(f"Updated config: '{transfer_config.name}'") |
45 | | - print(f"New display name: '{transfer_config.display_name}'") |
46 | | - # [END bigquerydatatransfer_update_config] |
47 | | - # Return the config name for testing purposes, so that it can be deleted. |
48 | | - return transfer_config |
49 | | - |
50 | | - |
51 | 16 | def update_credentials_with_service_account(override_values={}): |
52 | 17 | # [START bigquerydatatransfer_update_credentials] |
53 | 18 | from google.cloud import bigquery_datatransfer |
|
0 commit comments