|
|
# Copyright 2017 LasLabs Inc. # Copyright 2018 ACSONE SA/NV. # License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
from odoo import fields, api, models
class SaleOrder(models.Model): _inherit = 'sale.order'
is_contract = fields.Boolean( string='Is a contract', compute='_compute_is_contract' ) contract_count = fields.Integer(compute='_compute_contract_count') need_contract_creation = fields.Boolean( compute='_compute_need_contract_creation' )
@api.depends('order_line.contract_id', 'state') def _compute_need_contract_creation(self): for rec in self: if rec.state in ('sale', 'done'): line_to_create_contract = rec.order_line.filtered( lambda r: not r.contract_id and r.product_id.is_contract ) line_to_update_contract = rec.order_line.filtered( lambda r: r.contract_id and r.product_id.is_contract and r not in r.contract_id.contract_line_ids.mapped( 'sale_order_line_id' ) ) if line_to_create_contract or line_to_update_contract: rec.need_contract_creation = True
@api.depends('order_line') def _compute_is_contract(self): self.is_contract = any(self.order_line.mapped('is_contract'))
@api.multi def _prepare_contract_value(self, contract_template): self.ensure_one() return { 'name': '{template_name}: {sale_name}'.format( template_name=contract_template.name, sale_name=self.name ), 'partner_id': self.partner_id.id, 'company_id': self.company_id.id, 'contract_template_id': contract_template.id, 'user_id': self.user_id.id, 'payment_term_id': self.payment_term_id.id, 'fiscal_position_id': self.fiscal_position_id.id, 'invoice_partner_id': self.partner_invoice_id.id, }
@api.multi def action_create_contract(self): contract_model = self.env['contract.contract'] contracts = self.env['contract.contract'] for rec in self.filtered('is_contract'): line_to_create_contract = rec.order_line.filtered( lambda r: not r.contract_id and r.product_id.is_contract ) line_to_update_contract = rec.order_line.filtered( lambda r: r.contract_id and r.product_id.is_contract and r not in r.contract_id.contract_line_ids.mapped( 'sale_order_line_id' ) ) for contract_template in line_to_create_contract.mapped( 'product_id.contract_template_id' ): order_lines = line_to_create_contract.filtered( lambda r, template=contract_template: r.product_id.contract_template_id == template ) contract = contract_model.create( rec._prepare_contract_value(contract_template) ) contracts |= contract contract._onchange_contract_template_id() order_lines.create_contract_line(contract) order_lines.write({'contract_id': contract.id}) for line in line_to_update_contract: line.create_contract_line(line.contract_id) return contracts
@api.multi def action_confirm(self): """ If we have a contract in the order, set it up """ self.filtered( lambda order: ( order.company_id.create_contract_at_sale_order_confirmation ) ).action_create_contract() return super(SaleOrder, self).action_confirm()
@api.multi @api.depends("order_line") def _compute_contract_count(self): for rec in self: rec.contract_count = len(rec.order_line.mapped('contract_id'))
@api.multi def action_show_contracts(self): self.ensure_one() action = self.env.ref( "contract.action_customer_contract" ).read()[0] contracts = ( self.env['contract.line'] .search([('sale_order_line_id', 'in', self.order_line.ids)]) .mapped('contract_id') ) action["domain"] = [("id", "in", contracts.ids)] return action
|