Standardize URL approach

diff --git a/files/pc/tasks.py b/files/pc/tasks.py
index 79fe3fa24c0368f50a2ff96a2f52166710f84e19..c2cb8e13d220b194dce4e74241125a416550915f 100644 (file)
--- 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):