mirror of
https://bitbucket.org/presik/trytonpsk-hotel.git
synced 2023-12-14 07:52:52 +01:00
1549 lines
53 KiB
Python
1549 lines
53 KiB
Python
# This file is part of Tryton. The COPYRIGHT file at the top level of
|
|
# this repository contains the full copyright notices and license terms.
|
|
|
|
from __future__ import with_statement
|
|
from datetime import datetime, timedelta
|
|
from decimal import Decimal
|
|
from trytond.model import Workflow, ModelView, ModelSQL, fields
|
|
from trytond.wizard import Wizard, StateView, Button, StateTransition, StateReport
|
|
from trytond.report import Report
|
|
from trytond.pyson import Eval, If, In, Get, Not, Or, Equal, Bool
|
|
from trytond.transaction import Transaction
|
|
from trytond.pool import Pool
|
|
|
|
|
|
__all__ = [
|
|
'Booking', 'BookingLine', 'BookingReport', 'SelectRooms', 'SelectRoomsAsk',
|
|
'BookingSale', 'BookingVoucher', 'RoomsOccupancy', 'RoomsOccupancyStart',
|
|
'RoomsOccupancyReport', 'GuestsListReport', 'GuestsListStart',
|
|
'RegistrationCardReport', 'BookingForecastReport', 'GuestsList',
|
|
'Guest', 'BookingDailyStart', 'BookingDaily', 'BookingDailyReport'
|
|
]
|
|
|
|
STATE = [
|
|
(None, ''),
|
|
('offer', 'Offer'),
|
|
('confirmed', 'Confirmed'),
|
|
('not_confirmed', 'Not Confirmed'),
|
|
('cancelled', 'Cancelled'),
|
|
]
|
|
|
|
REGISTRATION_STATE = [
|
|
(None, ''),
|
|
('check_in', 'Check In'),
|
|
('check_out', 'Check Out'),
|
|
('no_show', 'No Show'),
|
|
]
|
|
|
|
STATES = {
|
|
'readonly': Eval('state') != 'offer',
|
|
}
|
|
|
|
STATES_CONFIRMED = {
|
|
'readonly': Eval('state') != 'offer',
|
|
'required': Eval('state') == 'confirmed',
|
|
}
|
|
|
|
STATES_PROCESSING = {
|
|
'readonly': Eval('state').in_(['check_in', 'no_show', 'cancelled']),
|
|
'required': Eval('state') == 'check_in',
|
|
}
|
|
|
|
SEGMENT = [
|
|
('', ''),
|
|
('sun_beach', 'Sun & Beach'),
|
|
('sport', 'Sport'),
|
|
('adventure', 'Adventure'),
|
|
('nature', 'Nature'),
|
|
('cultural', 'Cultural'),
|
|
('urban', 'Urban'),
|
|
('rural', 'Rural'),
|
|
('cruises', 'Cruises'),
|
|
('health', 'Health'),
|
|
('thematic_parks', 'Thematic Parks'),
|
|
('convention', 'Conventions & Meetings'),
|
|
('wedding', 'Weddings'),
|
|
('bussiness', 'Bussiness'),
|
|
]
|
|
|
|
GUARANTEE = [
|
|
('', ''),
|
|
('payment', 'Payment'),
|
|
('voucher', 'Voucher'),
|
|
('credit_card', 'Credit Card'),
|
|
('letter', 'Letter'),
|
|
]
|
|
|
|
SATISFACTION = [
|
|
('', ''),
|
|
('5', 'Excellent'),
|
|
('4', 'Good'),
|
|
('3', 'Regular'),
|
|
('2', 'Bad'),
|
|
('1', 'Worse'),
|
|
]
|
|
|
|
MEDIA = [
|
|
('', ''),
|
|
('phone', 'Phone'),
|
|
('fax', 'Fax'),
|
|
('mail', 'Mail'),
|
|
('chat', 'Chat'),
|
|
('direct', 'Direct'),
|
|
('web', 'Web'),
|
|
('channel_manager', 'Channel Manager'),
|
|
('other', 'Other'),
|
|
('ota', 'OTA'),
|
|
]
|
|
|
|
PLAN = [
|
|
('', ''),
|
|
('all_inclusive', 'All Inclusive'),
|
|
('bed_breakfast', 'Bed & Breakfast'),
|
|
('full_american', 'Full American'),
|
|
('half_american', 'Half American'),
|
|
]
|
|
|
|
SOURCE = [
|
|
('', ''),
|
|
('web', 'Web'),
|
|
('yellow_pages', 'Yellow Pages'),
|
|
('recommended', 'Recommended'),
|
|
('travel_agency', 'Travel Agency'),
|
|
('advertising', 'Advertising'),
|
|
('other', 'Other'),
|
|
]
|
|
|
|
INVOICE_METHOD = [
|
|
('by_booking', 'By Booking'),
|
|
('by_main_guest', 'By Main Guest'),
|
|
# ('guest_customer', 'Guest / Customer')
|
|
]
|
|
|
|
COMPLEMENTARY = [
|
|
('', ''),
|
|
('in_house', 'In House'),
|
|
('courtesy', 'courtesy')
|
|
]
|
|
|
|
_ZERO = Decimal('0.0')
|
|
|
|
|
|
class Booking(Workflow, ModelSQL, ModelView):
|
|
'Booking'
|
|
__name__ = 'hotel.booking'
|
|
_rec_name = 'number'
|
|
number = fields.Char('Number', readonly=True, select=True,
|
|
help="Sequence of reservation.")
|
|
party = fields.Many2One('party.party', 'Party', required=False,
|
|
select=True, help="Person or company owner of the booking.",
|
|
states={
|
|
'required': Eval('state') == 'check_in',
|
|
'readonly': Not(In(Eval('state'), ['offer', 'confirmed'])),
|
|
})
|
|
contact = fields.Char('Contact', states=STATES_PROCESSING)
|
|
payment_term = fields.Many2One('account.invoice.payment_term',
|
|
'Payment Term', states=STATES_PROCESSING)
|
|
booking_date = fields.DateTime('Booking Date', readonly=True,
|
|
states=STATES)
|
|
person_num = fields.Integer('Person Number', states=STATES)
|
|
group = fields.Boolean('Group', states=STATES)
|
|
complementary = fields.Boolean('Complementary', states=STATES)
|
|
type_complementary = fields.Selection(COMPLEMENTARY, 'Type Complementary', states={
|
|
'invisible': ~Bool(Eval('complementary')),
|
|
'required': Bool(Eval('complementary')),
|
|
})
|
|
party_seller = fields.Many2One('hotel.channel', 'Channel',
|
|
states=STATES_PROCESSING, help="Agency or channel that do reservation.")
|
|
state = fields.Selection(STATE, 'State', readonly=True, required=True)
|
|
registration_state = fields.Selection(REGISTRATION_STATE, 'State Registration',
|
|
readonly=True)
|
|
state_string = state.translated('state')
|
|
price_list = fields.Many2One('product.price_list', 'Price List',
|
|
states={
|
|
'readonly': Or(Not(Equal(Eval('state'), 'offer')),
|
|
Bool(Eval('lines'))),
|
|
}, depends=['state', 'company', 'lines'])
|
|
company = fields.Many2One('company.company', 'Company', required=True,
|
|
states=STATES, domain=[('id', If(In('company',
|
|
Eval('context', {})), '=', '!='), Get(Eval('context', {}),
|
|
'company', 0))], readonly=True)
|
|
lines = fields.One2Many('hotel.booking.line', 'booking', 'Lines',
|
|
states={
|
|
'required': Eval('state') == 'confirmed',
|
|
'readonly': Not(In(Eval('state'), ['offer', 'confirmed'])),
|
|
}, depends=['state', 'party'], context={'party': Eval('party')})
|
|
cancellation_policy = fields.Many2One('hotel.policy.cancellation',
|
|
'Cancellation Policy')
|
|
currency = fields.Many2One('currency.currency', 'Currency',
|
|
required=True, states={
|
|
'readonly': (Eval('state') != 'offer') |
|
|
(Eval('lines', [0]) & Eval('currency', 0)),
|
|
}, depends=['state'])
|
|
invoice_method = fields.Selection(INVOICE_METHOD, 'Invoice Method',
|
|
required=False, states=STATES_CONFIRMED)
|
|
satisfaction = fields.Selection(SATISFACTION, 'Satisfaction')
|
|
media = fields.Selection(MEDIA, 'Media', states=STATES_PROCESSING,
|
|
help="Media from booking coming from.")
|
|
media_string = media.translated('media')
|
|
plan = fields.Selection(PLAN, 'Commercial Plan', states=STATES_PROCESSING,
|
|
help="Plans offered by hotel and selected by guest for booking.")
|
|
plan_string = plan.translated('plan')
|
|
source_contact = fields.Selection(SOURCE, 'Source Contact',
|
|
states=STATES_PROCESSING,
|
|
help="Advertising source that create booking opportunity by guest.")
|
|
source_contact_string = source_contact.translated('source_contact')
|
|
comments = fields.Text('Comments', states=STATES_PROCESSING)
|
|
segment = fields.Selection(SEGMENT, 'Tourism Segment',
|
|
states=STATES_PROCESSING)
|
|
segment_string = segment.translated('segment')
|
|
guarantee = fields.Selection(GUARANTEE, 'Guarantee',
|
|
states=STATES_PROCESSING)
|
|
guarantee_string = guarantee.translated('guarantee')
|
|
untaxed_amount = fields.Function(fields.Numeric('Untaxed Amount',
|
|
digits=(16, 2), depends=['lines']), 'get_untaxed_amount')
|
|
tax_amount = fields.Function(fields.Numeric('Tax Amount',
|
|
digits=(16, 2), depends=['lines']), 'get_tax_amount')
|
|
total_amount = fields.Function(fields.Numeric('Total Amount',
|
|
digits=(16, 2), depends=['lines']), 'get_total_amount')
|
|
operations = fields.Function(fields.One2Many('hotel.operation', None,
|
|
'Operations'), 'get_operations', searcher='search_operations')
|
|
code = fields.Char('Code', states={'readonly': True})
|
|
booker = fields.Many2One('party.party', 'Booker', states={'readonly': True})
|
|
created_channel = fields.DateTime('Created Channel', states={'readonly': True})
|
|
sales = fields.Many2Many('hotel.booking-sale.sale', 'booking',
|
|
'sale', 'Sales', readonly=True)
|
|
# operations = fields.Many2Many('hotel.booking-hotel.operation', 'booking',
|
|
# 'sale', 'Sales', readonly=True)
|
|
vouchers = fields.Many2Many('hotel.booking-account.voucher', 'booking',
|
|
'voucher', 'Vouchers', states=STATES_PROCESSING, domain=[
|
|
], depends=['party'])
|
|
vip = fields.Boolean('V.I.P. Customer', states=STATES)
|
|
vehicles_num = fields.Integer('Vehicles Number', states=STATES,
|
|
help="Number of vehicles that bring with guests.")
|
|
vehicle_plate = fields.Integer('Vehicle Plate', states=STATES)
|
|
travel_cause = fields.Char('Travel Cause', states=STATES)
|
|
registration_card = fields.Char('Registration Card', readonly=True,
|
|
select=True, help="Unique sequence of card registration.")
|
|
taxes_exception = fields.Boolean('Taxes Exception', states=STATES)
|
|
|
|
@classmethod
|
|
def __setup__(cls):
|
|
super(Booking, cls).__setup__()
|
|
cls._order.insert(0, ('create_date', 'DESC'))
|
|
cls._transitions |= set((
|
|
('offer', 'confirmed'),
|
|
('confirmed', 'offer'),
|
|
('offer', 'not_confirmed'),
|
|
('offer', 'cancelled'),
|
|
('cancelled', 'offer'),
|
|
('confirmed', 'cancelled'),
|
|
('not_confirmed', 'confirmed'),
|
|
('not_confirmed', 'cancelled'),
|
|
))
|
|
cls._buttons.update({
|
|
'select_rooms': {
|
|
'invisible': Eval('state') != 'offer',
|
|
},
|
|
'cancel': {
|
|
'invisible': Eval('state').in_(['cancel']),
|
|
},
|
|
'offer': {
|
|
'invisible': ~Eval('state').in_([
|
|
'confirmed', 'cancelled', 'not_confirmed'
|
|
]),
|
|
},
|
|
'confirm': {
|
|
'invisible': ~Eval('state').in_(['offer', 'not_confirmed'])
|
|
},
|
|
'not_confirm': {
|
|
'invisible': Eval('state') != 'offer',
|
|
},
|
|
'check_in': {
|
|
'invisible': Eval('registration_state') == 'check_in',
|
|
},
|
|
'no_show': {
|
|
'invisible': Eval('registration_state') == 'no_show',
|
|
},
|
|
'check_out': {
|
|
'invisible': Eval('registration_state') == 'check_out',
|
|
},
|
|
'pay_advance': {
|
|
'invisible': Eval('state').in_(['cancelled', 'check_out']),
|
|
},
|
|
})
|
|
cls._error_messages.update({
|
|
'invalid_number_principal_guest': ('Must exist one principal guest'),
|
|
'payterm_missing': ('The payment term is missing!'),
|
|
'check_time_not_configured': ('The check out time is not configured!'),
|
|
'invalid_arrival_date': ('You cannot check in before the reserved date!'),
|
|
'missing_main_guest': ('Missing main guest in lines!'),
|
|
'missing_select_room': ('Missing select room!'),
|
|
'missing_default_configuration': ('Missing default configuration sequence!'),
|
|
'payterm_missing': ('The payment term is missing!'),
|
|
'room_no_clean': ('The room %s is not clean!'),
|
|
'missing_sequence_registration': ('Missing the configuration of \
|
|
registration card sequence!'),
|
|
})
|
|
|
|
@classmethod
|
|
def trigger_create(cls, records):
|
|
cls.set_number(records)
|
|
|
|
@staticmethod
|
|
def default_currency():
|
|
Company = Pool().get('company.company')
|
|
company = Transaction().context.get('company')
|
|
if company:
|
|
return Company(company).currency.id
|
|
|
|
@staticmethod
|
|
def default_company():
|
|
return Transaction().context.get('company') or False
|
|
|
|
@staticmethod
|
|
def default_payment_term():
|
|
PaymentTerm = Pool().get('account.invoice.payment_term')
|
|
payment_term = PaymentTerm.search([])
|
|
if len(payment_term) >= 1:
|
|
return payment_term[0].id
|
|
|
|
@staticmethod
|
|
def default_state():
|
|
return 'offer'
|
|
|
|
@staticmethod
|
|
def default_invoice_method():
|
|
return 'by_booking'
|
|
|
|
@staticmethod
|
|
def default_booking_date():
|
|
now = datetime.now()
|
|
return now
|
|
|
|
def get_person_num(self, name):
|
|
res = 0
|
|
for line in self.lines:
|
|
if line.num_adult:
|
|
res += line.num_adult
|
|
if line.num_children:
|
|
res += line.num_children
|
|
return res
|
|
|
|
@fields.depends('party', 'price_list')
|
|
def on_change_party(self):
|
|
if self.party and self.party.sale_price_list:
|
|
self.price_list = self.party.sale_price_list.id
|
|
self.price_list.rec_name = self.party.sale_price_list.rec_name
|
|
|
|
@classmethod
|
|
@ModelView.button_action('hotel.wizard_select_rooms')
|
|
def select_rooms(cls, bookings):
|
|
pass
|
|
|
|
@classmethod
|
|
@ModelView.button_action('hotel.wizard_booking_advance_voucher')
|
|
def pay_advance(cls, bookings):
|
|
pass
|
|
|
|
@classmethod
|
|
@ModelView.button
|
|
@Workflow.transition('offer')
|
|
def offer(cls, records):
|
|
for record in records:
|
|
record.delete_occupancy()
|
|
|
|
@classmethod
|
|
@ModelView.button
|
|
@Workflow.transition('cancelled')
|
|
def cancel(cls, records):
|
|
for record in records:
|
|
record.cancel_occupancy()
|
|
|
|
@classmethod
|
|
@ModelView.button
|
|
@Workflow.transition('not_confirmed')
|
|
def not_confirm(cls, records):
|
|
pass
|
|
|
|
@classmethod
|
|
@ModelView.button
|
|
@Workflow.transition('confirmed')
|
|
def confirm(cls, records):
|
|
# Occupancy must be created only when booking is confirmed
|
|
cls.set_number(records)
|
|
for rec in records:
|
|
# FIXME check if does not exist previous occupancy if exist update state
|
|
rec.create_occupancy()
|
|
|
|
@classmethod
|
|
@ModelView.button
|
|
def check_in(cls, records):
|
|
for record in records:
|
|
cls.set_registration_card_number([record])
|
|
for line in record.lines:
|
|
if line.main_guest is None:
|
|
cls.raise_user_error('missing_main_guest')
|
|
if line.room is None:
|
|
cls.raise_user_error('missing_select_room')
|
|
record.check_rooms()
|
|
cls.write([record], {'registration_state': 'check_in'})
|
|
record.update_occupancy(state='open')
|
|
|
|
@classmethod
|
|
@ModelView.button
|
|
@Workflow.transition('no_show')
|
|
def no_show(cls, records):
|
|
for record in records:
|
|
record.update_occupancy(state='cancelled')
|
|
record.cancel_occupancy()
|
|
|
|
@classmethod
|
|
@ModelView.button
|
|
@Workflow.transition('check_out')
|
|
def check_out(cls, records):
|
|
for record in records:
|
|
cls.write([record], {'registration_state': 'check_out'})
|
|
record.update_occupancy(state='closed')
|
|
pass
|
|
|
|
@classmethod
|
|
def set_number(cls, bookings):
|
|
"""
|
|
Fill the number field with the booking sequence
|
|
"""
|
|
pool = Pool()
|
|
Sequence = pool.get('ir.sequence')
|
|
Config = pool.get('hotel.configuration')
|
|
config = Config.get_configuration()
|
|
|
|
for booking in bookings:
|
|
if booking.number or not config.booking_sequence:
|
|
continue
|
|
number = Sequence.get_id(config.booking_sequence.id)
|
|
cls.write([booking], {'number': number})
|
|
|
|
def cancel_occupancy(self):
|
|
Operation = Pool().get('hotel.operation')
|
|
ops = Operation.search([('origin', '=', str(self))])
|
|
if ops:
|
|
Operation.cancel(ops)
|
|
|
|
@classmethod
|
|
def set_registration_card_number(cls, bookings):
|
|
"""
|
|
Fill the number field for registration card with sequence
|
|
"""
|
|
pool = Pool()
|
|
Sequence = pool.get('ir.sequence')
|
|
Config = pool.get('hotel.configuration')
|
|
config = Config.get_configuration()
|
|
|
|
for booking in bookings:
|
|
if booking.registration_card:
|
|
continue
|
|
if not config.registration_card_sequence:
|
|
cls.raise_user_error('missing_sequence_registration')
|
|
number = Sequence.get_id(config.registration_card_sequence.id)
|
|
cls.write([booking], {'registration_card': number})
|
|
|
|
def delete_occupancy(self):
|
|
Operation = Pool().get('hotel.operation')
|
|
for l in self.lines:
|
|
if l.operation:
|
|
Operation.delete([l.operation])
|
|
|
|
def create_occupancy(self):
|
|
pool = Pool()
|
|
Config = pool.get('hotel.configuration')
|
|
Line = pool.get('hotel.booking.line')
|
|
|
|
config = Config.get_configuration()
|
|
if config.check_out_time is None or config.check_in_time is None:
|
|
self.raise_user_error('check_time_not_configured')
|
|
|
|
if self.invoice_method == 'by_booking':
|
|
to_create = [(self.party, l, l.product.id, l.unit_price) for l in self.lines]
|
|
elif self.invoice_method == 'guest_customer':
|
|
to_create = [
|
|
(self.party, l, l.product.id, l.unit_price) for l in self.lines
|
|
]
|
|
to_create.extend([(l.main_guest, l, None, None) for l in self.lines])
|
|
else:
|
|
to_create = [(l.main_guest, l, l.product.id, l.unit_price) for l in self.lines]
|
|
|
|
for value in to_create:
|
|
Line.create_operation(value[0], value[1], value[2], value[3])
|
|
|
|
def check_rooms(self):
|
|
pool = Pool()
|
|
rooms_ids = []
|
|
for line in self.lines:
|
|
rooms_ids.append(line.room.id)
|
|
Housekeeping = pool.get('hotel.housekeeping')
|
|
housekeepings = Housekeeping.search([
|
|
('state', '!=', 'clean')
|
|
])
|
|
for hk in housekeepings:
|
|
if hk.room.id in rooms_ids:
|
|
self.raise_user_error('room_no_clean', hk.room.name)
|
|
|
|
def get_context_price(self, product):
|
|
context = {}
|
|
if getattr(self, 'currency', None):
|
|
context['currency'] = self.currency.id
|
|
if getattr(self, 'party', None):
|
|
context['customer'] = self.party.id
|
|
if getattr(self, 'booking_date', None):
|
|
context['sale_date'] = self.booking_date
|
|
if getattr(self, 'price_list', None):
|
|
context['price_list'] = self.price_list.id
|
|
|
|
context['uom'] = product.template.default_uom.id
|
|
|
|
# Set taxes before unit_price to have taxes in context of sale price
|
|
taxes = []
|
|
pattern = {}
|
|
if not self.taxes_exception:
|
|
for tax in product.customer_taxes_used:
|
|
if self.party and self.party.customer_tax_rule:
|
|
tax_ids = self.party.customer_tax_rule.apply(tax, pattern)
|
|
if tax_ids:
|
|
taxes.extend(tax_ids)
|
|
continue
|
|
taxes.append(tax.id)
|
|
if self.party and self.party.customer_tax_rule:
|
|
tax_ids = self.party.customer_tax_rule.apply(None, pattern)
|
|
if tax_ids:
|
|
taxes.extend(tax_ids)
|
|
|
|
context['taxes'] = taxes
|
|
return context
|
|
|
|
def create_sale(self):
|
|
pool = Pool()
|
|
Model = pool.get('ir.model')
|
|
SaleLine = pool.get('sale.line')
|
|
Configuration = pool.get('hotel.configuration')
|
|
configuration = Configuration.get_configuration()
|
|
date_ = datetime.today().date()
|
|
model = Model.search([('name', '=', str(self.__name__))])
|
|
ctx = {}
|
|
|
|
if not self.payment_term:
|
|
self.raise_user_error('payterm_missing')
|
|
|
|
if self.price_list:
|
|
ctx['price_list'] = self.price_list
|
|
ctx['sale_date'] = date_
|
|
ctx['currency'] = self.currency.id
|
|
if self.party:
|
|
ctx['customer'] = self.party.id
|
|
|
|
if not model:
|
|
model = 'Booking'
|
|
else:
|
|
model = model[0].name
|
|
|
|
if self.invoice_method == 'by_booking':
|
|
bookings_ = [(self.party, self.lines)]
|
|
else:
|
|
bookings_ = [
|
|
(line.main_guest, [line]) for line in self.lines
|
|
]
|
|
|
|
for party, lines in bookings_:
|
|
sale = self.get_sale(party, date_)
|
|
sale.save()
|
|
for line in lines:
|
|
if line.nights_quantity <= 0:
|
|
continue
|
|
|
|
if self.price_list:
|
|
with Transaction().set_context(ctx):
|
|
unit_price = self.price_list.compute(
|
|
self.party, line.product,
|
|
line.product.template.list_price,
|
|
line.nights_quantity, line.product.default_uom)
|
|
else:
|
|
unit_price = line.unit_price
|
|
|
|
unit_price = self.currency.round(line.unit_price)
|
|
new_line = self.create_sale_line(sale, line, unit_price)
|
|
SaleLine.create([new_line])
|
|
|
|
# Add and create default charges lines if exists
|
|
for product in configuration.default_charges:
|
|
taxes_ids = self.get_taxes(product)
|
|
new_line = {
|
|
'sale': sale.id,
|
|
'type': 'line',
|
|
'unit': product.template.default_uom.id,
|
|
'quantity': 1,
|
|
'unit_price': product.template.list_price,
|
|
'product': product.id,
|
|
'description': product.rec_name,
|
|
}
|
|
if taxes_ids:
|
|
new_line.update({'taxes': [('add', taxes_ids)]})
|
|
if new_line:
|
|
SaleLine.create([new_line])
|
|
self.write([self], {'sales': [('add', [sale.id])]})
|
|
|
|
def update_occupancy(self, state):
|
|
Operation = Pool().get('hotel.operation')
|
|
OperationVoucher = Pool().get('hotel.operation-account.voucher')
|
|
vouchers_added = False
|
|
for line in self.lines:
|
|
if state != 'open':
|
|
continue
|
|
guests = [{
|
|
'party': line.main_guest.id
|
|
}]
|
|
|
|
for g in line.guests:
|
|
# if line.main_guest.id == g.party.id:
|
|
# continue
|
|
guests.append({
|
|
'party': g.party.id
|
|
})
|
|
|
|
|
|
if line.operation:
|
|
Operation.write([line.operation], {
|
|
'state': state,
|
|
'main_guest': line.main_guest.id,
|
|
'party': self.party.id,
|
|
'unit_price': line.unit_price,
|
|
'start_date': line.arrival_date,
|
|
'end_date': line.departure_date,
|
|
'reference': self.registration_card,
|
|
'guests': [('create', guests)]
|
|
})
|
|
|
|
vouchers = None
|
|
if not vouchers_added and self.vouchers:
|
|
vouchers = self.vouchers
|
|
vouchers_added = True
|
|
if vouchers:
|
|
for v in vouchers:
|
|
OperationVoucher.create([{
|
|
'voucher': v.id,
|
|
'operation': line.operation.id
|
|
}])
|
|
|
|
def get_total_amount(self, name):
|
|
res = 0
|
|
if self.tax_amount or self.untaxed_amount:
|
|
res = self.tax_amount + self.untaxed_amount
|
|
return res
|
|
|
|
def get_taxes(self, product):
|
|
ctx = self.get_context_price(product)
|
|
return ctx['taxes']
|
|
|
|
def get_tax_amount(self, name):
|
|
Tax = Pool().get('account.tax')
|
|
|
|
res = _ZERO
|
|
for line in self.lines:
|
|
taxes_ids = self.get_taxes(line.product)
|
|
if taxes_ids:
|
|
taxes = Tax.browse(taxes_ids)
|
|
tax_list = Tax.compute(
|
|
taxes, line.unit_price or _ZERO, line.nights_quantity or 0
|
|
)
|
|
|
|
tax_amount = sum([t['amount'] for t in tax_list], _ZERO)
|
|
res += tax_amount
|
|
|
|
res = Decimal(round(res, 2))
|
|
return res
|
|
|
|
def get_untaxed_amount(self, name):
|
|
res = _ZERO
|
|
for line in self.lines:
|
|
res += line.total_amount
|
|
return res
|
|
|
|
def get_occupancies(self, name):
|
|
occupancies = set()
|
|
for line in self.lines:
|
|
if line.occupancy_line:
|
|
occupancies.add(line.occupancy_line.id)
|
|
return list(occupancies)
|
|
|
|
|
|
class BookingLine(ModelSQL, ModelView):
|
|
'Booking Line'
|
|
__name__ = 'hotel.booking.line'
|
|
booking = fields.Many2One('hotel.booking', 'Booking',
|
|
ondelete='CASCADE', select=True)
|
|
room = fields.Many2One('hotel.room', 'Room', select=True, states={
|
|
'required': Eval('state') == 'confirmed',
|
|
})
|
|
arrival_date = fields.Date('Arrival Date', required=True)
|
|
departure_date = fields.Date('Departure Date', required=True)
|
|
product = fields.Many2One('product.product', 'Product',
|
|
select=True, domain=[
|
|
('template.type', '=', 'service'),
|
|
('template.kind', '=', 'accommodation'),
|
|
], required=True)
|
|
unit_price = fields.Numeric('Unit Price', digits=(16, 4),
|
|
states={
|
|
'required': Bool(Eval('product')),
|
|
})
|
|
uom = fields.Many2One('product.uom', 'UOM', readonly=True)
|
|
main_guest = fields.Many2One('party.party', 'Main Guest', select=True,
|
|
states={
|
|
'required': Eval('state').in_(['check_in', 'check_out'])
|
|
}
|
|
)
|
|
state = fields.Selection(STATE, 'State')
|
|
contact = fields.Char('Contact')
|
|
num_children = fields.Function(fields.Integer('No. Children'),
|
|
'get_num_children')
|
|
nights_quantity = fields.Function(fields.Integer('Nights'),
|
|
'get_nights_quantity')
|
|
host_quantity = fields.Integer('Host')
|
|
unit_digits = fields.Function(fields.Integer('Unit Digits'), 'get_unit_digits')
|
|
notes = fields.Text('Notes')
|
|
total_amount = fields.Function(fields.Numeric('Total Amount',
|
|
digits=(16, 2)), 'get_total_amount')
|
|
total_commission = fields.Function(fields.Numeric('Channel Commission',
|
|
digits=(16, 2)), 'get_channel_commission')
|
|
guests = fields.One2Many('hotel.booking.guest', 'booking_line', 'Guests')
|
|
operation = fields.Many2One('hotel.operation', 'Operation')
|
|
nationality = fields.Many2One('party.nationality', 'Nationality')
|
|
origin_city = fields.Many2One('hotel.migration_city', 'Origin City',
|
|
select=True)
|
|
target_city = fields.Many2One('hotel.migration_city', 'Target City',
|
|
select=True)
|
|
|
|
@classmethod
|
|
def __setup__(cls):
|
|
super(BookingLine, cls).__setup__()
|
|
cls._error_messages.update({
|
|
'invalid_number_guests': ('Invalid number of guests for '
|
|
'the room %s.'),
|
|
'accommodation_not_defined': ('The room %s has not accommodation'),
|
|
'invalid_date': ('Departure date must be greater \
|
|
that arrival date'),
|
|
'occupied_room': ('The room is occupied in the date %s'),
|
|
'restring_room': ('The room %s is in restring access for quarantine protocolo!'),
|
|
})
|
|
|
|
|
|
@classmethod
|
|
def write(cls, *args):
|
|
Operation = Pool().get('hotel.operation')
|
|
lines, vals = args[0], args[-1]
|
|
for line in lines:
|
|
if not line.operation:
|
|
continue
|
|
values = {}
|
|
if vals.get('room'):
|
|
values.update({
|
|
'room': vals['room'],
|
|
})
|
|
if vals.get('main_guest'):
|
|
values.update({
|
|
'party': vals['main_guest'],
|
|
})
|
|
if vals.get('product'):
|
|
values.update({
|
|
'accommodation': vals['product'],
|
|
})
|
|
if vals.get('state'):
|
|
values.update({'state': vals['state']})
|
|
if vals.get('arrival_date'):
|
|
values.update({'start_date': vals['arrival_date']})
|
|
if vals.get('departure_date'):
|
|
values.update({'end_date': vals['departure_date']})
|
|
|
|
if values:
|
|
Operation.write([line.operation], values)
|
|
super(BookingLine, cls).write(*args)
|
|
|
|
@staticmethod
|
|
def default_main_guest():
|
|
party = Transaction().context.get('party')
|
|
return party
|
|
|
|
@staticmethod
|
|
def default_host_quantity():
|
|
return 1
|
|
|
|
@staticmethod
|
|
def default_state():
|
|
return 'offer'
|
|
|
|
@staticmethod
|
|
def default_accommodation():
|
|
Configuration = Pool().get('hotel.configuration')
|
|
configuration = Configuration.get_configuration()
|
|
|
|
if configuration.default_accommodation:
|
|
return configuration.default_accommodation.id
|
|
|
|
@classmethod
|
|
def validate(cls, lines):
|
|
super(BookingLine, cls).validate(lines)
|
|
for line in lines:
|
|
line.check_method()
|
|
|
|
@classmethod
|
|
def create_operation(cls, party, line, accommodation, unit_price):
|
|
Operation = Pool().get('hotel.operation')
|
|
|
|
kind = 'occupancy'
|
|
# if not accommodation:
|
|
# kind = 'statement'
|
|
|
|
values = {
|
|
'kind': kind,
|
|
'accommodation': accommodation,
|
|
'start_date': line.arrival_date,
|
|
'end_date': line.departure_date,
|
|
'currency': line.booking.currency.id,
|
|
'state': 'draft',
|
|
'unit_price': unit_price,
|
|
'party': party.id if party else None,
|
|
'origin': str(line),
|
|
'complementary': line.booking.complementary,
|
|
'type_complementary': line.booking.type_complementary,
|
|
'taxes_exception': line.booking.taxes_exception,
|
|
}
|
|
if line.room:
|
|
values['room'] = line.room.id
|
|
|
|
if line.main_guest:
|
|
values['main_guest'] = line.main_guest.id
|
|
|
|
operation, = Operation.create([values])
|
|
|
|
line.write([line], {'operation': operation.id})
|
|
|
|
@fields.depends('accommodation', 'product')
|
|
def on_change_accommodation(self):
|
|
if not self.accommodation:
|
|
self.product = None
|
|
|
|
@fields.depends('product', 'unit_price', 'uom', 'booking', 'nights_quantity')
|
|
def on_change_product(self):
|
|
Product = Pool().get('product.product')
|
|
if self.product and self.booking:
|
|
self.uom = self.product.default_uom.id
|
|
with Transaction().set_context(self.booking.get_context_price(self.product)):
|
|
self.unit_price = Product.get_sale_price([self.product],
|
|
self.nights_quantity or 0)[self.product.id]
|
|
if self.unit_price:
|
|
self.unit_price = self.booking.currency.round(self.unit_price)
|
|
else:
|
|
self.unit_price = self.product.list_price
|
|
|
|
|
|
@fields.depends('arrival_date', 'departure_date')
|
|
def on_change_arrival_date(self):
|
|
if not self.arrival_date or (self.departure_date and self.departure_date > self.arrival_date):
|
|
return
|
|
self.departure_date = self.arrival_date + timedelta(days=1)
|
|
|
|
def check_method(self):
|
|
"""
|
|
Check the methods.
|
|
"""
|
|
Date = Pool().get('ir.date')
|
|
if self.arrival_date < Date.today():
|
|
self.raise_user_error('invalid_arrival_date')
|
|
if self.arrival_date >= self.departure_date:
|
|
self.raise_user_error('invalid_date')
|
|
Operation = Pool().get('hotel.operation')
|
|
# operations = Operation.search([
|
|
# ('room', '=', self.room.id),
|
|
# ['AND',
|
|
# ['OR', [
|
|
# ('start_date', '>=', self.arrival_date),
|
|
# ('end_date', '<=', self.arrival_date),
|
|
# ], [
|
|
# ('start_date', '>=', self.departure_date),
|
|
# ('end_date', '<=', self.departure_date),
|
|
# ],
|
|
# ]]
|
|
# ])
|
|
# print(operations)
|
|
# if operations:
|
|
# self.raise_user_error('occupied_room', self.departure_date)
|
|
config = Pool().get('hotel.configuration')(1)
|
|
quarantine_days = config.quarantine_rooms
|
|
room_id = self.room.id
|
|
if quarantine_days:
|
|
delta = timedelta(days=int(quarantine_days))
|
|
_date = self.arrival_date - delta
|
|
operations = Operation.search([
|
|
['AND',
|
|
['OR', [
|
|
('room', '=', room_id),
|
|
('end_date', '=', _date)
|
|
],
|
|
[
|
|
('room', '=', room_id),
|
|
('end_date', '=', self.arrival_date)
|
|
]]]])
|
|
print(operations)
|
|
if operations:
|
|
self.raise_user_error('restring_room', self.room.name)
|
|
|
|
|
|
def get_state(self, name):
|
|
if self.booking:
|
|
return self.booking.state
|
|
|
|
def get_host_quantity(self, name):
|
|
res = 0
|
|
if self.num_adult:
|
|
res += self.num_adult
|
|
if self.num_children:
|
|
res += self.num_children
|
|
return res
|
|
|
|
@fields.depends('quantity', 'arrival_date', 'departure_date')
|
|
def get_nights_quantity(self, name=None):
|
|
"""
|
|
Compute nights between start and end
|
|
return a integer the mean days of occupancy.
|
|
"""
|
|
nights = 0
|
|
if not self.arrival_date or not self.departure_date:
|
|
return nights
|
|
nights = (self.departure_date - self.arrival_date).days
|
|
return nights
|
|
|
|
def get_total_amount(self, name):
|
|
"""
|
|
The total amount of booking based on room flat price.
|
|
TODO: If room fee is applied should be used for total price calculation
|
|
instead of flat price. Fee is linked to channel management.
|
|
"""
|
|
res = _ZERO
|
|
if self.nights_quantity and self.unit_price:
|
|
res = self.nights_quantity * self.unit_price
|
|
return res
|
|
|
|
def get_channel_commission(self, name):
|
|
"""
|
|
Calculation of sale commission for channel based on booking total amount
|
|
"""
|
|
res = Decimal(0)
|
|
if self.total_amount and self.booking.party_seller and \
|
|
self.booking.party_seller.party.sale_commission:
|
|
res = self.total_amount * self.booking.party_seller.sale_commission / 100
|
|
return res
|
|
|
|
|
|
class BookingReport(Report):
|
|
__name__ = 'hotel.booking'
|
|
|
|
@classmethod
|
|
def get_context(cls, records, data):
|
|
report_context = super(BookingReport, cls).get_context(records, data)
|
|
user = Pool().get('res.user')(Transaction().user)
|
|
report_context['company'] = user.company
|
|
return report_context
|
|
|
|
|
|
class SelectRoomsAsk(ModelView):
|
|
'Filter Rooms Assistant'
|
|
__name__ = 'hotel.booking.select_rooms.ask'
|
|
arrival_date = fields.Date('Arrival Date', required=True)
|
|
departure_date = fields.Date('Departure Date', required=True)
|
|
accommodation = fields.Many2One('product.product',
|
|
'Accommodation', domain=[
|
|
('template.kind', '=', 'accommodation'),
|
|
])
|
|
rooms = fields.Many2Many('hotel.room', None, None,
|
|
'Rooms', domain=[
|
|
('id', 'in', Eval('targets')),
|
|
])
|
|
overbooking = fields.Boolean('Overbooking')
|
|
targets = fields.Function(fields.Many2Many('hotel.room', None, None,
|
|
'Targets'), 'on_change_with_targets')
|
|
|
|
@staticmethod
|
|
def default_accommodation():
|
|
Configuration = Pool().get('hotel.configuration')
|
|
configuration = Configuration.get_configuration()
|
|
|
|
if configuration.default_accommodation:
|
|
return configuration.default_accommodation.id
|
|
|
|
@fields.depends('arrival_date', 'departure_date', 'accommodation', 'overbooking')
|
|
def on_change_with_targets(self, name=None):
|
|
pool = Pool()
|
|
Operation = pool.get('hotel.operation')
|
|
RoomTemplate = pool.get('hotel.room-product.template')
|
|
Room = pool.get('hotel.room')
|
|
res = []
|
|
if not self.accommodation or not self.arrival_date or not self.departure_date:
|
|
return res
|
|
|
|
if self.overbooking:
|
|
return [r.id for r in Room.search([])]
|
|
|
|
room_templates = RoomTemplate.search([
|
|
('template.accommodation_capacity', '>=', self.accommodation.accommodation_capacity)
|
|
])
|
|
rooms_ids = [t.room.id for t in room_templates]
|
|
|
|
rooms_available_ids = Operation.get_available_rooms(
|
|
self.arrival_date,
|
|
self.departure_date,
|
|
rooms_ids=rooms_ids
|
|
)
|
|
|
|
return rooms_available_ids
|
|
|
|
|
|
class SelectRooms(Wizard):
|
|
'Select Rooms'
|
|
__name__ = 'hotel.booking.select_rooms'
|
|
"""
|
|
this is the wizard that allows the front desk employee to select
|
|
rooms, based on the requirements listed by the customer.
|
|
"""
|
|
start = StateView('hotel.booking.select_rooms.ask',
|
|
'hotel.view_select_rooms_form', [
|
|
Button('Exit', 'end', 'tryton-cancel'),
|
|
Button('Add and Continue', 'add_continue', 'tryton-forward'),
|
|
Button('Add', 'add_rooms', 'tryton-ok'),
|
|
]
|
|
)
|
|
add_rooms = StateTransition()
|
|
add_continue = StateTransition()
|
|
|
|
def transition_add_rooms(self):
|
|
self._add_rooms()
|
|
return 'end'
|
|
|
|
def transition_add_continue(self):
|
|
self._add_rooms()
|
|
return 'start'
|
|
|
|
def _add_rooms(self):
|
|
pool = Pool()
|
|
Line = pool.get('hotel.booking.line')
|
|
Booking = pool.get('hotel.booking')
|
|
booking = Booking(Transaction().context.get('active_id'))
|
|
lines_to_create = []
|
|
ctx = {}
|
|
if booking.price_list:
|
|
ctx['price_list'] = booking.price_list
|
|
ctx['sale_date'] = self.start.arrival_date
|
|
ctx['currency'] = booking.currency.id
|
|
if booking.party:
|
|
ctx['customer'] = booking.party.id
|
|
product = self.start.accommodation
|
|
|
|
quantity = (self.start.departure_date - self.start.arrival_date).days
|
|
unit_price = product.template.list_price
|
|
if booking.price_list:
|
|
with Transaction().set_context(ctx):
|
|
unit_price = booking.price_list.compute(
|
|
booking.party, product, unit_price,
|
|
quantity, product.default_uom)
|
|
unit_price = booking.currency.round(unit_price)
|
|
|
|
for room in self.start.rooms:
|
|
values = {
|
|
'booking': booking.id,
|
|
'product': product.id,
|
|
'room': room.id,
|
|
'arrival_date': self.start.arrival_date,
|
|
'departure_date': self.start.departure_date,
|
|
'unit_price': unit_price,
|
|
}
|
|
if booking.party:
|
|
values['main_guest'] = booking.party.id
|
|
values.update({'product': product.id})
|
|
lines_to_create.append(values)
|
|
Line.create(lines_to_create)
|
|
booking.save()
|
|
|
|
|
|
class BookingSale(ModelSQL):
|
|
'Booking - Sale'
|
|
__name__ = 'hotel.booking-sale.sale'
|
|
_table = 'booking_sales_rel'
|
|
booking = fields.Many2One('hotel.booking', 'Booking', ondelete='CASCADE',
|
|
required=True)
|
|
sale = fields.Many2One('sale.sale', 'Sale', ondelete='RESTRICT',
|
|
required=True)
|
|
|
|
|
|
class BookingVoucher(ModelSQL):
|
|
'Booking - Voucher'
|
|
__name__ = 'hotel.booking-account.voucher'
|
|
_table = 'booking_vouchers_rel'
|
|
booking = fields.Many2One('hotel.booking', 'Booking',
|
|
ondelete='CASCADE', select=True, required=True)
|
|
voucher = fields.Many2One('account.voucher', 'Voucher', select=True,
|
|
domain=[('voucher_type', '=', 'receipt')], ondelete='RESTRICT',
|
|
required=True)
|
|
|
|
@classmethod
|
|
def set_voucher_origin(cls, voucher_id, booking_id):
|
|
cls.create([{
|
|
'voucher': voucher_id,
|
|
'booking': booking_id,
|
|
}])
|
|
|
|
|
|
class Guest(ModelSQL, ModelView):
|
|
'Guest'
|
|
__name__ = 'hotel.booking.guest'
|
|
_rec_name = 'party'
|
|
booking_line = fields.Many2One('hotel.booking.line', 'Booking Line',
|
|
required=True, ondelete='CASCADE')
|
|
party = fields.Many2One('party.party', 'Party', required=False)
|
|
type_guest = fields.Selection([
|
|
('adult', 'Adult'),
|
|
('child', 'Child'),
|
|
], 'Type Guest')
|
|
type_guest_string = type_guest.translated('type_guest')
|
|
nationality = fields.Many2One('party.nationality', 'Nationality')
|
|
origin_city = fields.Many2One('hotel.migration_city', 'Origin City',
|
|
select=True)
|
|
target_city = fields.Many2One('hotel.migration_city', 'Target City',
|
|
select=True)
|
|
|
|
# New fields for speed reason
|
|
type_document = fields.Selection([
|
|
('12', 'Tarjeta de Identidad'),
|
|
('13', 'Cedula de Ciudadania'),
|
|
('21', 'Tarjeta de Extranjeria'),
|
|
('22', 'Cedula de Extranjeria'),
|
|
('41', 'Pasaporte'),
|
|
], 'Document Type')
|
|
doc_number = fields.Char('Doc. Id', select=True)
|
|
name = fields.Char('Name', select=True)
|
|
mobile = fields.Char('Mobile', select=True)
|
|
email = fields.Char('Email', select=True)
|
|
birthday = fields.Date('Birthday', select=True)
|
|
sex = fields.Selection([
|
|
('female', 'Female'),
|
|
('male', 'Male'),
|
|
('', ''),
|
|
], 'Sex')
|
|
|
|
def get_rec_name(self, name):
|
|
if self.party:
|
|
return self.party.name
|
|
|
|
@staticmethod
|
|
def default_type_guest():
|
|
return 'adult'
|
|
|
|
@staticmethod
|
|
def default_sex():
|
|
return 'male'
|
|
|
|
@staticmethod
|
|
def default_type_document():
|
|
return '13'
|
|
|
|
@staticmethod
|
|
def default_principal_guest():
|
|
return False
|
|
|
|
@fields.depends('origin_city', 'target_city')
|
|
def on_change_origin_city(self):
|
|
if self.origin_city:
|
|
self.target_city = self.origin_city.id
|
|
|
|
@classmethod
|
|
def create(cls, vlist):
|
|
Party = Pool().get('party.party')
|
|
new_values = []
|
|
for v in vlist:
|
|
# if not v.get('doc_number'):
|
|
# continue
|
|
party = v.get('party')
|
|
if not party:
|
|
parties = Party.search([
|
|
('id_number', '=', v['doc_number']),
|
|
])
|
|
if parties:
|
|
party = parties[0]
|
|
else:
|
|
party, = Party.create([{
|
|
'name': v['name'],
|
|
'id_number': v['doc_number'],
|
|
'type_document': v['type_document'],
|
|
'birthday': v['birthday'],
|
|
'sex': v['sex'],
|
|
'contact_mechanisms': [
|
|
('create', [
|
|
{'type': 'email', 'value': v['email']},
|
|
{'type': 'mobile', 'value': v['mobile']},
|
|
])
|
|
]
|
|
}])
|
|
v['party'] = party.id
|
|
new_values.append(v)
|
|
|
|
super(Guest, cls).create(new_values)
|
|
|
|
|
|
class BookingForecastStart(ModelView):
|
|
'Booking Forecast Start'
|
|
__name__ = 'hotel.print_booking_forecast.start'
|
|
date = fields.Date('Start Date', required=True)
|
|
company = fields.Many2One('company.company', 'Company', required=True)
|
|
|
|
@staticmethod
|
|
def default_date():
|
|
Date_ = Pool().get('ir.date')
|
|
return Date_.today()
|
|
|
|
@staticmethod
|
|
def default_company():
|
|
return Transaction().context.get('company')
|
|
|
|
|
|
class BookingForecast(Wizard):
|
|
'Booking Forecast'
|
|
__name__ = 'hotel.print_booking_forecast'
|
|
start = StateView('hotel.print_booking_forecast.start',
|
|
'hotel.print_booking_forecast_start_view_form', [
|
|
Button('Cancel', 'end', 'tryton-cancel'),
|
|
Button('Print', 'print_', 'tryton-print', default=True),
|
|
])
|
|
print_ = StateReport('hotel.booking_forecast.report')
|
|
|
|
def do_print_(self, action):
|
|
company = self.start.company
|
|
data = {
|
|
'date': self.start.date,
|
|
'company': company.id,
|
|
}
|
|
return action, data
|
|
|
|
def transition_print_(self):
|
|
return 'end'
|
|
|
|
|
|
class BookingForecastReport(Report):
|
|
__name__ = 'hotel.booking_forecast.report'
|
|
|
|
@classmethod
|
|
def get_context(cls, records, data):
|
|
report_context = super(BookingForecastReport, cls).get_context(records, data)
|
|
MAX_DAYS = 30
|
|
pool = Pool()
|
|
Company = pool.get('company.company')
|
|
Room = pool.get('hotel.room')
|
|
BookingLine = pool.get('hotel.booking.line')
|
|
|
|
rooms = Room.search([])
|
|
alldays = {}
|
|
alldays_convert = {}
|
|
for nd in range(MAX_DAYS):
|
|
day_n = 'day' + str((nd + 1))
|
|
tdate = data['date'] + timedelta(nd)
|
|
data[day_n] = tdate
|
|
data['total_' + day_n] = 0
|
|
data[('revenue_' + day_n)] = 0
|
|
data[('rate_' + day_n)] = 0
|
|
alldays[day_n] = ''
|
|
alldays_convert[tdate] = day_n
|
|
|
|
date_init = data['date']
|
|
date_limit = data['date'] + timedelta(MAX_DAYS)
|
|
|
|
dom = [['OR',
|
|
[
|
|
('arrival_date', '<=', date_init),
|
|
('departure_date', '>=', date_init),
|
|
], [
|
|
('arrival_date', '>=', date_init),
|
|
('arrival_date', '<=', date_limit),
|
|
],
|
|
|
|
], ('booking.state', 'not in', ['no_show', 'cancelled'])]
|
|
|
|
lines = BookingLine.search(dom)
|
|
drooms = {}
|
|
|
|
for room in rooms:
|
|
drooms[room.id] = {'name': room.name}
|
|
drooms[room.id].update(alldays.copy())
|
|
|
|
for line in lines:
|
|
_delta = (line.departure_date - line.arrival_date).days
|
|
for i in range(_delta):
|
|
dt = line.arrival_date + timedelta(i)
|
|
if dt >= date_init and dt < date_limit \
|
|
and dt >= data['date']:
|
|
dayn = alldays_convert[dt]
|
|
drooms[line.room.id][dayn] = "X"
|
|
data['total_' + dayn] += 1
|
|
data['revenue_' + dayn] += float(line.unit_price) / 1000000
|
|
for i in range(MAX_DAYS):
|
|
day_n = 'day' + str((i + 1))
|
|
data['rate_' + day_n] = (data['total_' + day_n] * 100.0) / len(rooms)
|
|
|
|
report_context['records'] = list(drooms.values())
|
|
report_context['date'] = data['date']
|
|
report_context['company'] = Company(data['company']).party.name
|
|
return report_context
|
|
|
|
|
|
class RegistrationCardReport(Report):
|
|
__name__ = 'hotel.occupancy.registration_card'
|
|
|
|
@classmethod
|
|
def get_context(cls, records, data):
|
|
report_context = super(RegistrationCardReport, cls).get_context(records, data)
|
|
user = Pool().get('res.user')(Transaction().user)
|
|
report_context['company'] = user.company
|
|
return report_context
|
|
|
|
|
|
class GuestsListStart(ModelView):
|
|
'Guests List Start'
|
|
__name__ = 'hotel.print_guests_list.start'
|
|
date = fields.Date('Date', required=True)
|
|
company = fields.Many2One('company.company', 'Company', required=True)
|
|
|
|
@staticmethod
|
|
def default_date():
|
|
Date_ = Pool().get('ir.date')
|
|
return Date_.today()
|
|
|
|
@staticmethod
|
|
def default_company():
|
|
return Transaction().context.get('company')
|
|
|
|
|
|
class GuestsList(Wizard):
|
|
'Guests List'
|
|
__name__ = 'hotel.print_guests_list'
|
|
start = StateView('hotel.print_guests_list.start',
|
|
'hotel.print_guests_list_start_view_form', [
|
|
Button('Cancel', 'end', 'tryton-cancel'),
|
|
Button('Open', 'print_', 'tryton-print', default=True),
|
|
])
|
|
print_ = StateReport('hotel.guests_list.report')
|
|
|
|
def do_print_(self, action):
|
|
company = self.start.company
|
|
data = {
|
|
'date': self.start.date,
|
|
'company': company.id,
|
|
}
|
|
return action, data
|
|
|
|
def transition_print_(self):
|
|
return 'end'
|
|
|
|
|
|
class GuestsListReport(Report):
|
|
__name__ = 'hotel.guests_list.report'
|
|
|
|
@classmethod
|
|
def get_context(cls, records, data):
|
|
report_context = super(GuestsListReport, cls).get_context(records, data)
|
|
pool = Pool()
|
|
Company = pool.get('company.company')
|
|
Operation = pool.get('hotel.operation')
|
|
Room = pool.get('hotel.room')
|
|
User = pool.get('res.user')
|
|
user_id = Transaction().user
|
|
|
|
user = User(user_id)
|
|
start_date = data['date']
|
|
operations = Operation.search([
|
|
('start_date', '<=', start_date),
|
|
('kind', '=', 'occupancy'),
|
|
('state', '=', 'check_in'),
|
|
], order=[('room.code', 'ASC')])
|
|
|
|
total_guests = []
|
|
rooms_occupied = []
|
|
for op in operations:
|
|
total_guests.append(len(op.guests))
|
|
rooms_occupied.append(op.room.id)
|
|
|
|
rooms = Room.search_read([])
|
|
|
|
_rooms_occupied = len(set(rooms_occupied))
|
|
occupancy_rate = round(_rooms_occupied / len(rooms), 2)
|
|
|
|
report_context['records'] = operations
|
|
report_context['total_rooms'] = _rooms_occupied
|
|
report_context['total_guests'] = sum(total_guests)
|
|
report_context['occupancy_rate'] = occupancy_rate
|
|
report_context['company'] = Company(data['company']).party.name
|
|
report_context['date'] = data['date']
|
|
report_context['print_date'] = datetime.now()
|
|
report_context['user'] = user.name
|
|
return report_context
|
|
|
|
|
|
class RoomsOccupancyStart(ModelView):
|
|
'Rooms Occupancy Start'
|
|
__name__ = 'hotel.print_rooms_occupancy.start'
|
|
date = fields.Date('Date', required=True)
|
|
company = fields.Many2One('company.company', 'Company', required=True)
|
|
|
|
@staticmethod
|
|
def default_date():
|
|
Date_ = Pool().get('ir.date')
|
|
return Date_.today()
|
|
|
|
@staticmethod
|
|
def default_company():
|
|
return Transaction().context.get('company')
|
|
|
|
|
|
class RoomsOccupancy(Wizard):
|
|
'Rooms Occupancy'
|
|
__name__ = 'hotel.print_rooms_occupancy'
|
|
start = StateView('hotel.print_rooms_occupancy.start',
|
|
'hotel.print_rooms_occupancy_start_view_form', [
|
|
Button('Cancel', 'end', 'tryton-cancel'),
|
|
Button('Open', 'print_', 'tryton-print', default=True),
|
|
])
|
|
print_ = StateReport('hotel.rooms_occupancy.report')
|
|
|
|
def do_print_(self, action):
|
|
company = self.start.company
|
|
data = {
|
|
'date': self.start.date,
|
|
'company': company.id,
|
|
}
|
|
return action, data
|
|
|
|
def transition_print_(self):
|
|
return 'end'
|
|
|
|
|
|
class RoomsOccupancyReport(Report):
|
|
__name__ = 'hotel.rooms_occupancy.report'
|
|
|
|
@classmethod
|
|
def get_context(cls, records, data):
|
|
report_context = super(RoomsOccupancyReport, cls).get_context(records, data)
|
|
pool = Pool()
|
|
Company = pool.get('company.company')
|
|
Room = pool.get('hotel.room')
|
|
Operation = pool.get('hotel.operation')
|
|
start_date = data['date']
|
|
all_rooms = Room.search([], order=[('code', 'ASC')])
|
|
operations = Operation.search([
|
|
('start_date', '<=', start_date),
|
|
('state', '=', 'check_in'),
|
|
('kind', '=', 'occupancy'),
|
|
])
|
|
|
|
def _get_default_room(r):
|
|
res = {
|
|
'room': r.name,
|
|
'guest': None,
|
|
'num_guest': None,
|
|
'party': None,
|
|
'arrival': None,
|
|
'departure': None,
|
|
'booking': None,
|
|
'reference': None,
|
|
'amount': None,
|
|
'balance': None,
|
|
'state': None,
|
|
}
|
|
return res
|
|
|
|
rooms_map = {room.id: _get_default_room(room) for room in all_rooms}
|
|
occupancy_rooms = 0
|
|
for op in operations:
|
|
rooms_map[op.room.id].update({
|
|
'guest': op.main_guest.name if op.main_guest else op.party.name,
|
|
'num_guest': len(op.guests),
|
|
'party': op.party.name,
|
|
'arrival': op.start_date,
|
|
'departure': op.end_date,
|
|
'reference': op.reference,
|
|
'amount': op.total_amount,
|
|
'balance': op.total_amount_today,
|
|
'state': op.state,
|
|
})
|
|
occupancy_rooms += 1
|
|
|
|
if all_rooms:
|
|
occupancy_rate = (float(len(operations)) / len(all_rooms)) * 100
|
|
else:
|
|
occupancy_rate = 0
|
|
|
|
user = Pool().get('res.user')(Transaction().user)
|
|
report_context['records'] = rooms_map
|
|
report_context['occupancy_rate'] = occupancy_rate
|
|
report_context['occupancy_rooms'] = occupancy_rooms
|
|
report_context['company'] = Company(data['company']).party.name
|
|
report_context['date'] = data['date']
|
|
report_context['user'] = user.rec_name
|
|
return report_context
|
|
|
|
|
|
class BookingDailyStart(ModelView):
|
|
'Booking Daily Start'
|
|
__name__ = 'hotel.print_booking_daily.start'
|
|
date = fields.Date('Date', required=True)
|
|
company = fields.Many2One('company.company', 'Company', required=True)
|
|
|
|
@staticmethod
|
|
def default_date():
|
|
Date_ = Pool().get('ir.date')
|
|
return Date_.today()
|
|
|
|
@staticmethod
|
|
def default_company():
|
|
return Transaction().context.get('company')
|
|
|
|
|
|
class BookingDaily(Wizard):
|
|
'Rooms Occupancy'
|
|
__name__ = 'hotel.booking_daily'
|
|
start = StateView('hotel.print_booking_daily.start',
|
|
'hotel.print_booking_daily_start_view_form', [
|
|
Button('Cancel', 'end', 'tryton-cancel'),
|
|
Button('Open', 'print_', 'tryton-print', default=True),
|
|
])
|
|
print_ = StateReport('hotel.booking_daily.report')
|
|
|
|
def do_print_(self, action):
|
|
company = self.start.company
|
|
data = {
|
|
'date': self.start.date,
|
|
'company': company.id,
|
|
}
|
|
return action, data
|
|
|
|
def transition_print_(self):
|
|
return 'end'
|
|
|
|
|
|
class BookingDailyReport(Report):
|
|
__name__ = 'hotel.booking_daily.report'
|
|
|
|
@classmethod
|
|
def get_context(cls, records, data):
|
|
report_context = super(BookingDailyReport, cls).get_context(records, data)
|
|
pool = Pool()
|
|
Company = pool.get('company.company')
|
|
records = BookingLine.search([
|
|
('arrival_date', '=', data['date']),
|
|
], order=[('room.code', 'ASC')])
|
|
|
|
report_context['records'] = records
|
|
report_context['company'] = Company(data['company']).party.name
|
|
report_context['date'] = data['date']
|
|
return report_context
|