From 2c9c6afac9bd893453364281330b2c3351078c31 Mon Sep 17 00:00:00 2001
From: Pablo <pabloqb@gmail.com>
Date: Tue, 6 Aug 2019 11:54:07 +0200
Subject: [PATCH] [FIX] availability in cancelled modifications

---
 .../models/hotel_reservation/importer.py      | 26 ++++++++++++-------
 1 file changed, 16 insertions(+), 10 deletions(-)

diff --git a/hotel_channel_connector_wubook/models/hotel_reservation/importer.py b/hotel_channel_connector_wubook/models/hotel_reservation/importer.py
index 1d656e543..6f4112c24 100644
--- a/hotel_channel_connector_wubook/models/hotel_reservation/importer.py
+++ b/hotel_channel_connector_wubook/models/hotel_reservation/importer.py
@@ -273,6 +273,18 @@ class HotelReservationImporter(Component):
 
         return checkin_utc_dt, checkout_utc_dt
 
+    def _force_update_availability_wubook(self, binding):
+        # WuBook always add +1 in the channel manager for cancelled reservation
+        # However, the quota in Odoo has preference in the availability
+        cancelled_dates = binding.reservation_line_ids.mapped('date')
+        channel_availability = self.env['channel.hotel.room.type.availability'].search([
+            ('backend_id', '=', binding.backend_id.id),
+            ('date', 'in', cancelled_dates)
+        ])
+        channel_availability.write({'channel_pushed': False})
+        # Force an update with the correct availability
+        channel_availability.push_availability(binding.backend_id)
+
     def _update_reservation_binding(self, binding, book):
         is_cancellation = book['status'] in WUBOOK_STATUS_BAD
         binding.with_context({'connector_no_export': True}).write({
@@ -291,16 +303,7 @@ class HotelReservationImporter(Component):
             binding.odoo_id.with_context({
                 'connector_no_export': True,
                 'ota_limits': False}).action_cancel()
-            # WuBook always add +1 in the channel manager for cancelled reservation
-            # However, the quota in Odoo has preference in the availability
-            cancelled_dates = binding.reservation_line_ids.mapped('date')
-            channel_availability = self.env['channel.hotel.room.type.availability'].search([
-                ('backend_id', '=', binding.backend_id.id),
-                ('date', 'in', cancelled_dates)
-            ])
-            channel_availability.write({'channel_pushed': False})
-            # Force an update with the correct availability
-            channel_availability.push_availability(binding.backend_id)
+            self._force_update_availability_wubook(binding)
         elif binding.state == 'cancelled':
             binding.with_context({
                 'connector_no_export': True,
@@ -357,6 +360,8 @@ class HotelReservationImporter(Component):
                 reservations -= reservation
             else:
                 new_books.append(broom)
+                # Review quota if new reservation is a modification not recognized in this method
+                # because quota is __always__ decreased when creating reservation, even in the overlapped days
         return new_books, reservations
 
 
@@ -431,6 +436,7 @@ class HotelReservationImporter(Component):
                                 'connector_no_export': True,
                                 'ota_limits': False,
                                 'no_penalty': True}).action_cancel()
+                            self._force_update_availability_wubook(res.odoo_id)
                     if len(new_books) == 0:
                         processed_rids.append(rcode)
                         continue