prepared for cli args

This commit is contained in:
2024-12-15 11:54:58 +01:00
parent 1b4a8bcd57
commit 6c61653f6c
+15 -9
View File
@@ -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}")