[IMP] pms_api_rest: add room type class service

This commit is contained in:
Sara
2022-05-12 11:46:02 +02:00
committed by Darío Lodeiros
parent 4da5273417
commit f5afcaf4e9
4 changed files with 82 additions and 0 deletions

View File

@@ -4,6 +4,7 @@ from . import pms_folio
from . import pms_room
from . import pms_room_type
from . import pms_room_type_class
from . import pms_reservation

View File

@@ -0,0 +1,17 @@
from marshmallow import fields
from odoo.addons.datamodel.core import Datamodel
class PmsRoomTypeClassSearchParam(Datamodel):
_name = "pms.room.type.class.search.param"
id = fields.Integer(required=False, allow_none=True)
name = fields.String(required=False, allow_none=True)
pms_property_ids = fields.List(fields.Integer(), required=False)
class PmsRoomTypeClassInfo(Datamodel):
_name = "pms.room.type.class.info"
id = fields.Integer(required=False, allow_none=True)
name = fields.String(required=False, allow_none=True)
pms_property_ids = fields.List(fields.Integer(), required=False)

View File

@@ -9,3 +9,4 @@ from . import pms_property_service
from . import pms_login_service
from . import pms_pricelist_service
from . import pms_availability_plan_service
from . import pms_room_type_class_service

View File

@@ -0,0 +1,63 @@
from odoo.addons.base_rest import restapi
from odoo.addons.base_rest_datamodel.restapi import Datamodel
from odoo.addons.component.core import Component
class PmsRoomTypeClassService(Component):
_inherit = "base.rest.service"
_name = "pms.room.type.class.service"
_usage = "room-type-class"
_collection = "pms.services"
@restapi.method(
[
(
[
"/",
],
"GET",
)
],
input_param=Datamodel("pms.room.type.class.search.param"),
output_param=Datamodel("pms.room.type.class.info", is_list=True),
auth="jwt_api_pms",
)
def get_room_type_class(self, room_type_class_search_param):
room_type_class_all_properties = self.env["pms.room.type.class"].search(
[("pms_property_ids", "=", False)]
)
if room_type_class_search_param.pms_property_ids:
room_type_class = set()
for index, prop in enumerate(room_type_class_search_param.pms_property_ids):
room_type_class_with_query_property = self.env[
"pms.room.type.class"
].search([("pms_property_ids", "=", prop)])
if index == 0:
room_type_class = set(room_type_class_with_query_property.ids)
else:
room_type_class = room_type_class.intersection(
set(room_type_class_with_query_property.ids)
)
room_type_class_total = list(
set(list(room_type_class) + room_type_class_all_properties.ids)
)
else:
room_type_class_total = list(room_type_class_all_properties.ids)
domain = [
("id", "in", room_type_class_total),
]
result_room_type_class = []
PmsRoomTypeClassInfo = self.env.datamodels["pms.room.type.class.info"]
for room in self.env["pms.room.type.class"].search(
domain,
):
result_room_type_class.append(
PmsRoomTypeClassInfo(
id=room.id,
name=room.name,
pms_property_ids=room.pms_property_ids.mapped("id"),
)
)
return result_room_type_class