rename home to remote

2022-02-24 21:50:37 +00:00
parent 7807e3b1e8
commit 9e873ac5f7
11 changed files with 806 additions and 145 deletions
@@ -0,0 +1,118 @@
+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()