Create a partition table for SLI/SLO. (GoogleCloudPlatform#507)

jw207427 · Jul 8, 2020 · cdbb290 · cdbb290
1 parent 894cc28
commit cdbb290
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/tools/slo-generator/setup.py b/tools/slo-generator/setup.py
@@ -29,7 +29,7 @@
     long_description = f.read()
 
 setup(name='slo-generator',
-      version='1.0.1',
+      version='1.1.0',
       description='SLO generator',
       long_description=long_description,
       long_description_content_type='text/markdown',

diff --git a/tools/slo-generator/slo_generator/exporters/bigquery.py b/tools/slo-generator/slo_generator/exporters/bigquery.py
@@ -93,6 +93,9 @@ def create_table(self, project_id, dataset_id, table_id, schema=None):
         table_name = f"{project_id}.{dataset_id}.{table_id}"
         LOGGER.info(f"Creating table {table_name}", table_name)
         table = bigquery.Table(table_name, schema=pyschema)
+        table.time_partitioning = bigquery.TimePartitioning(
+            type_=bigquery.TimePartitioningType.DAY,
+        )
         return self.client.create_table(table)