Move download all setting to config.py

This commit is contained in:
Jan Bader
2026-04-12 22:37:35 +02:00
parent 038d239033
commit 0eee28968f
3 changed files with 3 additions and 4 deletions

View File

@@ -19,6 +19,7 @@ class Config:
BROWSER_HEADLESS = os.getenv("BROWSER_HEADLESS", "true").lower() == "true" BROWSER_HEADLESS = os.getenv("BROWSER_HEADLESS", "true").lower() == "true"
BROWSER_TIMEOUT = int(os.getenv("BROWSER_TIMEOUT", "60000")) BROWSER_TIMEOUT = int(os.getenv("BROWSER_TIMEOUT", "60000"))
LOG_LEVEL = os.getenv("LOG_LEVEL", "INFO") LOG_LEVEL = os.getenv("LOG_LEVEL", "INFO")
DOWNLOAD_ALL_YEARS = os.getenv("DOWNLOAD_ALL_YEARS", "false").lower() in ("1", "true", "yes", "y")
@classmethod @classmethod
def validate(cls): def validate(cls):

View File

@@ -1,6 +1,5 @@
import html import html
import logging import logging
import os
import re import re
import time import time
from pathlib import Path from pathlib import Path
@@ -370,7 +369,7 @@ def download_all_invoices() -> list[Path]:
years = get_year_options(page) years = get_year_options(page)
if not years: if not years:
raise RuntimeError("No years found on billing page.") raise RuntimeError("No years found on billing page.")
download_all_years = os.getenv("DOWNLOAD_ALL_YEARS", "").lower() in ("1", "true", "yes", "y") download_all_years = Config.DOWNLOAD_ALL_YEARS
if not download_all_years: if not download_all_years:
years = [years[0]] years = [years[0]]
logger.info("Group '%s' - years: %s", group["label"], years) logger.info("Group '%s' - years: %s", group["label"], years)

View File

@@ -1,7 +1,6 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
import argparse import argparse
import logging import logging
import os
import sys import sys
from config import Config from config import Config
@@ -42,7 +41,7 @@ def main():
if args.notes: if args.notes:
Config.INVOICE_NOTES = args.notes Config.INVOICE_NOTES = args.notes
if args.all_years: if args.all_years:
os.environ["DOWNLOAD_ALL_YEARS"] = "true" Config.DOWNLOAD_ALL_YEARS = True
try: try:
Config.validate() Config.validate()