Add functionality to future_pto_calc, remove calendar_conflict step from goal

tools/future_pto_calc.py

@@ -1,17 +1,57 @@
+import json
+import pandas
+from pathlib import Path
+from datetime import date, datetime
+from dateutil.relativedelta import relativedelta
+
+
 def future_pto_calc(args: dict) -> dict:
     
-    start_date = args.get("start_date")
-    end_date = args.get("end_date")
+    file_path = Path(__file__).resolve().parent / "data" / "employee_pto_data.json"
+    if not file_path.exists():
+        return {"error": "Data file not found."}
+    
+    start_date = datetime.strptime(args.get("start_date"), "%Y-%m-%d").date()
+    end_date = datetime.strptime(args.get("end_date"), "%Y-%m-%d").date()
+    email = args.get("email")
 
-    # get rate of accrual - need email?
-    # get total hrs of PTO available as of start date (accrual * time between today and start date)
-        # take into account other booked PTO??
-    # calculate number of business hours of PTO: between start date and end date
+    #Next, set up the ability to calculate how much PTO will be added to the user's total by the start of the PTO request
+    today = date.today()
 
-    # enough_pto = total PTO as of start date - num biz hours of PTO > 0
-    # pto_hrs_remaining_after = total PTO as of start date - num biz hours of PTO
+    if today > start_date:
+        return_msg = "PTO start date " + args.get("start_date") + "cannot be in the past"
+        return {"error": return_msg}
+    
+    if end_date < start_date:
+        return_msg = "PTO end date " + args.get("end_date") + " must be after PTO start date " + args.get("start_date")
+        return {"error": return_msg}
+    
+    #Get the number of business days, and then business hours (assume 8 hr biz day), included in the PTO request
+    biz_days_of_request = len(pandas.bdate_range(start=start_date, end=end_date, inclusive="both"))
+    biz_hours_of_request = biz_days_of_request * 8
+    
+    #Assume PTO is added on the first of every month - month math compares rolling dates, so compare the PTO request with the first day of the current month.
+    today_first_of_month = date(today.year, today.month, 1)
+    time_difference = relativedelta(start_date, today_first_of_month)
+    months_to_accrue = time_difference.years * 12 + time_difference.months
+    
+    data = json.load(open(file_path))
+    employee_list = data["theCompany"]["employees"]
 
-    return {
-        "enough_pto": True, 
-        "pto_hrs_remaining_after": 410,
-    }
+    enough_pto = False
+
+    for employee in employee_list:
+        if employee["email"] == email:
+            current_pto_hours = int(employee["currentPTOHrs"])
+            hrs_added_per_month = int(employee["hrsAddedPerMonth"])
+            pto_available_at_start = current_pto_hours + (months_to_accrue * hrs_added_per_month)
+            pto_hrs_remaining_after = pto_available_at_start - biz_hours_of_request
+            if pto_hrs_remaining_after >= 0:
+                enough_pto = True
+            return {
+                "enough_pto": enough_pto, 
+                "pto_hrs_remaining_after": str(pto_hrs_remaining_after),
+            }
+
+    return_msg = "Employee not found with email address " + email
+    return {"error": return_msg}