Cleanup

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

diff --git a/files/pc/tasks.py b/files/pc/tasks.py
index c2cb8e13d220b194dce4e74241125a416550915f..a15202b315aaed8410efbdec09b9a356bb6ee6a0 100644 (file)
--- a/files/pc/tasks.py
+++ b/files/pc/tasks.py
@@ -8,7 +8,6 @@ HOME = os.path.expanduser('~')
 TODAY = dt.now().strftime('%Y.%m.%d')
 PAYROLL = os.environ.get('PAYROLL')
 SCHOOL = f"{HOME}/School"
-PAYROLL = f"{HOME}/Nanny.Payroll"
 
 expect_URLs = [
 'budget',
@@ -16,6 +15,7 @@ expect_URLs = [
 'allowance'
 ]
 
+
 URLS = {}
 for url in expect_URLs:
     key = f'URL_{url.upper()}'
@@ -94,16 +94,21 @@ def allowance(c):
     browse(c, URLS['allowance'])
 
 @task
-def payroll(c):
-
-    allowance(c)
-
+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)
+    payroll_folders(c)
 
     # For editing files
-    folder_payroll(c)
+    code(c, f"{PAYROLL}")
     payroll_jupyter(c)
 
 @task
@@ -112,10 +117,6 @@ def payroll_jupyter(c):
     # 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):
@@ -151,8 +152,8 @@ def mod_luanti(c):
 
 @task
 def budget(c):
-    browse(c, BUDGET_URL)
-    browse(c, CU_URL)
+    browse(c, URLS['budget'])
+    browse(c, URLS['cu'])
 
 
 @task