Merge pull request #8 from KristijanEftimov/main

When calculating the forecasted costs, the end date must be after the…
kristijaneftimovpcg · Sep 30, 2024 · 5482fa3 · 5482fa3
2 parents 183290e + aa92184
commit 5482fa3
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/lambda.py b/lambda.py
@@ -48,6 +48,9 @@ def get_forecast_dates():
         next_month = start_date.replace(day=28) + datetime.timedelta(days=4)  # Move to the next month
         end_date = next_month.replace(day=1) - datetime.timedelta(days=1)  # Last day of the current month
 
+        if start_date == end_date:
+            end_date = end_date + datetime.timedelta(days=1)
+
         return start_date, end_date
 
     #Function to get the forecast cost for the current month