diff options
Diffstat (limited to 'sqlglot/dialects/databricks.py')
-rw-r--r-- | sqlglot/dialects/databricks.py | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/sqlglot/dialects/databricks.py b/sqlglot/dialects/databricks.py index 2498c62..2e058e8 100644 --- a/sqlglot/dialects/databricks.py +++ b/sqlglot/dialects/databricks.py @@ -4,6 +4,7 @@ from sqlglot import exp from sqlglot.dialects.dialect import parse_date_delta from sqlglot.dialects.spark import Spark from sqlglot.dialects.tsql import generate_date_delta_with_unit_sql +from sqlglot.tokens import TokenType class Databricks(Spark): @@ -21,3 +22,11 @@ class Databricks(Spark): exp.DateAdd: generate_date_delta_with_unit_sql, exp.DateDiff: generate_date_delta_with_unit_sql, } + + PARAMETER_TOKEN = "$" + + class Tokenizer(Spark.Tokenizer): + SINGLE_TOKENS = { + **Spark.Tokenizer.SINGLE_TOKENS, + "$": TokenType.PARAMETER, + } |