Support the new BigQuery types in charts

datalab/notebook/static/charting.ts

@@ -440,11 +440,20 @@ module Charting {
     // Convert any string fields that are date type to JS Dates.
     public static convertDates(data:any):void {
       for (var i = 0; i < data.cols.length; i++) {
-        if (data.cols[i].type == 'datetime') {
+        if (data.cols[i].type == 'date' || data.cols[i].type == 'datetime') {
           var rows = data.rows;
           for (var j = 0; j < rows.length; j++) {
             rows[j].c[i].v = new Date(rows[j].c[i].v);
           }
+        } else if (data.cols[i].type == 'timeofday') {
+          var rows = data.rows;
+          for (var j = 0; j < rows.length; j++) {
+            var timeInSeconds = rows[j].c[i].v.split('.')[0];
+            rows[j].c[i].v = timeInSeconds.split(':').map(
+              function(n:string) {
+                return parseInt(n, 10);
+              });
+          }
         }
       }
     }

datalab/utils/commands/_utils.py

@@ -98,16 +98,23 @@ def _get_cols(fields, schema):
   """ Get column metadata for Google Charts based on field list and schema. """
   typemap = {
     'STRING': 'string',
+    'INT64': 'number',
     'INTEGER': 'number',
     'FLOAT': 'number',
+    'FLOAT64': 'number',
+    'BOOL': 'boolean',
     'BOOLEAN': 'boolean',
+    'DATE': 'date',
+    'TIME': 'timeofday',
+    'DATETIME': 'datetime',
     'TIMESTAMP': 'datetime'
   }
   cols = []
   for col in fields:
     if schema:
       f = schema[col]
-      cols.append({'id': f.name, 'label': f.name, 'type': typemap[f.data_type]})
+      t = typemap.get(f.data_type, 'string')
+      cols.append({'id': f.name, 'label': f.name, 'type': t})
     else:
       # This will only happen if we had no rows to infer a schema from, so the type
       # is not really important, except that GCharts will choke if we pass such a schema