Merge pull request tylertreat#117 from nrfk/master

tylertreat · web-flow · commit d4ccdb97cb4d · 2016-10-17T14:55:37.000-05:00
Add possibility to decide location (EU or US) of a dataset when creat…
diff --git a/bigquery/client.py b/bigquery/client.py
@@ -1655,7 +1655,7 @@ def _raise_executing_exception_if_error(self, job):
     # DataSet manipulation methods
     #
     def create_dataset(self, dataset_id, friendly_name=None, description=None,
-                       access=None):
+                       access=None, location=None):
         """Create a new BigQuery dataset.
 
         Parameters
@@ -1670,6 +1670,9 @@ def create_dataset(self, dataset_id, friendly_name=None, description=None,
         access : list, optional
             Indicating access permissions (see
             https://developers.google.com/bigquery/docs/reference/v2/datasets#resource)
+        location : str, optional
+            Indicating where dataset should be stored: EU or US (see
+            https://developers.google.com/bigquery/docs/reference/v2/datasets#resource)
 
         Returns
         -------
@@ -1682,7 +1685,8 @@ def create_dataset(self, dataset_id, friendly_name=None, description=None,
             dataset_data = self.dataset_resource(dataset_id,
                                                  friendly_name=friendly_name,
                                                  description=description,
-                                                 access=access)
+                                                 access=access,
+                                                 location=location)
 
             response = datasets.insert(projectId=self.project_id,
                                        body=dataset_data).execute()
@@ -1843,7 +1847,7 @@ def patch_dataset(self, dataset_id, friendly_name=None, description=None,
                 return {}
 
     def dataset_resource(self, ref_id, friendly_name=None, description=None,
-                         access=None):
+                         access=None, location=None):
         """See
         https://developers.google.com/bigquery/docs/reference/v2/datasets#resource
 
@@ -1857,6 +1861,8 @@ def dataset_resource(self, ref_id, friendly_name=None, description=None,
             An optional description for the dataset
         access : list, optional
             Indicating access permissions
+        location: str, optional, 'EU' or 'US'
+            An optional geographical location for the dataset(EU or US)
 
         Returns
         -------
@@ -1875,6 +1881,8 @@ def dataset_resource(self, ref_id, friendly_name=None, description=None,
             data["description"] = description
         if access:
             data["access"] = access
+        if location:
+            data["location"] = location
 
         return data
 
