Add week ending and week start to grain

apache · Mar 21, 2016 · a9ba47f · a9ba47f
1 parent f09f3d7
commit a9ba47f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/dashed/models.py b/dashed/models.py
@@ -285,6 +285,10 @@ def grains(self):
                 Grain('Time Column', '{col}'),
                 Grain('week', "date_trunc('week', CAST({col} AS DATE))"),
                 Grain('month', "date_trunc('month', CAST({col} AS DATE))"),
+                Grain('week_ending', "date_add('day', 5, \
+                    date_trunc('week', date_add('day', 1, CAST({col} AS DATE))))")
+                Grain('week_start', "date_add('day', -1, \
+                    date_trunc('week', date_add('day', 1, CAST({col} AS DATE))))")
             ),
             'mysql': (
                 Grain('Time Column', '{col}'),