Ross
119 lines
3.5 KiB
Python
119 lines
3.5 KiB
Python
import csv
|
|
import re
|
|
|
|
date_re = r"[\d]{1,2}\/[\d]{1,2}\/[\d]{2}"
|
|
|
|
|
|
def load_leave(Rota, use_live_rota):
|
|
|
|
file_name = "leave.csv"
|
|
|
|
with open(file_name, newline="") as f:
|
|
shifts = Rota.get_shift_names()
|
|
|
|
workers = {}
|
|
|
|
reader = csv.reader(f, delimiter=",", quotechar='"')
|
|
|
|
leave = {}
|
|
requests = {}
|
|
site_prefs = {}
|
|
|
|
order = []
|
|
|
|
n = 0
|
|
for row in reader:
|
|
row_title = row[0]
|
|
r = row[2:]
|
|
# header, extract names
|
|
if n == 1:
|
|
names = r
|
|
order = names
|
|
for name in names:
|
|
|
|
workers[name] = {}
|
|
workers[name]["leave"] = []
|
|
workers[name]["requests"] = []
|
|
workers[name]["site_pref"] = []
|
|
workers[name]["shift_balance_extra"] = {}
|
|
workers[name]["bank_holiday_extra"] = 0
|
|
|
|
for i in range(len(r)):
|
|
try:
|
|
a = order[i]
|
|
except IndexError:
|
|
continue
|
|
|
|
worker = workers[order[i]]
|
|
|
|
lower_item = r[i].lower()
|
|
if lower_item == "derriford":
|
|
lower_item = "plymouth"
|
|
|
|
if lower_item == "derriford twilights":
|
|
lower_item = "plymouth_twilights"
|
|
|
|
if row_title == "PROC site":
|
|
worker["site_pref"] = lower_item
|
|
|
|
if row_title == "Rotation":
|
|
worker["site"] = lower_item
|
|
|
|
elif row_title == "Grade":
|
|
worker["grade"] = lower_item
|
|
|
|
elif row_title == "%FTE":
|
|
worker["fte"] = lower_item
|
|
|
|
elif row_title == "NWD":
|
|
worker["nwd"] = lower_item
|
|
|
|
elif row_title == "Flexible NWD":
|
|
worker["flexible_nwd"] = lower_item
|
|
|
|
elif row_title == "End Date":
|
|
worker["end_date"] = lower_item
|
|
|
|
elif row_title == "OOP":
|
|
worker["oop"] = lower_item
|
|
|
|
elif row_title == "Group":
|
|
worker["group"] = lower_item
|
|
|
|
elif row_title == "Pair":
|
|
if lower_item == "":
|
|
worker["pair"] = 0
|
|
else:
|
|
worker["pair"] = int(lower_item)
|
|
|
|
elif row_title.startswith("Extra_"):
|
|
if lower_item:
|
|
shift = row_title[len("Extra_") :]
|
|
if shift == "twilight":
|
|
shift = f"{worker['site']}_{shift}"
|
|
elif shift == "weekend":
|
|
shift = f"weekend_{worker['site']}"
|
|
worker["shift_balance_extra"][shift] = float(lower_item)
|
|
|
|
elif row_title == "bank_holidays_worked":
|
|
if lower_item:
|
|
print("BH", lower_item)
|
|
worker["bank_holiday_extra"] = int(lower_item)
|
|
|
|
elif re.match(date_re, row_title) is not None:
|
|
|
|
if lower_item != "":
|
|
# This may be easier to do as a dict
|
|
if lower_item in shifts:
|
|
worker["requests"].append((row[0], lower_item))
|
|
|
|
else:
|
|
worker["leave"].append((row[0], lower_item))
|
|
|
|
n = n + 1
|
|
|
|
return workers
|
|
|
|
|
|
# load_leave()
|