Merge pull request googleapis#15 from shopkick/fix_autogenerated_metrics

mxmzdlv · web-flow · commit 94f39a99117c · 2018-05-14T12:24:53.000+12:00
Fix autogenerated metrics
diff --git a/pybigquery/sqlalchemy_bigquery.py b/pybigquery/sqlalchemy_bigquery.py
@@ -11,6 +11,7 @@
 from sqlalchemy.sql.compiler import SQLCompiler, IdentifierPreparer
 from sqlalchemy.engine.default import DefaultDialect, DefaultExecutionContext
 from sqlalchemy.engine.base import Engine
+from sqlalchemy.sql.schema import Column
 
 
 class UniversalSet(object):
@@ -67,6 +68,12 @@ def create_cursor(self):
 
 
 class BigQueryCompiler(SQLCompiler):
+    def __init__(self, dialect, statement, column_keys=None,
+                 inline=False, **kwargs):
+        if isinstance(statement, Column):
+             kwargs['compile_kwargs'] = util.immutabledict({'include_table': False})
+        super(BigQueryCompiler, self).__init__(dialect, statement, column_keys, inline, **kwargs)
+
     def visit_label(self, *args, **kwargs):
         # Use labels in GROUP BY clause
         if len(kwargs) == 0 or len(kwargs) == 1:
