Files
pms/hotel_node_master/wizards/wizard_hotel_node_reservation.py
2018-10-18 13:53:01 +02:00

199 lines
8.4 KiB
Python

# Copyright 2018 Pablo Q. Barriuso
# Copyright 2018 Alexandre Díaz
# Copyright 2018 Dario Lodeiros
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
import wdb
import logging
import urllib.error
import odoorpc.odoo
from datetime import timedelta
from odoo import models, fields, api, _
from odoo.addons import decimal_precision as dp
from odoo.exceptions import ValidationError
from odoo.tools import (
DEFAULT_SERVER_DATE_FORMAT)
_logger = logging.getLogger(__name__)
class HotelNodeReservationWizard(models.TransientModel):
_name = "hotel.node.reservation.wizard"
_description = "Hotel Node Reservation Wizard"
@api.model
def _get_default_checkin(self):
pass
@api.model
def _get_default_checkout(self):
pass
node_id = fields.Many2one('project.project', 'Hotel', required=True)
partner_id = fields.Many2one('res.partner', string="Customer", required=True)
checkin = fields.Date('Check In', required=True,
default=_get_default_checkin)
checkout = fields.Date('Check Out', required=True,
default=_get_default_checkout)
room_type_wizard_ids = fields.One2many('node.room.type.wizard', 'node_reservation_wizard_id',
string="Room Types")
@api.onchange('node_id')
def _onchange_node_id(self):
self.ensure_one()
if self.node_id:
_logger.info('onchange_node_id(self): %s', 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)
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)
# rooms_availability = noderpc.env['hotel.room.type'].check_availability_room(checkin, checkout) # return str
# TODO add check_availability_room in a hotel slave node module
reservation_ids = noderpc.env['hotel.reservation'].search([
('reservation_line_ids.date', '>=', checkin),
('reservation_line_ids.date', '<', checkout),
('state', '!=', 'cancelled'),
('overbooking', '=', False)
])
reservation_room_ids = []
# do not trust even your father
if reservation_ids:
reservation_room_ids = noderpc.env['hotel.reservation'].browse(reservation_ids).mapped('room_id.id')
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', 'not in', reservation_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)
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"
_description = "Node Room Type Wizard"
def _default_checkin(self):
today = fields.Date.context_today(self.with_context())
return self.node_reservation_wizard_id.checkin or \
fields.Date.from_string(today).strftime(DEFAULT_SERVER_DATE_FORMAT)
def _default_checkout(self):
today = fields.Date.context_today(self.with_context())
return self.node_reservation_wizard_id.checkin or \
(fields.Date.from_string(today) + timedelta(days=1)).strftime(DEFAULT_SERVER_DATE_FORMAT)
node_reservation_wizard_id = fields.Many2one('hotel.node.reservation.wizard')
node_id = fields.Many2one(related='node_reservation_wizard_id.node_id')
room_type_id = fields.Many2one('hotel.node.room.type', 'Rooms Type')
room_type_name = fields.Char('Name', related='room_type_id.name')
room_type_availability = fields.Integer('Availability') #, compute="_compute_room_type_availability")
rooms_qty = fields.Integer('Quantity', default=0)
checkin = fields.Date('Check In', required=True,
default=_default_checkin)
checkout = fields.Date('Check Out', required=True,
default=_default_checkout)
price_unit = fields.Float('Unit Price', required=True,
digits=dp.get_precision('Unit Price'),
default=0.0)
discount = fields.Float(string='Discount (%)',
digits=dp.get_precision('Discount'), default=0.0)
# price_total #compute
# json_days #enchufar como texto literal la cadena devuelta por el método prepare_reservation_lines del hotel.reservation del nodo.(para que funcione
# #es necesario que Darío modifique el método en el modulo Hotel haciendolo independiente del self.
# compute and search fields, in the same order that fields declaration
# @api.depends('node_id')
# def _compute_room_type_availability(self):
# pass
# for record in self:
# record.room_type_availability = 42
#
@api.onchange('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)
# Conectar con nodo para traer dispo(availability) y precio por habitación(price_unit)
# 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')
# def _compute_price_total(self):
# # Unidades x precio unidad (el precio de unidad ya incluye el conjunto de días)
#