sale_loyalty/models/loyalty_program.py

28 lines
1.0 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 fields, models
class LoyaltyProgram(models.Model):
_inherit = 'loyalty.program'
order_count = fields.Integer(compute='_compute_order_count')
sale_ok = fields.Boolean(string="Sales", default=True)
def _compute_order_count(self):
# An order should count only once PER program but may appear in multiple programs
read_group_res = self.env['sale.order.line']._read_group(
[('reward_id', 'in', self.reward_ids.ids)], ['order_id'], ['reward_id:array_agg'])
for program in self:
program_reward_ids = program.reward_ids.ids
program.order_count = sum(
any(id_ in reward_ids for id_ in program_reward_ids)
for __, reward_ids in read_group_res
)
def _compute_total_order_count(self):
super()._compute_total_order_count()
for program in self:
program.total_order_count += program.order_count