36 lines
1.3 KiB
Python
36 lines
1.3 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
# Part of Odoo. See LICENSE file for full copyright and licensing details.
|
||
|
|
||
|
from odoo import api, fields, models
|
||
|
|
||
|
|
||
|
class SaleOrder(models.Model):
|
||
|
_inherit = 'sale.order'
|
||
|
|
||
|
event_booth_ids = fields.One2many('event.booth', 'sale_order_id', string='Booths')
|
||
|
event_booth_count = fields.Integer(string='Booth Count', compute='_compute_event_booth_count')
|
||
|
|
||
|
@api.depends('event_booth_ids')
|
||
|
def _compute_event_booth_count(self):
|
||
|
if self.ids:
|
||
|
slot_data = self.env['event.booth']._read_group(
|
||
|
[('sale_order_id', 'in', self.ids)],
|
||
|
['sale_order_id'], ['sale_order_id']
|
||
|
)
|
||
|
slot_mapped = dict((data['sale_order_id'][0], data['sale_order_id_count']) for data in slot_data)
|
||
|
else:
|
||
|
slot_mapped = dict()
|
||
|
for so in self:
|
||
|
so.event_booth_count = slot_mapped.get(so.id, 0)
|
||
|
|
||
|
def action_confirm(self):
|
||
|
res = super(SaleOrder, self).action_confirm()
|
||
|
for so in self:
|
||
|
so.order_line._update_event_booths()
|
||
|
return res
|
||
|
|
||
|
def action_view_booth_list(self):
|
||
|
action = self.env['ir.actions.act_window']._for_xml_id('event_booth.event_booth_action')
|
||
|
action['domain'] = [('sale_order_id', 'in', self.ids)]
|
||
|
return action
|