29 lines
1.2 KiB
Python
29 lines
1.2 KiB
Python
# Copyright 2017-2021 Akretion France (http://www.akretion.com/)
|
|
# @author: Alexis de Lattre <alexis.delattre@akretion.com>
|
|
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
|
|
|
|
from odoo import fields, models
|
|
|
|
|
|
class PosCategory(models.Model):
|
|
_inherit = 'pos.category'
|
|
|
|
product_count = fields.Integer(
|
|
'# Products', compute='_compute_product_count',
|
|
help="The number of products under this point of sale category "
|
|
"(children categories included)")
|
|
|
|
# inspired by the code of odoo/addons/product/models/product_category.py
|
|
def _compute_product_count(self):
|
|
read_group_res = self.env['product.template'].read_group(
|
|
[('pos_categ_id', 'child_of', self.ids)],
|
|
['pos_categ_id'], ['pos_categ_id'])
|
|
group_data = dict(
|
|
(data['pos_categ_id'][0], data['pos_categ_id_count']) for data
|
|
in read_group_res)
|
|
for pos_categ in self:
|
|
product_count = 0
|
|
for sub_categ_id in pos_categ.search([('id', 'child_of', pos_categ.ids)]).ids:
|
|
product_count += group_data.get(sub_categ_id, 0)
|
|
pos_categ.product_count = product_count
|