Fix utc time calculation if provided datetime has tz info (#1287)

apache · Oct 14, 2016 · 65c744f · 65c744f
1 parent 82bcadf
commit 65c744f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/caravel/utils.py b/caravel/utils.py
@@ -10,6 +10,7 @@
 import functools
 import json
 import logging
+import pytz
 import numpy
 import signal
 import uuid
@@ -368,6 +369,9 @@ def json_iso_dttm_ser(obj):
 
 
 def datetime_to_epoch(dttm):
+    if dttm.tzinfo:
+        epoch_with_tz = pytz.utc.localize(EPOCH)
+        return (dttm - epoch_with_tz).total_seconds() * 1000
     return (dttm - EPOCH).total_seconds() * 1000