update for current rota

This commit is contained in:
Ross
2024-01-01 14:56:48 +00:00
parent 8c28e7155a
commit 5d55ed77a8
2 changed files with 38 additions and 22 deletions
+11 -5
View File
@@ -21,11 +21,14 @@ def load_leave(Rota):
"Pragma": "no-cache"
}
download = s.get("https://docs.google.com/spreadsheets/d/e/2PACX-1vTQBxt0mHoVS1D5EkQcrekqGveWyTG2MCMyV1NnrSWkptqWbdI6_myaeLT1vpbkbujXJqU6Z5nrzTh9/pub?gid=0&single=true&output=csv", headers=headers)
download = s.get("https://docs.google.com/spreadsheets/d/e/2PACX-1vRYqSKq-OYW1tNnGJtx9LIMo_FXVfcHBRPF4HEMcBxWPfmXimWtr5oEDR_rIUSXrOndb9YjP3iC_4bS/pub?gid=0&single=true&output=csv", headers=headers)
decoded_content = download.content.decode('utf-8')
reader = csv.reader(decoded_content.splitlines(), delimiter=',')
#with open("/home/ross/Downloads/Draft PROC Rota March 24 - Sept 24 - Sheet1.csv") as f:
# reader = csv.reader(f.read().splitlines(), delimiter=',')
leave = {}
requests = {}
site_prefs = {}
@@ -48,6 +51,9 @@ def load_leave(Rota):
workers[name]["site_pref"] = []
workers[name]["shift_balance_extra"] = {}
workers[name]["bank_holiday_extra"] = 0
workers[name]["end_date"] = None
workers[name]["start_date"] = None
workers[name]["pair"] = None
for i in range(len(r)):
try:
@@ -64,10 +70,10 @@ def load_leave(Rota):
if lower_item == "derriford twilights":
lower_item = "plymouth_twilights"
if row_title == "PROC site":
if "PROC site" in row_title:
worker["site_pref"] = lower_item
if row_title == "Rotation":
if row_title in ("Rotation", "Site"):
worker["site"] = lower_item
elif row_title == "Grade":
@@ -76,7 +82,7 @@ def load_leave(Rota):
elif row_title == "%FTE":
worker["fte"] = lower_item
elif row_title == "NWD":
elif "NWD" in row_title:
worker["nwd"] = lower_item
elif row_title == "Flexible NWD":
@@ -97,7 +103,7 @@ def load_leave(Rota):
else:
worker["start_date"] = None
elif row_title == "OOP":
elif "OOP" in row_title:
worker["oop"] = lower_item
elif row_title == "Group":