update

leave.py

@@ -1,13 +1,22 @@
 import csv
 
 
-def load_leave(Rota, file_name="leave.csv"):
+def load_leave(Rota, use_live_rota):
 
-    shifts = Rota.get_shift_names()
-    
-    workers = {}
+    if use_live_rota:
+        from gsheets import Sheets
+        sheets = Sheets.from_files('~/client_secrets.json', '~/storage.json')
+        url = 'https://docs.google.com/spreadsheets/d/1SQh433Y2lj_KMyH3Ie7CeAl_x_Iq5XQ2GHeFm9ZDXYw/'
+        s = sheets.get(url)
+        s.sheets[2].to_csv("leave.csv", encoding='utf-8', dialect='excel')
+
+    file_name="leave.csv"
 
     with open(file_name, newline="") as f:
+        shifts = Rota.get_shift_names()
+        
+        workers = {}
+
         reader = csv.reader(f, delimiter=',', quotechar='|')
 
         leave = {}
@@ -46,28 +55,28 @@ def load_leave(Rota, file_name="leave.csv"):
                     workers[order[i]]["site_pref"] = lower_item
                 
                 if row_title == "Rotation":
-                	workers[order[i]]["site"] = lower_item
-                	
+                    workers[order[i]]["site"] = lower_item
+                    
                 elif row_title == "Grade":
-                	workers[order[i]]["grade"] = lower_item
-                	
+                    workers[order[i]]["grade"] = lower_item
+                    
                 elif row_title == "%FTE":
                     workers[order[i]]["fte"] = lower_item
-                	
+                    
                 elif row_title == "NWD":
                     workers[order[i]]["nwd"] = lower_item
-                	
+                    
                 elif row_title == "Flexible NWD":
-                	workers[order[i]]["flexible_nwd"] = lower_item
-                	
+                    workers[order[i]]["flexible_nwd"] = lower_item
+                    
                 elif row_title == "End Date":
-                	workers[order[i]]["end_date"] = lower_item
-                	
+                    workers[order[i]]["end_date"] = lower_item
+                    
                 elif row_title == "OOP":
-                	workers[order[i]]["oop"] = lower_item
-                	
+                    workers[order[i]]["oop"] = lower_item
+                    
                 elif row_title == "Group":
-                	workers[order[i]]["group"] = lower_item
+                    workers[order[i]]["group"] = lower_item
 
                 elif n > 11:
 
@@ -83,5 +92,4 @@ def load_leave(Rota, file_name="leave.csv"):
 
         return workers
 
-
 #load_leave()