Thanks to visit codestin.com
Credit goes to github.com

Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 7 additions & 0 deletions pybigquery/sqlalchemy_bigquery.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down Expand Up @@ -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:
Expand Down