Files
pms/hotel/models/hotel_checkin_partner.py
2018-11-07 11:48:59 +01:00

75 lines
2.7 KiB
Python

# Copyright 2017 Dario Lodeiros
# Copyright 2018 Alexandre Diaz
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
import datetime
from odoo import models, fields, api, _
from odoo.exceptions import ValidationError
class HotelCheckinPartner(models.Model):
_name = 'hotel.checkin.partner'
def _default_reservation_id(self):
if 'reservation_id' in self.env.context:
reservation = self.env['hotel.reservation'].browse([
self.env.context['reservation_id']
])
return reservation
return False
def _default_enter_date(self):
if 'reservation_id' in self.env.context:
reservation = self.env['hotel.reservation'].browse([
self.env.context['reservation_id']
])
return reservation.checkin
return False
def _default_exit_date(self):
if 'reservation_id' in self.env.context:
reservation = self.env['hotel.reservation'].browse([
self.env.context['reservation_id']
])
return reservation.checkout
return False
def _default_partner_id(self):
if 'reservation_id' in self.env.context:
reservation = self.env['hotel.reservation'].browse([
self.env.context['reservation_id']
])
return reservation.partner_id
return False
partner_id = fields.Many2one('res.partner', default=_default_partner_id,
required=True)
reservation_id = fields.Many2one(
'hotel.reservation',
default=_default_reservation_id, readonly=True)
enter_date = fields.Date(default=_default_enter_date, required=True)
exit_date = fields.Date(default=_default_exit_date, required=True)
# Validation for Departure date is after arrival date.
@api.multi
@api.constrains('exit_date','enter_date')
def _check_exit_date(self):
for record in self:
date_in = fields.Date.from_string(record.enter_date)
date_out = fields.Date.from_string(record.exit_date)
if date_out < date_in:
raise models.ValidationError(
_('Departure date (%s) is prior to arrival on %s') %
(date_out, date_in))
@api.onchange('enter_date', 'exit_date')
def _onchange_enter_date(self):
date_in = fields.Date.from_string(self.enter_date)
date_out = fields.Date.from_string(self.exit_date)
if date_out <= date_in:
date_out = date_in + datetime.timedelta(days=1)
self.update({'exit_date': date_out})
raise ValidationError(
_('Departure date, is prior to arrival. Check it now. %s') %
date_out)