From 72eae4d5b3fb69df75a66489914c7029cef2155e Mon Sep 17 00:00:00 2001 From: "Dustin C. Hatch" Date: Sun, 7 Jul 2024 18:24:33 -0500 Subject: [PATCH] Add CLI argument for selecting banks When debugging a failure for one bank's website, I often want to run the fetch for just that bank. To date, I've been commenting out the other bank, but that is silly. Now, `xactfetch` can target a subset of banks by specifying their name slug(s) as CLI arguments. --- xactfetch.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/xactfetch.py b/xactfetch.py index 9f632c2..4937daf 100644 --- a/xactfetch.py +++ b/xactfetch.py @@ -6,6 +6,7 @@ import json import logging import os import random +import sys import subprocess import urllib.parse from pathlib import Path @@ -690,10 +691,13 @@ def main() -> None: browser = pw.firefox.launch(headless=headless) page = browser.new_page() failed = False - if not download_commerce(page, end_date, token, importer): - failed = True - if not download_chase(page, end_date, token, importer): - failed = True + banks = sys.argv[1:] or list(ACCOUNTS.keys()) + if 'commerce' in banks: + if not download_commerce(page, end_date, token, importer): + failed = True + if 'chase' in banks: + if not download_chase(page, end_date, token, importer): + failed = True raise SystemExit(1 if failed else 0)