Files
pms/hotel/models/inherit_res_partner.py
Dario Lodeiros a9a4e59882 First Commit
2018-07-26 13:06:48 +02:00

29 lines
1003 B
Python

# -*- coding: utf-8 -*-
# Copyright 2017 Alexandre Díaz
# Copyright 2017 Dario Lodeiros
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
from openerp import models, fields, api, _
class ResPartner(models.Model):
_inherit = 'res.partner'
reservations_count = fields.Integer('Reservations',
compute='_compute_reservations_count')
folios_count = fields.Integer('Folios', compute='_compute_folios_count')
def _compute_reservations_count(self):
hotel_reservation_obj = self.env['hotel.reservation']
for partner in self:
partner.reservations_count = hotel_reservation_obj.search_count([
('partner_id.id', '=', partner.id)
])
def _compute_folios_count(self):
hotel_folio_obj = self.env['hotel.folio']
for partner in self:
partner.folios_count = hotel_folio_obj.search_count([
('partner_id.id', '=', partner.id)
])