From 395998de8871d432230a15f827fad393c7e2e066 Mon Sep 17 00:00:00 2001 From: Jan Bader Date: Sun, 12 Apr 2026 20:59:06 +0200 Subject: [PATCH] only get current year by default --- downloader.py | 7 +++++++ main.py | 4 ++++ 2 files changed, 11 insertions(+) diff --git a/downloader.py b/downloader.py index e63d699..7a77a8b 100644 --- a/downloader.py +++ b/downloader.py @@ -1,7 +1,9 @@ import html import logging +import os import re import time +from datetime import datetime from pathlib import Path from playwright.sync_api import sync_playwright, Page, BrowserContext @@ -369,6 +371,11 @@ def download_all_invoices() -> list[Path]: years = get_year_options(page) if not years: years = ["all"] + else: + download_all_years = os.getenv("DOWNLOAD_ALL_YEARS", "").lower() in ("1", "true", "yes", "y") + if not download_all_years: + current_year = str(datetime.now().year) + years = [current_year] if current_year in years else years[:1] logger.info("Group '%s' - years: %s", group["label"], years) for year in years: diff --git a/main.py b/main.py index 51c6d01..120c12a 100644 --- a/main.py +++ b/main.py @@ -1,6 +1,7 @@ #!/usr/bin/env python3 import argparse import logging +import os import sys from config import Config @@ -16,6 +17,7 @@ def main(): parser.add_argument("--document-type", help="Document type to select") parser.add_argument("--company", help="Company name to fill") parser.add_argument("--notes", help="Notes to fill on invoices") + parser.add_argument("--all-years", action="store_true", help="Download invoices for all years") parser.add_argument("--verbose", "-v", action="store_true", help="Verbose logging") args = parser.parse_args() @@ -39,6 +41,8 @@ def main(): Config.INVOICE_COMPANY = args.company if args.notes: Config.INVOICE_NOTES = args.notes + if args.all_years: + os.environ["DOWNLOAD_ALL_YEARS"] = "true" try: Config.validate()