From 7d8d6bf0d6a71607066f4715ba0084a8c7868ff2 Mon Sep 17 00:00:00 2001
From: Sara Lago <saralago126@gmail.com>
Date: Tue, 1 Mar 2022 10:37:10 +0100
Subject: [PATCH] [IMP] pms_api_rest: add daily invoincing service

---
 pms_api_rest/services/pms_calendar_service.py | 44 +++++++++++++++++++
 1 file changed, 44 insertions(+)

diff --git a/pms_api_rest/services/pms_calendar_service.py b/pms_api_rest/services/pms_calendar_service.py
index 44b28cb9c..60960b2ec 100644
--- a/pms_api_rest/services/pms_calendar_service.py
+++ b/pms_api_rest/services/pms_calendar_service.py
@@ -128,3 +128,47 @@ class PmsCalendarService(Component):
 
         lines_room_a._compute_occupies_availability()
         lines_room_b._compute_occupies_availability()
+
+    @restapi.method(
+        [
+            (
+                [
+                    "/daily-invoicing",
+                ],
+                "GET",
+            )
+        ],
+        input_param=Datamodel("pms.calendar.search.param", is_list=False),
+        auth="jwt_api_pms",
+    )
+    def get_daily_invoincing(self, pms_calendar_search_param):
+        reservation_lines = self.env["pms.reservation.line"].search(
+            [
+                ("date", ">=", pms_calendar_search_param.date_from),
+                ("date", "<=", pms_calendar_search_param.date_to),
+                ("pms_property_id", "=", pms_calendar_search_param.pms_property_id),
+            ]
+        )
+        service_lines = self.env["pms.service.line"].search(
+            [
+                ("date", ">=", pms_calendar_search_param.date_from),
+                ("date", "<=", pms_calendar_search_param.date_to),
+                ("pms_property_id", "=", pms_calendar_search_param.pms_property_id),
+            ]
+        )
+
+        date_from = datetime.strptime(pms_calendar_search_param.date_from, '%Y-%m-%d').date()
+        date_to = datetime.strptime(pms_calendar_search_param.date_to, '%Y-%m-%d').date()
+
+        result = []
+        for day in (date_from + timedelta(d) for d in range((date_to-date_from).days+1)):
+            reservation_lines_by_day = reservation_lines.filtered(lambda d: d.date == day)
+            service_lines_by_day = service_lines.filtered(lambda d: d.date == day)
+            daily_invoicing = {
+                "date": str(day),
+                "invoicing_total":
+                    sum(reservation_lines_by_day.mapped("price"))+sum(service_lines_by_day.mapped("price_day_total"))
+            }
+            result.append(daily_invoicing)
+
+        return result