From 53eb12e05fe635ef5799e02c3d4d019b49892686 Mon Sep 17 00:00:00 2001
From: mesmacosta <mesmacosta@gmail.com>
Date: Mon, 13 Apr 2020 10:33:52 -0300
Subject: [PATCH] Update Data Catalog samples to V1

---
 datacatalog/cloud-client/lookup_entry.py  | 24 +++++++++++------------
 datacatalog/cloud-client/requirements.txt |  2 +-
 2 files changed, 13 insertions(+), 13 deletions(-)

diff --git a/datacatalog/cloud-client/lookup_entry.py b/datacatalog/cloud-client/lookup_entry.py
index b6c1e321f7f4..4b6e8c5873fb 100644
--- a/datacatalog/cloud-client/lookup_entry.py
+++ b/datacatalog/cloud-client/lookup_entry.py
@@ -27,9 +27,9 @@
 def lookup_bigquery_dataset(project_id, dataset_id):
     # [START datacatalog_lookup_dataset]
     """Retrieves Data Catalog entry for the given BigQuery Dataset."""
-    from google.cloud import datacatalog_v1beta1
+    from google.cloud import datacatalog_v1
 
-    datacatalog = datacatalog_v1beta1.DataCatalogClient()
+    datacatalog = datacatalog_v1.DataCatalogClient()
 
     resource_name = '//bigquery.googleapis.com/projects/{}/datasets/{}'\
         .format(project_id, dataset_id)
@@ -42,9 +42,9 @@ def lookup_bigquery_dataset_sql_resource(project_id, dataset_id):
     """Retrieves Data Catalog entry for the given BigQuery Dataset by
     sql_resource.
     """
-    from google.cloud import datacatalog_v1beta1
+    from google.cloud import datacatalog_v1
 
-    datacatalog = datacatalog_v1beta1.DataCatalogClient()
+    datacatalog = datacatalog_v1.DataCatalogClient()
 
     sql_resource = 'bigquery.dataset.`{}`.`{}`'.format(project_id, dataset_id)
 
@@ -53,9 +53,9 @@ def lookup_bigquery_dataset_sql_resource(project_id, dataset_id):
 
 def lookup_bigquery_table(project_id, dataset_id, table_id):
     """Retrieves Data Catalog entry for the given BigQuery Table."""
-    from google.cloud import datacatalog_v1beta1
+    from google.cloud import datacatalog_v1
 
-    datacatalog = datacatalog_v1beta1.DataCatalogClient()
+    datacatalog = datacatalog_v1.DataCatalogClient()
 
     resource_name = '//bigquery.googleapis.com/projects/{}/datasets/{}' \
                     '/tables/{}'\
@@ -68,9 +68,9 @@ def lookup_bigquery_table_sql_resource(project_id, dataset_id, table_id):
     """Retrieves Data Catalog entry for the given BigQuery Table by
     sql_resource.
     """
-    from google.cloud import datacatalog_v1beta1
+    from google.cloud import datacatalog_v1
 
-    datacatalog = datacatalog_v1beta1.DataCatalogClient()
+    datacatalog = datacatalog_v1.DataCatalogClient()
 
     sql_resource = 'bigquery.table.`{}`.`{}`.`{}`'.format(
         project_id, dataset_id, table_id)
@@ -80,9 +80,9 @@ def lookup_bigquery_table_sql_resource(project_id, dataset_id, table_id):
 
 def lookup_pubsub_topic(project_id, topic_id):
     """Retrieves Data Catalog entry for the given Pub/Sub Topic."""
-    from google.cloud import datacatalog_v1beta1
+    from google.cloud import datacatalog_v1
 
-    datacatalog = datacatalog_v1beta1.DataCatalogClient()
+    datacatalog = datacatalog_v1.DataCatalogClient()
 
     resource_name = '//pubsub.googleapis.com/projects/{}/topics/{}'\
         .format(project_id, topic_id)
@@ -94,9 +94,9 @@ def lookup_pubsub_topic_sql_resource(project_id, topic_id):
     """Retrieves Data Catalog entry for the given Pub/Sub Topic by
     sql_resource.
     """
-    from google.cloud import datacatalog_v1beta1
+    from google.cloud import datacatalog_v1
 
-    datacatalog = datacatalog_v1beta1.DataCatalogClient()
+    datacatalog = datacatalog_v1.DataCatalogClient()
 
     sql_resource = 'pubsub.topic.`{}`.`{}`'.format(project_id, topic_id)
 
diff --git a/datacatalog/cloud-client/requirements.txt b/datacatalog/cloud-client/requirements.txt
index 8a41223bc3ff..11fe8151a0e8 100644
--- a/datacatalog/cloud-client/requirements.txt
+++ b/datacatalog/cloud-client/requirements.txt
@@ -1 +1 @@
-google-cloud-datacatalog==0.5.0
+google-cloud-datacatalog==0.7.0