from datetime import datetime, timedelta from odoo.exceptions import MissingError from odoo.addons.base_rest import restapi from odoo.addons.base_rest_datamodel.restapi import Datamodel from odoo.addons.component.core import Component class PmsAvailabilityPlanService(Component): _inherit = "base.rest.service" _name = "pms.availability.plan.service" _usage = "availability-plans" _collection = "pms.services" @restapi.method( [ ( [ "/", ], "GET", ) ], input_param=Datamodel("pms.search.param", is_list=False), output_param=Datamodel("pms.availability.plan.info", is_list=True), auth="jwt_api_pms", ) def get_availability_plans(self, pms_search_param, **args): availability_plans_all_properties = self.env["pms.availability.plan"].search( [("pms_property_ids", "=", False)] ) availabilities = set() if pms_search_param.pmsPropertyIds: for index, prop in enumerate(pms_search_param.pmsPropertyIds): availabilities_with_query_property = self.env[ "pms.availability.plan" ].search([("pms_property_ids", "=", prop)]) if index == 0: availabilities = set(availabilities_with_query_property.ids) else: availabilities = availabilities.intersection( set(availabilities_with_query_property.ids) ) availabilities_total = list( set(list(availabilities) + availability_plans_all_properties.ids) ) else: availabilities_total = list(availability_plans_all_properties.ids) domain = [ ("id", "in", availabilities_total), ] PmsAvialabilityPlanInfo = self.env.datamodels["pms.availability.plan.info"] result_availabilities = [] for availability in self.env["pms.availability.plan"].search(domain): result_availabilities.append( PmsAvialabilityPlanInfo( id=availability.id, name=availability.name, pmsPropertyIds=availability.pms_property_ids.mapped("id"), ) ) return result_availabilities @restapi.method( [ ( [ "//availability-plan-rules", ], "GET", ) ], input_param=Datamodel("pms.availability.plan.rule.search.param", is_list=False), output_param=Datamodel("pms.availability.plan.rule.info", is_list=True), auth="jwt_api_pms", ) def get_availability_plan_rules( self, availability_plan_id, availability_plan_rule_search_param ): result = [] record_availability_plan_id = self.env["pms.availability.plan"].browse( availability_plan_id ) if not record_availability_plan_id: raise MissingError PmsAvailabilityPlanRuleInfo = self.env.datamodels[ "pms.availability.plan.rule.info" ] rooms = self.env["pms.room"].search( [ ( "pms_property_id", "=", availability_plan_rule_search_param.pmsPropertyId, ) ] ) date_from = datetime.strptime( availability_plan_rule_search_param.dateFrom, "%Y-%m-%d" ).date() date_to = datetime.strptime( availability_plan_rule_search_param.dateTo, "%Y-%m-%d" ).date() for date in ( date_from + timedelta(d) for d in range((date_to - date_from).days + 1) ): for room_type in self.env["pms.room.type"].search( [("id", "in", rooms.mapped("room_type_id").ids)] ): rule = self.env["pms.availability.plan.rule"].search( [ ("date", "=", date), ( "availability_plan_id", "=", record_availability_plan_id.id, ), ("room_type_id", "=", room_type.id), ] ) if rule: availability_plan_rule_info = PmsAvailabilityPlanRuleInfo( roomTypeId=room_type.id, date=datetime.combine(date, datetime.min.time()).isoformat(), availabilityRuleId=rule.id, minStay=rule.min_stay, minStayArrival=rule.min_stay_arrival, maxStay=rule.max_stay, maxStayArrival=rule.max_stay_arrival, closed=rule.closed, closedDeparture=rule.closed_departure, closedArrival=rule.closed_arrival, quota=rule.quota, ) result.append(availability_plan_rule_info) return result @restapi.method( [ ( [ "//availability-plan-rules", ], "POST", ) ], input_param=Datamodel("pms.availability.plan.rule.info", is_list=False), auth="jwt_api_pms", ) def create_availability_plan_rule( self, availability_plan_id, pms_avail_plan_rule_info ): day = datetime.strptime( pms_avail_plan_rule_info.date[:10], "%Y-%m-%d" ) + timedelta(days=1) vals = { "room_type_id": pms_avail_plan_rule_info.roomTypeId, "date": day, "pms_property_id": pms_avail_plan_rule_info.pmsPropertyId, "availability_plan_id": availability_plan_id, } if pms_avail_plan_rule_info.minStay: vals.update({"min_stay": pms_avail_plan_rule_info.minStay}) if pms_avail_plan_rule_info.minStayArrival: vals.update({"min_stay_arrival": pms_avail_plan_rule_info.minStayArrival}) if pms_avail_plan_rule_info.maxStay: vals.update({"max_stay": pms_avail_plan_rule_info.maxStay}) if pms_avail_plan_rule_info.maxStayArrival: vals.update({"max_stay_arrival": pms_avail_plan_rule_info.maxStayArrival}) if pms_avail_plan_rule_info.closed: vals.update({"closed": pms_avail_plan_rule_info.closed}) if pms_avail_plan_rule_info.closedDeparture: vals.update({"closed_departure": pms_avail_plan_rule_info.closedDeparture}) if pms_avail_plan_rule_info.closedArrival: vals.update({"closed_arrival": pms_avail_plan_rule_info.closedArrival}) if pms_avail_plan_rule_info.quota: vals.update({"quota": pms_avail_plan_rule_info.quota}) avail_plan_rule = self.env["pms.availability.plan.rule"].create(vals) return avail_plan_rule.id @restapi.method( [ ( [ "//" "availability-plan-rules/", ], "PATCH", ) ], input_param=Datamodel("pms.availability.plan.rule.info", is_list=False), auth="jwt_api_pms", ) def write_availability_plan_rule( self, availability_plan_id, availability_plan_rule_id, pms_avail_plan_rule_info ): vals = dict() avail_rule = self.env["pms.availability.plan.rule"].search( [ ("availability_plan_id", "=", availability_plan_id), ("id", "=", availability_plan_rule_id), ] ) if avail_rule: if pms_avail_plan_rule_info.minStay: vals.update({"min_stay": pms_avail_plan_rule_info.minStay}) if pms_avail_plan_rule_info.minStayArrival: vals.update( {"min_stay_arrival": pms_avail_plan_rule_info.minStayArrival} ) if pms_avail_plan_rule_info.maxStay: vals.update({"max_stay": pms_avail_plan_rule_info.maxStay}) if pms_avail_plan_rule_info.maxStayArrival: vals.update( {"max_stay_arrival": pms_avail_plan_rule_info.maxStayArrival} ) if pms_avail_plan_rule_info.closed: vals.update({"closed": pms_avail_plan_rule_info.closed}) if pms_avail_plan_rule_info.closedDeparture: vals.update( {"closed_departure": pms_avail_plan_rule_info.closedDeparture} ) if pms_avail_plan_rule_info.closedArrival: vals.update({"closed_arrival": pms_avail_plan_rule_info.closedArrival}) if pms_avail_plan_rule_info.quota: vals.update({"quota": pms_avail_plan_rule_info.quota}) avail_rule.write(vals)