diff --git a/we_collector.py b/we_collector.py index e656ee9..0f7c363 100644 --- a/we_collector.py +++ b/we_collector.py @@ -6,6 +6,7 @@ from wci import wci from credentials import users from folder_mgr.folder_mgr import get_date, date_has_changed, FolderMgr, create_timestamp from jay_diff import jay_diff_full +import argparse INTERVAL = 1*60 @@ -92,7 +93,7 @@ class WeCollector: print(f"Exception: {e}") -def request_loop(_wc_list: list[WeCollector]): +def request_loop(wc: WeCollector): _date_last = None _file_timestamp = None @@ -106,14 +107,12 @@ def request_loop(_wc_list: list[WeCollector]): _file_timestamp = create_timestamp(with_ms=False).replace(':', '-').replace('.', '-') # Request service - for wc in wc_list: - wc.init(_file_timestamp) + wc.init(_file_timestamp) # Request service _date = get_date() timestamp = create_timestamp() - for wc in wc_list: - wc.request(_date, timestamp) + wc.request(_date, timestamp) # wait until next loop stop = time.time() @@ -125,13 +124,20 @@ def request_loop(_wc_list: list[WeCollector]): if __name__ == '__main__': - wc_list = [] + parser = argparse.ArgumentParser(prog="We Collect", description="Collects vehicle data") + parser.add_argument("username", help="Username to collect data for (must exist in credentials.py)") + parser.add_argument("--loop", help="Execute once or create event loop", type=bool, default=False) + + args = parser.parse_args() + print(args) + + wc = None for user in users: - wc_list.append(WeCollector(user)) + if args.username in user['username']: + wc = WeCollector(user) - print(wc_list) try: - request_loop(wc_list) + request_loop(wc) except Exception as e: print(f"Terminated with Exception {e}")