Files
pms/pms_api_rest/services/pms_sale_channel_service.py

96 lines
3.2 KiB
Python

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 PmsSaleChannelService(Component):
_inherit = "base.rest.service"
_name = "pms.sale.channel.service"
_usage = "sale-channels"
_collection = "pms.services"
@restapi.method(
[
(
[
"/",
],
"GET",
)
],
input_param=Datamodel("pms.sale.channel.search.param"),
output_param=Datamodel("pms.sale.channel.info", is_list=True),
auth="jwt_api_pms",
)
def get_sale_channels(self, sale_channel_search_param):
sale_channels_all_properties = self.env["pms.sale.channel"].search(
[("pms_property_ids", "=", False)]
)
if sale_channel_search_param.pmsPropertyIds:
sale_channels = set()
for index, prop in enumerate(sale_channel_search_param.pmsPropertyIds):
sale_channels_with_query_property = self.env["pms.sale.channel"].search(
[("pms_property_ids", "=", prop)]
)
if index == 0:
sale_channels = set(sale_channels_with_query_property.ids)
else:
sale_channels = sale_channels.intersection(
set(sale_channels_with_query_property.ids)
)
sale_channels_total = list(
set(list(sale_channels) + sale_channels_all_properties.ids)
)
else:
sale_channels_total = list(sale_channels_all_properties.ids)
domain = [
("id", "in", sale_channels_total),
]
result_sale_channels = []
PmsSaleChannelInfo = self.env.datamodels["pms.sale.channel.info"]
for sale_channel in self.env["pms.sale.channel"].search(
domain,
):
result_sale_channels.append(
PmsSaleChannelInfo(
id=sale_channel.id,
name=sale_channel.name if sale_channel.name else None,
channelType=sale_channel.channel_type
if sale_channel.channel_type
else None,
icon=sale_channel.icon
if sale_channel.icon
else None,
)
)
return result_sale_channels
@restapi.method(
[
(
[
"/<int:sale_channel_id>",
],
"GET",
)
],
output_param=Datamodel("pms.sale.channel.info", is_list=False),
auth="jwt_api_pms",
)
def get_sale_channel(self, sale_channel_id):
sale_channel = self.env["pms.sale.channel"].search(
[("id", "=", sale_channel_id)]
)
if sale_channel:
PmsSaleChannelInfo = self.env.datamodels["pms.sale.channel.info"]
return PmsSaleChannelInfo(
id=sale_channel.id,
name=sale_channel.name if sale_channel else None,
)
else:
raise MissingError(_("Sale Channel not found"))