# -*- coding: utf-8 -*-
# Part of Odoo. See LICENSE file for full copyright and licensing details.
from odoo import api, fields, models


class StockRule(models.Model):
    _inherit = 'stock.rule'

    def _prepare_mo_vals(self, product_id, product_qty, product_uom, location_id, name, origin, company_id, values,
                         bom):
        res = super()._prepare_mo_vals(product_id, product_qty, product_uom, location_id, name, origin, company_id,
                                       values, bom)
        if self.env.context.get('demand_plan_line_id'):
            res['demand_plan_line_id'] = self.env.context.get('demand_plan_line_id')
        return res

    @api.model
    def _prepare_purchase_request_line(self, request_id, procurement):
        res = super()._prepare_purchase_request_line(request_id, procurement)
        if self.env.context.get('demand_plan_line_id'):
            res['demand_plan_line_id'] = self.env.context.get('demand_plan_line_id')
        return res