From b426735fbdbe45dedff9a38987c5e541bd29b4c7 Mon Sep 17 00:00:00 2001 From: Edward Delaporte Date: Thu, 26 Mar 2026 17:32:03 -0500 Subject: [PATCH] Standardize URL approach --- files/pc/tasks.py | 19 +++++++++++++------ 1 file changed, 13 insertions(+), 6 deletions(-) diff --git a/files/pc/tasks.py b/files/pc/tasks.py index 79fe3fa..c2cb8e1 100644 --- a/files/pc/tasks.py +++ b/files/pc/tasks.py @@ -6,15 +6,23 @@ import os HOME = os.path.expanduser('~') TODAY = dt.now().strftime('%Y.%m.%d') -KIDS_ALLOWANCE_URL=os.environ.get('KIDS_ALLOWANCE_URL') PAYROLL = os.environ.get('PAYROLL') -BUDGET_URL = os.environ.get('BUDGET_URL') -CU_URL = os.environ.get('CU_URL') SCHOOL = f"{HOME}/School" PAYROLL = f"{HOME}/Nanny.Payroll" +expect_URLs = [ +'budget', +'cu', +'allowance' +] - +URLS = {} +for url in expect_URLs: + key = f'URL_{url.upper()}' + result = os.environ.get(key) + if not result: + raise ValueError(f'Missing expected {key} in environment.') + URLS[url] = result def browse(c, url): c.run(f"librefox {url}", asynchronous=True) @@ -83,8 +91,7 @@ def debug(c): @task def allowance(c): - browse(c, KIDS_ALLOWANCE_URL) - + browse(c, URLS['allowance']) @task def payroll(c): -- 2.47.3