mirror of
https://github.com/OCA/pms.git
synced 2025-01-29 00:17:45 +02:00
127 lines
4.1 KiB
Python
127 lines
4.1 KiB
Python
# Copyright 2017 Alexandre Díaz
|
|
# Copyright 2017 Dario Lodeiros
|
|
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
|
|
import logging
|
|
|
|
from odoo import _, api, fields, models
|
|
|
|
_logger = logging.getLogger(__name__)
|
|
|
|
|
|
class ResPartner(models.Model):
|
|
_inherit = "res.partner"
|
|
|
|
# Fields declaration
|
|
main_partner_id = fields.Many2one(
|
|
"res.partner", string="Destination Partner fusion"
|
|
)
|
|
reservations_count = fields.Integer(
|
|
"Reservations", compute="_compute_reservations_count"
|
|
)
|
|
folios_count = fields.Integer("Folios", compute="_compute_folios_count")
|
|
unconfirmed = fields.Boolean("Unconfirmed", default=True)
|
|
is_agency = fields.Boolean("Is Agency")
|
|
sale_channel_id = fields.Many2one(
|
|
"pms.sale.channel",
|
|
string="Sale Channel",
|
|
ondelete="restrict",
|
|
domain=[("channel_type", "=", "indirect")],
|
|
)
|
|
default_commission = fields.Integer("Commission")
|
|
apply_pricelist = fields.Boolean("Apply Pricelist")
|
|
invoice_agency = fields.Boolean("Invoice Agency")
|
|
|
|
# Compute and Search methods
|
|
def _compute_reservations_count(self):
|
|
pms_reservation_obj = self.env["pms.reservation"]
|
|
for record in self:
|
|
record.reservations_count = pms_reservation_obj.search_count(
|
|
[
|
|
(
|
|
"partner_id.id",
|
|
"=",
|
|
record.id if isinstance(record.id, int) else False,
|
|
)
|
|
]
|
|
)
|
|
|
|
def _compute_folios_count(self):
|
|
pms_folio_obj = self.env["pms.folio"]
|
|
for record in self:
|
|
record.folios_count = pms_folio_obj.search_count(
|
|
[
|
|
(
|
|
"partner_id.id",
|
|
"=",
|
|
record.id if isinstance(record.id, int) else False,
|
|
)
|
|
]
|
|
)
|
|
|
|
# ORM Overrides
|
|
@api.model
|
|
def name_search(self, name, args=None, operator="ilike", limit=100):
|
|
if not args:
|
|
args = []
|
|
domain = [
|
|
"|",
|
|
"|",
|
|
("phone", operator, name),
|
|
("mobile", operator, name),
|
|
("email", operator, name),
|
|
]
|
|
partners = self.search(
|
|
domain + args,
|
|
limit=limit,
|
|
)
|
|
res = partners.name_get()
|
|
if limit:
|
|
limit_rest = limit - len(partners)
|
|
else:
|
|
limit_rest = limit
|
|
if limit_rest or not limit:
|
|
args += [("id", "not in", partners.ids)]
|
|
res += super(ResPartner, self).name_search(
|
|
name, args=args, operator=operator, limit=limit_rest
|
|
)
|
|
return res
|
|
|
|
@api.constrains("is_agency", "sale_channel_id")
|
|
def _check_is_agency(self):
|
|
for record in self:
|
|
if record.is_agency and not record.sale_channel_id:
|
|
raise models.ValidationError(_("Sale Channel must be entered"))
|
|
if not record.is_agency and record.sale_channel_id:
|
|
record.sale_channel_id = None
|
|
|
|
# REVIEW: problems with odoo demo data
|
|
# @api.constrains("mobile", "email")
|
|
# def _check_duplicated(self):
|
|
# for record in self:
|
|
# partner, field = record._search_duplicated()
|
|
# if partner:
|
|
# raise models.ValidationError(
|
|
# _(
|
|
# "Partner %s found with same %s (%s)",
|
|
# partner.name,
|
|
# partner._fields[field].string,
|
|
# getattr(record, field),
|
|
# )
|
|
# )
|
|
|
|
def _search_duplicated(self):
|
|
self.ensure_one()
|
|
partner = False
|
|
for field in self._get_key_fields():
|
|
if getattr(self, field):
|
|
partner = self.search(
|
|
[(field, "=", getattr(self, field)), ("id", "!=", self.id)]
|
|
)
|
|
if partner:
|
|
field = field
|
|
return partner, field
|
|
|
|
@api.model
|
|
def _get_key_fields(self):
|
|
return ["mobile", "email"]
|