google_cloud_datacatalog_search_bigquery_datasets_quickstart.py

1import argparse
2
3from google.cloud import datacatalog_v1
4
5
6def quickstart(project_id: str):
7    # [START datacatalog_quickstart]
8    # Initialize the Data Catalog client.
9    datacatalog = datacatalog_v1.DataCatalogClient()
10
11    # Create a scope for the search.
12    # See: https://cloud.google.com/data-catalog/docs/reference/rest/v1/projects.locations.entryGroups/search
13    scope = datacatalog_v1.types.SearchCatalogRequest.Scope()
14    scope.include_project_ids.append(project_id)
15
16    # Search for BigQuery datasets.
17    # The query is written in the Data Catalog search syntax.
18    # See: https://cloud.google.com/data-catalog/docs/how-to/search-reference
19    results = datacatalog.search_catalog(
20        scope=scope, query="system=bigquery type=dataset"
21    )
22
23    print("Search results:")
24    for result in results:
25        print(result)
26    # [END datacatalog_quickstart]
27
28
29if __name__ == "__main__":
30    parser = argparse.ArgumentParser(
31        description=__doc__, formatter_class=argparse.RawDescriptionHelpFormatter
32    )
33    parser.add_argument("project_id", help="Your Google Cloud project ID")
34
35    args = parser.parse_args()
36
37    quickstart(args.project_id)