From 07e7ac78d7bb5e281ce6764a7178741bfe6abea1 Mon Sep 17 00:00:00 2001
From: Pablo <pabloqb@gmail.com>
Date: Tue, 23 Oct 2018 13:56:49 +0200
Subject: [PATCH] [WIP] Wizard Node Reservation

---
 hotel_node_master/views/hotel_node.xml        |   5 +
 .../wizards/wizard_hotel_node_reservation.py  | 165 ++++++++++--------
 .../wizards/wizard_hotel_node_reservation.xml |  16 +-
 3 files changed, 107 insertions(+), 79 deletions(-)

diff --git a/hotel_node_master/views/hotel_node.xml b/hotel_node_master/views/hotel_node.xml
index 8bc080464..30029783f 100644
--- a/hotel_node_master/views/hotel_node.xml
+++ b/hotel_node_master/views/hotel_node.xml
@@ -8,6 +8,11 @@
             <form string="Hotel Node">
                 <sheet string="Hotel Node">
                     <div class="oe_button_box" name="button_box" groups="base.group_user">
+                        <button name="%(hotel_node_reservation_action)d" type="action"
+                                string="Reserve" help="Make a reservation in this hotel"
+                                class="oe_stat_button" icon="fa-suitcase"
+                                context="{'node_id': id}">
+                        </button>
                         <button class="oe_stat_button" type="action"
                             name="" icon="fa-tasks">
                             <field string="Tasks" name="task_count" widget="statinfo" options="{'label_field': 'label_tasks'}"/>
diff --git a/hotel_node_master/wizards/wizard_hotel_node_reservation.py b/hotel_node_master/wizards/wizard_hotel_node_reservation.py
index e6349dc14..1706e8d2e 100644
--- a/hotel_node_master/wizards/wizard_hotel_node_reservation.py
+++ b/hotel_node_master/wizards/wizard_hotel_node_reservation.py
@@ -2,6 +2,7 @@
 # Copyright 2018  Alexandre Díaz
 # Copyright 2018  Dario Lodeiros
 # License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
+from builtins import list
 
 import wdb
 import logging
@@ -21,6 +22,10 @@ class HotelNodeReservationWizard(models.TransientModel):
     _name = "hotel.node.reservation.wizard"
     _description = "Hotel Node Reservation Wizard"
 
+    @api.model
+    def _get_default_node_id(self):
+        return self._context.get('node_id') or None
+
     @api.model
     def _get_default_checkin(self):
         pass
@@ -29,7 +34,8 @@ class HotelNodeReservationWizard(models.TransientModel):
     def _get_default_checkout(self):
         pass
 
-    node_id = fields.Many2one('project.project', 'Hotel', required=True)
+    node_id = fields.Many2one('project.project', 'Hotel', required=True,
+                              default=_get_default_node_id)
 
     partner_id = fields.Many2one('res.partner', string="Customer", required=True)
 
@@ -50,84 +56,101 @@ class HotelNodeReservationWizard(models.TransientModel):
             try:
                 noderpc = odoorpc.ODOO(self.node_id.odoo_host, self.node_id.odoo_protocol, self.node_id.odoo_port)
                 noderpc.login(self.node_id.odoo_db, self.node_id.odoo_user, self.node_id.odoo_password)
+
+                today = fields.Date.context_today(self.with_context())
+
+                # TODO check hotel timezone
+                checkin = fields.Date.from_string(today).strftime(
+                    DEFAULT_SERVER_DATE_FORMAT) if not self.checkin else fields.Date.from_string(self.checkin)
+
+                checkout = (fields.Date.from_string(today) + timedelta(days=1)).strftime(
+                    DEFAULT_SERVER_DATE_FORMAT) if not self.checkout else fields.Date.from_string(self.checkout)
+
+                if checkin >= checkout:
+                    checkout = checkin + timedelta(days=1)
+
+                free_room_ids = noderpc.env['hotel.room.type'].check_availability_room_ids(checkin, checkout)
+
+                room_type_availability = {}
+                for room_type in self.node_id.room_type_ids:
+                    availability_real = noderpc.env['hotel.room'].search_count([
+                        ('id', 'in', free_room_ids),
+                        ('room_type_id', '=', room_type.remote_room_type_id),
+                    ])
+                    availability_plan = noderpc.env['hotel.room.type.availability'].search_read([
+                        ('date', '>=', checkin),
+                        ('date', '<', checkout),
+                        ('room_type_id', '=', room_type.remote_room_type_id),
+
+                    ], ['avail']) or float('inf')
+
+                    if isinstance(availability_plan, list):
+                        availability_plan = min([r['avail'] for r in availability_plan])
+
+                    room_type_availability[room_type.id] = min(
+                        availability_real, availability_plan)
+
+                cmds = self.node_id.room_type_ids.mapped(lambda room_type_id: (0, False, {
+                    'room_type_id': room_type_id.id,
+                    'checkin': checkin,
+                    'checkout': checkout,
+                    'room_type_availability': room_type_availability[room_type_id.id],
+                    'node_reservation_wizard_id': self.id,
+
+                }))
+                self.update({
+                    'checkin': checkin,
+                    'checkout': checkout,
+                    'room_type_wizard_ids': cmds,
+                })
+                noderpc.logout()
             except (odoorpc.error.RPCError, odoorpc.error.InternalError, urllib.error.URLError) as err:
                 raise ValidationError(err)
 
-            today = fields.Date.context_today(self.with_context())
-
-            # TODO check hotel timezone
-            checkin = fields.Date.from_string(today).strftime(
-                DEFAULT_SERVER_DATE_FORMAT) if not self.checkin else fields.Date.from_string(self.checkin)
-
-            checkout = (fields.Date.from_string(today) + timedelta(days=1)).strftime(
-                DEFAULT_SERVER_DATE_FORMAT) if not self.checkout else fields.Date.from_string(self.checkout)
-
-            if checkin >= checkout:
-                checkout = checkin + timedelta(days=1)
-
-            free_room_ids = noderpc.env['hotel.room.type'].check_availability_room_ids(checkin, checkout)
-
-            room_type_availability = {}
-            for room_type in self.node_id.room_type_ids:
-                room_type_availability[room_type.id] = noderpc.env['hotel.room'].search_count([
-                    ('id', 'in', free_room_ids),
-                    ('room_type_id', '=', room_type.remote_room_type_id)
-                ])
-
-            cmds = self.node_id.room_type_ids.mapped(lambda room_type_id: (0, False, {
-                'room_type_id': room_type_id.id,
-                'checkin': checkin,
-                'checkout': checkout,
-                'room_type_availability': room_type_availability[room_type_id.id],
-                'node_reservation_wizard_id': self.id,
-
-            }))
-            self.update({
-                'checkin': checkin,
-                'checkout': checkout,
-                'room_type_wizard_ids': cmds,
-            })
-            noderpc.logout()
-
     @api.multi
     def create_node_reservation(self):
         try:
             noderpc = odoorpc.ODOO(self.node_id.odoo_host, self.node_id.odoo_protocol, self.node_id.odoo_port)
             noderpc.login(self.node_id.odoo_db, self.node_id.odoo_user, self.node_id.odoo_password)
+
+            # prepare required fields for hotel folio
+            remote_partner_id = noderpc.env['res.partner'].search([('email','=',self.partner_id.email)])
+            vals = {
+                'partner_id': remote_partner_id,
+                'checkin': self.checkin,
+                'checkout': self.checkout,
+            }
+            # prepare hotel folio room_lines
+            room_lines = []
+            for line in self.room_type_wizard_ids:
+                if line.room_qty > 0:
+                    vals_reservation_lines = {
+                        'partner_id': remote_partner_id,
+                        'room_type_id': line.room_type_id.remote_room_type_id,
+                    }
+                    reservation_line_ids = noderpc.env['hotel.reservation'].prepare_reservation_lines(
+                        line.checkin,
+                        (fields.Date.from_string(line.checkout) - fields.Date.from_string(line.checkin)).days,
+                        vals_reservation_lines
+                    ) # [[5, 0, 0], ¿?
+
+                    room_lines.append((0, False, {
+                        'room_type_id': line.room_type_id.remote_room_type_id,
+                        'checkin': line.checkin,
+                        'checkout': line.checkout,
+                        'reservation_line_ids': reservation_line_ids['reservation_line_ids'],
+                    }))
+            vals.update({'room_lines': room_lines})
+
+            from pprint import pprint
+            pprint(vals)
+
+            # x = noderpc.env['hotel.reservation'].create(vals)
+
+            noderpc.logout()
         except (odoorpc.error.RPCError, odoorpc.error.InternalError, urllib.error.URLError) as err:
             raise ValidationError(err)
 
-        # prepare required fields for hotel folio
-        vals = {
-            'partner_id': self.partner_id.id,
-            'checkin': self.checkin,
-            'checkout': self.checkout,
-        }
-        # prepare hotel folio room_lines
-        room_lines = []
-        for line in self.room_type_wizard_ids:
-            if line.rooms_qty > 0:
-                vals_reservation_lines = {
-                    'partner_id': self.partner_id.id,
-                    'room_type_id': line.room_type_id.remote_room_type_id,
-                }
-                reservation_line_ids = noderpc.env['hotel.reservation'].prepare_reservation_lines(
-                    line.checkin,
-                    (fields.Date.from_string(line.checkout) - fields.Date.from_string(line.checkin)).days,
-                    vals_reservation_lines
-                )
-
-                room_lines.append((0, False, {
-                    'room_type_id': line.room_type_id.remote_room_type_id,
-                    'checkin': line.checkin,
-                    'checkout': line.checkout,
-                    'reservation_line_ids': reservation_line_ids['reservation_line_ids'],
-                }))
-        vals.update({'room_lines': room_lines})
-
-        x = noderpc.env['hotel.reservation'].create(vals)
-
-        noderpc.logout()
 
 class NodeRoomTypeWizard(models.TransientModel):
     _name = "node.room.type.wizard"
@@ -171,7 +194,7 @@ class NodeRoomTypeWizard(models.TransientModel):
         # for record in self:
         #     record.room_type_availability = 42
 #
-    @api.onchange('node_id','checkin','checkout')
+    @api.depends('node_id','checkin','checkout')
     def _onchange_dates(self):
         if self.checkin and self.checkout:
             _logger.info('_onchange_dates for room type %s', self.room_type_id)
@@ -179,9 +202,9 @@ class NodeRoomTypeWizard(models.TransientModel):
          # availability: search de hotel.room.type.availability filtrando por room_type y date y escogiendo el min avail en el rango
          # preci_unit y json_days: usando prepare_reservation_lines
 
-    @api.onchange('rooms_qty')
+    @api.onchange('room_qty')
     def _compute_price_total(self):
-        self.price_total
+        _logger.info('_compute_price_total')
         for record in self:
             record.price_total = record.room_qty * (record.price_unit * record.discount * 0.01)
          # Unidades x precio unidad (el precio de unidad ya incluye el conjunto de días)
diff --git a/hotel_node_master/wizards/wizard_hotel_node_reservation.xml b/hotel_node_master/wizards/wizard_hotel_node_reservation.xml
index 9a57be24f..ddfe3c975 100644
--- a/hotel_node_master/wizards/wizard_hotel_node_reservation.xml
+++ b/hotel_node_master/wizards/wizard_hotel_node_reservation.xml
@@ -23,19 +23,17 @@
                         </group>
                         <group colspan="2">
                             <field name="room_type_wizard_ids" nolabel="1">
-                                <tree editable="bottom" create="false" delete="false">
+                                <tree editable="bottom" create="false" delete="false"
+                                      decoration-muted="room_type_availability == 0">
                                     <field name="room_type_id" string="Room Type" readonly="1" force_save="1"/>
                                     <field name="room_type_availability" readonly="1" force_save="1"/>
                                     <field name="room_qty"/>
-                                    <field name="price_unit" widget="monetary" readonly="1" force_save="1"/>
-                                    <!--<field name="currency_id" invisible="1"/>-->
-                                    <field name="discount" widget="monetary"/>
                                     <field name="checkin" widget="date" />
                                     <field name="checkout" widget="date" />
-                                    <!--<field name="price" attrs="{'readonly': [('can_confirm','=',False)]}"/>-->
-                                    <!--<field name="amount_reservation" readonly="1" />-->
-                                    <!--<field name="total_price" invisible="1" />-->
-                                    <!--<field name="can_confirm" invisible="1" />-->
+                                    <field name="price_unit" widget="monetary" readonly="1" force_save="1"/>
+                                    <!--<field name="currency_id" invisible="1"/>-->
+                                    <field name="discount"/>
+                                    <field name="price_total" widget="monetary" readonly="1" force_save="1"/>
                                 </tree>
                             </field>
                         </group>
@@ -50,6 +48,8 @@
                     <footer attrs="{'invisible':[('node_id','=',False)]}">
                         <button name="create_node_reservation" string="Create Reservations" type="object"
 							class="oe_highlight" />
+                        <button name="cancel" string="Cancel" special="cancel"
+							class="oe_link" />
                     </footer>
 				</sheet>
 			</form>