1 files changed, 8 insertions, 3 deletions
diff --git a/python/download_daily.py b/python/download_daily.py
index 51cb17ba..a0f1a6e7 100644
--- a/python/download_daily.py
+++ b/python/download_daily.py
@@ -7,6 +7,12 @@ import config
 import sys
 import re
 
+try:
+    import pandas as pd
+    from pandas.tseries.offsets import BDay
+except ImportError:
+    pass
+
 if os.name =='nt':
     root = "//WDsentinel/share/Daily"
 elif os.name == 'posix':
@@ -18,8 +24,8 @@ def get_ped(s):
         PED = datetime.datetime.strptime(regex.group(1), "%Y-%m-%d").date()
     else:
         regex = re.search("(.+?)\.", s)
-        PED = datetime.datetime.strptime(regex.group(1), "%Y%m%d").date() - \
-              datetime.timedelta(1)
+        PED = pd.to_datetime(regex.group(1), "%Y%m%d") - BDay(1)
+        PED = PED.date()
     return PED
 
 def key_fun(s):
@@ -42,7 +48,6 @@ def download_data(workdate):
                 "Profit" in filename if get_ped(filename) < workdate]
     valuationfiles = [filename for filename in files if "csv" in filename and \
                       "Valuation" in filename if get_ped(filename) < workdate]
-
     pnlfile = sorted(pnlfiles, key=key_fun, reverse=True)[0]
     valuationfile = sorted(valuationfiles, key=key_fun, reverse=True)[0]