sale_margin/models/sale_order.py

32 lines
1.5 KiB
Python
Raw Permalink Normal View History

# -*- 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"
margin = fields.Monetary("Margin", compute='_compute_margin', store=True)
margin_percent = fields.Float("Margin (%)", compute='_compute_margin', store=True, group_operator="avg")
@api.depends('order_line.margin', 'amount_untaxed')
def _compute_margin(self):
if not all(self._ids):
for order in self:
order.margin = sum(order.order_line.mapped('margin'))
order.margin_percent = order.amount_untaxed and order.margin/order.amount_untaxed
else:
# On batch records recomputation (e.g. at install), compute the margins
# with a single read_group query for better performance.
# This isn't done in an onchange environment because (part of) the data
# may not be stored in database (new records or unsaved modifications).
grouped_order_lines_data = self.env['sale.order.line']._read_group(
[
('order_id', 'in', self.ids),
], ['order_id'], ['margin:sum'])
mapped_data = {order.id: margin for order, margin in grouped_order_lines_data}
for order in self:
order.margin = mapped_data.get(order.id, 0.0)
order.margin_percent = order.amount_untaxed and order.margin/order.amount_untaxed