|
19 | 19 | """
|
20 | 20 | Example Airflow DAG that interacts with Google Data Catalog service
|
21 | 21 | """
|
22 |
| -from google.cloud.datacatalog_v1beta1 import FieldType, TagField, TagTemplateField |
| 22 | +from google.cloud.datacatalog_v1beta1.proto.tags_pb2 import FieldType, TagField, TagTemplateField |
23 | 23 |
|
24 | 24 | from airflow import models
|
25 | 25 | from airflow.operators.bash_operator import BashOperator
|
|
91 | 91 | entry_id=ENTRY_ID,
|
92 | 92 | entry={
|
93 | 93 | "display_name": "Wizard",
|
94 |
| - "type_": "FILESET", |
| 94 | + "type": "FILESET", |
95 | 95 | "gcs_fileset_spec": {"file_patterns": ["gs://test-datacatalog/**"]},
|
96 | 96 | },
|
97 | 97 | )
|
|
144 | 144 | "display_name": "Awesome Tag Template",
|
145 | 145 | "fields": {
|
146 | 146 | FIELD_NAME_1: TagTemplateField(
|
147 |
| - display_name="first-field", type_=dict(primitive_type="STRING") |
| 147 | + display_name="first-field", type=FieldType(primitive_type="STRING") |
148 | 148 | )
|
149 | 149 | },
|
150 | 150 | },
|
|
172 | 172 | tag_template=TEMPLATE_ID,
|
173 | 173 | tag_template_field_id=FIELD_NAME_2,
|
174 | 174 | tag_template_field=TagTemplateField(
|
175 |
| - display_name="second-field", type_=FieldType(primitive_type="STRING") |
| 175 | + display_name="second-field", type=FieldType(primitive_type="STRING") |
176 | 176 | ),
|
177 | 177 | )
|
178 | 178 | # [END howto_operator_gcp_datacatalog_create_tag_template_field]
|
|
305 | 305 | # [START howto_operator_gcp_datacatalog_lookup_entry_result]
|
306 | 306 | lookup_entry_result = BashOperator(
|
307 | 307 | task_id="lookup_entry_result",
|
308 |
| - bash_command="echo \"{{ task_instance.xcom_pull('lookup_entry')['display_name'] }}\"", |
| 308 | + bash_command="echo \"{{ task_instance.xcom_pull('lookup_entry')['displayName'] }}\"", |
309 | 309 | )
|
310 | 310 | # [END howto_operator_gcp_datacatalog_lookup_entry_result]
|
311 | 311 |
|
|
0 commit comments