diff --git a/taptap/tap_weekly_mail.py b/taptap/tap_weekly_mail.py index f6bdaa3..5643102 100644 --- a/taptap/tap_weekly_mail.py +++ b/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()