Odoo18-Base/addons/purchase_stock/report/stock_forecasted.py
2025-03-10 11:12:23 +07:00

39 lines
2.0 KiB
Python

# -*- coding: utf-8 -*-
# Part of Odoo. See LICENSE file for full copyright and licensing details.
from odoo import models
class ReplenishmentReport(models.AbstractModel):
_inherit = 'report.stock.report_product_product_replenishment'
def _serialize_docs(self, docs, product_template_ids=False, product_variant_ids=False):
res = super()._serialize_docs(docs, product_template_ids, product_variant_ids)
res['draft_purchase_orders'] = docs['draft_purchase_orders'].read(fields=['id', 'name'])
return res
def _compute_draft_quantity_count(self, product_template_ids, product_variant_ids, wh_location_ids):
res = super()._compute_draft_quantity_count(product_template_ids, product_variant_ids, wh_location_ids)
domain = [('state', 'in', ['draft', 'sent', 'to approve'])]
domain += self._product_purchase_domain(product_template_ids, product_variant_ids)
warehouse_id = self.env['stock.warehouse']._get_warehouse_id_from_context()
if warehouse_id:
domain += [('order_id.picking_type_id.warehouse_id', '=', warehouse_id)]
po_lines = self.env['purchase.order.line'].search(domain)
in_sum = sum(po_lines.mapped('product_uom_qty'))
res['draft_purchase_qty'] = in_sum
res['draft_purchase_orders'] = po_lines.mapped("order_id").sorted(key=lambda po: po.name)
res['draft_purchase_orders_matched'] = self.env.context.get('purchase_line_to_match_id') in po_lines.ids
res['qty']['in'] += in_sum
return res
def _product_purchase_domain(self, product_template_ids, product_variant_ids):
if product_variant_ids:
return [('product_id', 'in', product_variant_ids)]
elif product_template_ids:
products = self.env['product.product'].search_read(
[('product_tmpl_id', 'in', product_template_ids)], ['id']
)
product_ids = [product['id'] for product in products]
return [('product_id', 'in', product_ids)]