Fixup

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

diff --git a/files/pc/tasks.py b/files/pc/tasks.py
index 4c0cdb483465c4c9b58dae2beaf6f171a20f0dbd..a2a5df7770c104c493529fe9f0f4e2e17c39c164 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,20 +91,24 @@ def debug(c):
 
 @task
 def allowance(c):
 
 @task
 def allowance(c):

-    browse(c, KIDS_ALLOWANCE_URL)
+    browse(c, URLS['allowance'])
+
+@task
+def payroll_folders(c):
+    # For copying the latest file in
+    folder(c, f"{HOME}/pixel7adocuments")
+    folder(c, f"{PAYROLL}/data/payroll_inbox")
+    folder(c, f"{PAYROLL}")

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

-
-    # For copying the latest file in
-    folder(c, f"{HOME}/pixel7adocuments")
-    folder(c, f"{PAYROLL}/data/payroll_inbox")
+    payroll_folders(c)

 
     # For editing files
 
     # For editing files

-    folder_payroll(c)
+    code(c, f"{PAYROLL}")

     payroll_jupyter(c)
 
 @task
     payroll_jupyter(c)
 
 @task


@@ -93,10 +117,6 @@ def payroll_jupyter(c):
     # For running the payroll scripts
     c.run(f"cd {PAYROLL};{HOME}/envs/jupyter/bin/jupyter-lab")
 
     # For running the payroll scripts
     c.run(f"cd {PAYROLL};{HOME}/envs/jupyter/bin/jupyter-lab")
 

-@task
-def folder_payroll(c):
-    code(c, f"{PAYROLL}")
-

 
 @task()
 def screenshare(c):
 
 @task()
 def screenshare(c):