First Commit

hotel/models/inherit_res_partner.py

@@ -0,0 +1,28 @@
+# -*- 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)
+            ])