pms/pms_api_rest/services/pms_reservation_line_service.py

from datetime import datetime

from odoo import _
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 PmsReservationLineService(Component):
    _inherit = "base.rest.service"
    _name = "pms.reservation.line.service"
    _usage = "reservation-lines"
    _collection = "pms.services"

    @restapi.method(
        [
            (
                [
                    "/<int:reservation_line_id>",
                ],
                "GET",
            )
        ],
        output_param=Datamodel("pms.reservation.line.info", is_list=False),
        auth="jwt_api_pms",
    )
    def get_reservation_line(self, reservation_line_id):
        reservation_line = self.env["pms.reservation.line"].search(
            [("id", "=", reservation_line_id)]
        )
        if reservation_line:
            PmsReservationLineInfo = self.env.datamodels["pms.reservation.line.info"]
            return PmsReservationLineInfo(
                id=reservation_line.id,
                date=datetime.combine(
                    reservation_line.date, datetime.min.time()
                ).isoformat(),
                price=reservation_line.price,
                discount=reservation_line.discount,
                cancelDiscount=reservation_line.cancel_discount,
                roomId=reservation_line.room_id.id,
                reservationId=reservation_line.reservation_id.id,
                pmsPropertyId=reservation_line.pms_property_id.id,
            )
        else:
            raise MissingError(_("Reservation Line not found"))

    @restapi.method(
        [
            (
                [
                    "/",
                ],
                "GET",
            )
        ],
        input_param=Datamodel("pms.reservation.line.search.param"),
        output_param=Datamodel("pms.reservation.line.info", is_list=True),
        auth="jwt_api_pms",
    )
    def get_reservation_lines(self, reservation_lines_search_param):
        domain = []
        if reservation_lines_search_param.date:
            domain.append(("date", "=", reservation_lines_search_param.date))
        if reservation_lines_search_param.reservationId:
            domain.append(
                ("reservation_id", "=", reservation_lines_search_param.reservationId)
            )
        if reservation_lines_search_param.pmsPropertyId:
            domain.extend(
                [
                    (
                        "pms_property_id",
                        "=",
                        reservation_lines_search_param.pmsPropertyId,
                    ),
                ]
            )

        result_lines = []
        PmsReservationLineInfo = self.env.datamodels["pms.reservation.line.info"]
        for reservation_line in self.env["pms.reservation.line"].search(
            domain,
        ):
            result_lines.append(
                PmsReservationLineInfo(
                    id=reservation_line.id,
                    date=datetime.combine(
                        reservation_line.date, datetime.min.time()
                    ).isoformat(),
                    price=reservation_line.price,
                    discount=reservation_line.discount,
                    cancelDiscount=reservation_line.cancel_discount,
                    roomId=reservation_line.room_id.id,
                    reservationId=reservation_line.reservation_id.id,
                    pmsPropertyId=reservation_line.pms_property_id.id,
                )
            )
        return result_lines

    @restapi.method(
        [
            (
                [
                    "/<int:reservation_line_id>",
                ],
                "PATCH",
            )
        ],
        input_param=Datamodel("pms.reservation.line.info"),
        auth="jwt_api_pms",
    )
    def update_reservation_line(self, reservation_line_id, reservation_line_info):
        reservation_line = self.env["pms.reservation.line"].search(
            [("id", "=", reservation_line_id)]
        )
        vals = dict()
        if reservation_line:
            if reservation_line_info.price:
                vals["price"] = reservation_line_info.price
            if reservation_line_info.discount:
                vals["discount"] = reservation_line_info.discount
            if reservation_line_info.cancelDiscount:
                vals["cancel_discount"] = reservation_line_info.cancelDiscount
            if reservation_line_info.roomId:
                vals["room_id"] = reservation_line_info.roomId
            reservation_line.write(vals)
        else:
            raise MissingError(_("Reservation Line not found"))