# © 2017 Eficent Business and IT Consulting Services S.L.
#        (http://www.eficent.com)
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl.html).

from odoo import api, models


class BomStructureReport(models.AbstractModel):
    _inherit = 'report.mrp.report_bom_structure'

    @api.model
    def _get_bom_lines(self, bom, bom_quantity, product, line_id, level):
        res = super(BomStructureReport, self)._get_bom_lines(
            bom, bom_quantity, product, line_id, level)
        line_ids = self.env['mrp.bom.line'].search([('bom_id', '=', bom.id)])
        for line in res[0]:
            line_id = line_ids.filtered(
                lambda l: l.location_id and l.id == line['line_id'])
            line['location_id'] = line_id.location_id or ''
        return res

    @api.model
    def _get_pdf_line(self, bom_id, product_id=False, qty=1,
                      child_bom_ids=None, unfolded=False):
        res = super(BomStructureReport, self)._get_pdf_line(
            bom_id, product_id, qty, child_bom_ids, unfolded)
        line_ids = self.env['mrp.bom.line'].search([('bom_id', '=', bom_id)])
        for line in res['lines']:
            line_id = line_ids.filtered(
                lambda l: l.location_id and
                l.product_id.display_name == line['name'])
            line['location_name'] = line_id.location_id.complete_name or ''
        return res