diff options
Diffstat (limited to 'sqlglot/dialects/tableau.py')
-rw-r--r-- | sqlglot/dialects/tableau.py | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/sqlglot/dialects/tableau.py b/sqlglot/dialects/tableau.py index 45aa041..63e7275 100644 --- a/sqlglot/dialects/tableau.py +++ b/sqlglot/dialects/tableau.py @@ -1,7 +1,7 @@ -from sqlglot import exp +from __future__ import annotations + +from sqlglot import exp, generator, parser from sqlglot.dialects.dialect import Dialect -from sqlglot.generator import Generator -from sqlglot.parser import Parser def _if_sql(self, expression): @@ -20,17 +20,17 @@ def _count_sql(self, expression): class Tableau(Dialect): - class Generator(Generator): + class Generator(generator.Generator): TRANSFORMS = { - **Generator.TRANSFORMS, + **generator.Generator.TRANSFORMS, # type: ignore exp.If: _if_sql, exp.Coalesce: _coalesce_sql, exp.Count: _count_sql, } - class Parser(Parser): + class Parser(parser.Parser): FUNCTIONS = { - **Parser.FUNCTIONS, + **parser.Parser.FUNCTIONS, "IFNULL": exp.Coalesce.from_arg_list, "COUNTD": lambda args: exp.Count(this=exp.Distinct(expressions=args)), } |