diff --git a/taptap/taptap_report.py b/taptap/taptap_report.py index fef365c..82c19f6 100644 --- a/taptap/taptap_report.py +++ b/taptap/taptap_report.py @@ -168,13 +168,17 @@ class TapTapReport: def main(): - if len(sys.argv) == 2: + if len(sys.argv) == 3: day = sys.argv[1] + methods = sys.argv[2] else: day = (datetime.date.today() - datetime.timedelta(days=1)).strftime('%Y-%m-%d') tap = TapTapReport(day) - #tap.get_order() - tap.build_report() - + if methods == 'data': + tap.get_order() + elif methods == "build": + tap.build_report() + else: + print("PLS input methods in ('data','build')") if __name__ == "__main__": main()