Add dataset helper methods

pirsquare · pirsquare · commit 606330e5c316 · 2015-06-06T23:38:26.000+08:00
diff --git a/bigquery/client.py b/bigquery/client.py
@@ -321,6 +321,33 @@ def get_query_rows(self, job_id, offset=None, limit=None, timeout=0):
             records += [self._transform_row(row, schema) for row in rows]
         return records
 
+    def check_dataset(self, dataset_id):
+        """Check to see if a dataset exists.
+        Args:
+            dataset: dataset unique id
+        Returns:
+            bool indicating if the table exists.
+        """
+        dataset = self.get_dataset(dataset_id)
+        return bool(dataset)
+
+    def get_dataset(self, dataset_id):
+        """
+        Retrieve a dataset if it exists, otherwise return an empty dict.
+        Args:
+            dataset: dataset unique id
+        Returns:
+            dictionary containing the dataset object if it exists, otherwise
+            an empty dictionary
+        """
+        try:
+            dataset = self.bigquery.datasets().get(
+                projectId=self.project_id, datasetId=dataset_id).execute()
+        except HttpError:
+            dataset = {}
+
+        return dataset
+
     def check_table(self, dataset, table):
         """Check to see if a table exists.
 
