fixup

[dotfiles.git] / files / pc / tasks.py

diff --git a/files/pc/tasks.py b/files/pc/tasks.py
index 4c0cdb483465c4c9b58dae2beaf6f171a20f0dbd..a6f0c05c2d3953de70c8db4b9b5130de04f0303f 100644 (file)
--- a/files/pc/tasks.py
+++ b/files/pc/tasks.py
@@ -6,12 +6,23 @@ import os
 
 HOME = os.path.expanduser('~')
 TODAY = dt.now().strftime('%Y.%m.%d')
 
 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')
 PAYROLL = os.environ.get('PAYROLL')

-BUDGET_URL = os.environ.get('BUDGET_URL')
-CU_URL = os.environ.get('CU_URL')

 SCHOOL = f"{HOME}/School"
 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)
 
 def browse(c, url):
     c.run(f"librefox {url}", asynchronous=True)


@@ -24,6 +35,15 @@ def code(c, path):
 def folder(c, path):
     c.run(f"nemo {path}")
 
 def folder(c, path):
     c.run(f"nemo {path}")
 

+@task
+def go(c, bookmark):
+    var = f'URL_{bookmark.upper()}'
+    url = os.environ.get(var)
+    browse(c, url)
+
+@task
+def cloud_panel(c):
+    go(c, 'cloud')

 
 @task
 def debug(c):
 
 @task
 def debug(c):


@@ -71,8 +91,7 @@ def debug(c):
 
 @task
 def allowance(c):
 
 @task
 def allowance(c):

-    browse(c, KIDS_ALLOWANCE_URL)
-
+    browse(c, URLS['allowance'])

 
 @task
 def payroll(c):
 
 @task
 def payroll(c):