From 8726822463ca892b5e8f9c571b5a847158623caa Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Dar=C3=ADo=20Lodeiros?= <dario@commitsun.com>
Date: Wed, 15 Jun 2022 12:50:17 +0200
Subject: [PATCH] [ADD]pms_api_rest: extra beds service

---
 pms_api_rest/datamodels/__init__.py           |  1 +
 pms_api_rest/datamodels/pms_extra_bed.py      | 20 ++++++
 pms_api_rest/services/__init__.py             |  1 +
 .../services/pms_extra_beds_service.py        | 72 +++++++++++++++++++
 4 files changed, 94 insertions(+)
 create mode 100644 pms_api_rest/datamodels/pms_extra_bed.py
 create mode 100644 pms_api_rest/services/pms_extra_beds_service.py

diff --git a/pms_api_rest/datamodels/__init__.py b/pms_api_rest/datamodels/__init__.py
index 429799837..8f5e0178d 100644
--- a/pms_api_rest/datamodels/__init__.py
+++ b/pms_api_rest/datamodels/__init__.py
@@ -29,3 +29,4 @@ from . import res_city_zip
 
 from . import pms_search_param
 from . import pms_ubication
+from . import pms_extra_bed
diff --git a/pms_api_rest/datamodels/pms_extra_bed.py b/pms_api_rest/datamodels/pms_extra_bed.py
new file mode 100644
index 000000000..9cfc9f7f2
--- /dev/null
+++ b/pms_api_rest/datamodels/pms_extra_bed.py
@@ -0,0 +1,20 @@
+from marshmallow import fields
+
+from odoo.addons.datamodel.core import Datamodel
+
+
+class PmsExtraBedSearchParam(Datamodel):
+    _name = "pms.extra.beds.search.param"
+    id = fields.Integer(required=False, allow_none=True)
+    name = fields.String(required=False, allow_none=True)
+    pmsPropertyId = fields.Integer(required=True, allow_none=False)
+    dateFrom = fields.String(required=False, allow_none=True)
+    dateTo = fields.String(required=False, allow_none=True)
+
+
+class PmsExtraBedInfo(Datamodel):
+    _name = "pms.extra.bed.info"
+    id = fields.Integer(required=False, allow_none=True)
+    name = fields.String(required=False, allow_none=True)
+    dailyLimitConfig = fields.Integer(required=False, allow_none=True)
+    dailyLimitAvail = fields.Integer(required=False, allow_none=True)
diff --git a/pms_api_rest/services/__init__.py b/pms_api_rest/services/__init__.py
index 694726f08..14419c344 100644
--- a/pms_api_rest/services/__init__.py
+++ b/pms_api_rest/services/__init__.py
@@ -15,3 +15,4 @@ from . import res_partner_category_services
 from . import res_city_zip_service
 from . import pms_room_type_class_service
 from . import pms_ubication_service
+from . import pms_extra_beds_service
diff --git a/pms_api_rest/services/pms_extra_beds_service.py b/pms_api_rest/services/pms_extra_beds_service.py
new file mode 100644
index 000000000..ad6d75826
--- /dev/null
+++ b/pms_api_rest/services/pms_extra_beds_service.py
@@ -0,0 +1,72 @@
+from odoo.addons.base_rest import restapi
+from odoo.addons.base_rest_datamodel.restapi import Datamodel
+from odoo.addons.component.core import Component
+
+
+class PmsExtraBedsService(Component):
+    _inherit = "base.rest.service"
+    _name = "pms.extra.beds.service"
+    _usage = "extra-beds"
+    _collection = "pms.services"
+
+    @restapi.method(
+        [
+            (
+                [
+                    "/",
+                ],
+                "GET",
+            )
+        ],
+        input_param=Datamodel("pms.extra.beds.search.param"),
+        output_param=Datamodel("pms.extra.bed.info", is_list=True),
+        auth="jwt_api_pms",
+    )
+    def get_extra_beds(self, extra_beds_search_param):
+        domain = [("is_extra_bed", "=", True)]
+        if extra_beds_search_param.name:
+            domain.append(("name", "like", extra_beds_search_param.name))
+        if extra_beds_search_param.id:
+            domain.append(("id", "=", extra_beds_search_param.id))
+        if extra_beds_search_param.pmsPropertyId:
+            domain.extend(
+                [
+                    "|",
+                    ("pms_property_ids", "in", extra_beds_search_param.pmsPropertyId),
+                    ("pms_property_ids", "=", False),
+                ]
+            )
+
+        result_extra_beds = []
+        PmsExtraBed = self.env.datamodels["pms.extra.bed.info"]
+
+        for bed in self.env["product.product"].search(
+            domain,
+        ):
+            avail = -1
+            if extra_beds_search_param.dateFrom and extra_beds_search_param.dateTo:
+                qty_for_day = self.env["pms.service.line"].read_group(
+                    [
+                        ("product_id", "=", bed.id),
+                        ("date", ">=", extra_beds_search_param.dateFrom),
+                        ("date", "<=", extra_beds_search_param.dateTo),
+                        ("cancel_discount", "=", 0),
+                    ],
+                    ["day_qty:sum"],
+                    ["date:day"],
+                )
+                max_daily_used = max(date["day_qty"] for date in qty_for_day)
+                avail = bed.daily_limit - max_daily_used
+                # Avoid send negative values in avail
+                avail = avail if avail >= 0 else 0
+
+            result_extra_beds.append(
+                PmsExtraBed(
+                    id=bed.id,
+                    name=bed.name,
+                    dailyLimitConfig=bed.daily_limit,
+                    dailyLimitAvail=avail,
+                )
+            )
+
+        return result_extra_beds