summaryrefslogtreecommitdiffstats
path: root/sqlglot/dialects/spark.py
diff options
context:
space:
mode:
Diffstat (limited to 'sqlglot/dialects/spark.py')
-rw-r--r--sqlglot/dialects/spark.py3
1 files changed, 3 insertions, 0 deletions
diff --git a/sqlglot/dialects/spark.py b/sqlglot/dialects/spark.py
index b9aaa66..7c8982b 100644
--- a/sqlglot/dialects/spark.py
+++ b/sqlglot/dialects/spark.py
@@ -52,6 +52,9 @@ class Spark(Spark2):
TRANSFORMS = {
**Spark2.Generator.TRANSFORMS,
exp.StartsWith: rename_func("STARTSWITH"),
+ exp.TimestampAdd: lambda self, e: self.func(
+ "DATEADD", e.args.get("unit") or "DAY", e.expression, e.this
+ ),
}
TRANSFORMS.pop(exp.DateDiff)
TRANSFORMS.pop(exp.Group)