use correct agg measure reference. addresses #2

DataBrewery · Jun 4, 2016 · 860dbcf · 860dbcf
1 parent 08d4b9f
commit 860dbcf
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/cubes-mongo/browser.py b/cubes-mongo/browser.py
@@ -403,7 +403,7 @@ def _date_transform(item, date_field):
                 group = phys.group
             elif function:
                 group_applicator = function["group_by"]
-                group = group_applicator(escape_level(agg.ref()))
+                group = group_applicator(escape_level(agg.measure.ref()))
             else:
                 raise ModelError("Neither function or mapping group specified "
                                  "for aggregate '%s' in cube '%s'"

diff --git a/cubes-mongo/datesupport.py b/cubes-mongo/datesupport.py
@@ -1,9 +1,12 @@
+# -*-coding=utf -*-
+
+from __future__ import absolute_import
+
 from dateutil.relativedelta import relativedelta
 from datetime import datetime, timedelta
 from dateutil.tz import *
 
 from functools import partial
-import pytz
 
 
 DATE_PARTS = ['year', 'month', 'day']