fixbug

taptap/tap_weekly_mail.py

@@ -6,6 +6,7 @@ import datetime
 import json
 from collections import defaultdict
 import pdb
+import sys
 define_logger("/data/logs/ops/tap_weekly.log")
 log = logging.getLogger(__name__)
 db_conf = {'user': 'mytga', 'pswd': 'gzVwh4HGR68G', 'host': '10.10.3.5', 'db': 'external_data'}
@@ -145,7 +146,10 @@ class TapWeeklyReport:
 
 
 def main():
-    day = (datetime.date.today() - datetime.timedelta(days=2)).strftime('%Y-%m-%d')
+    if len(sys.argv) == 2:
+        day = sys.argv[1]
+    else:
+        day = (datetime.date.today() - datetime.timedelta(days=2)).strftime('%Y-%m-%d')
     twr = TapWeeklyReport(day)
     twr.run()