Merging upstream version 20.1.0.

Signed-off-by: Daniel Baumann <mail@daniel-baumann.ch>

1 files changed, 9 insertions, 4 deletions

diff --git a/sqlglot/dialects/databricks.py b/sqlglot/dialects/databricks.py
index b777db0..1c10a8b 100644
--- a/sqlglot/dialects/databricks.py
+++ b/sqlglot/dialects/databricks.py
@@ -1,13 +1,18 @@
 from __future__ import annotations
 
 from sqlglot import exp, transforms
-from sqlglot.dialects.dialect import parse_date_delta, timestamptrunc_sql
+from sqlglot.dialects.dialect import (
+    date_delta_sql,
+    parse_date_delta,
+    timestamptrunc_sql,
+)
 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):
+    SAFE_DIVISION = False
+
     class Parser(Spark.Parser):
         LOG_DEFAULTS_TO_LN = True
         STRICT_CAST = True
@@ -27,8 +32,8 @@ class Databricks(Spark):
     class Generator(Spark.Generator):
         TRANSFORMS = {
             **Spark.Generator.TRANSFORMS,
-            exp.DateAdd: generate_date_delta_with_unit_sql,
-            exp.DateDiff: generate_date_delta_with_unit_sql,
+            exp.DateAdd: date_delta_sql("DATEADD"),
+            exp.DateDiff: date_delta_sql("DATEDIFF"),
             exp.DatetimeAdd: lambda self, e: self.func(
                 "TIMESTAMPADD", e.text("unit"), e.expression, e.this
             ),