# Copyright (C) 2015-2019 Akretion (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 collections import OrderedDict

from odoo import api, fields, models


class AccountMove(models.Model):
    _inherit = "account.move"

    # sale_ids is kind of the symetric field of invoice_ids on sale.order
    sale_ids = fields.Many2many(
        "sale.order", string="Sale Orders", compute="_compute_sale_ids"
    )
    sale_count = fields.Integer(string="Sale Order Count", compute="_compute_sale_ids")

    @api.depends("invoice_line_ids.sale_line_ids")
    def _compute_sale_ids(self):
        for invoice in self:
            if invoice.move_type == "out_invoice":
                sales = invoice.invoice_line_ids.mapped("sale_line_ids").mapped(
                    "order_id"
                )
                invoice.sale_ids = sales.ids
                invoice.sale_count = len(sales.ids)
            else:
                invoice.sale_ids = []
                invoice.sale_count = 0

    def show_sale_orders(self):
        self.ensure_one()
        action = self.env.ref("sale.action_orders").read()[0]
        sales = self.sale_ids
        if len(sales) > 1:
            action["domain"] = [("id", "in", sales.ids)]
        else:
            action.update(
                {
                    "res_id": sales.id,
                    "view_mode": "form,tree,kanban,calendar,pivot,graph,activity",
                    "views": False,
                }
            )
        return action

    def py3o_lines_layout_groupby_order(self, subtotal=True):
        # This method is an alternative to the method py3o_lines_layout()
        # defined above: you just have to change the call in the invoice
        # ODT template
        self.ensure_one()
        res1 = OrderedDict()
        # {categ(1): {'lines': [l1, l2], 'subtotal': 23.32}}
        soo = self.env["sale.order"]
        for line in self.invoice_line_ids:
            order = (
                not line.display_type
                and line.sale_line_ids
                and line.sale_line_ids[0].order_id
                or soo
            )
            if order in res1:
                res1[order]["lines"].append(line)
                res1[order]["subtotal"] += line.price_subtotal
            else:
                res1[order] = {"lines": [line], "subtotal": line.price_subtotal}
        # from pprint import pprint
        # pprint(res1)
        res2 = []
        if len(res1) == 1 and not list(res1)[0]:
            # No order at all
            for line in list(res1.values())[0]["lines"]:
                res2.append({"line": line})
        else:
            for order, ldict in res1.items():
                res2.append({"categ": order})
                for line in ldict["lines"]:
                    res2.append({"line": line})
                if subtotal:
                    res2.append({"subtotal": ldict["subtotal"]})
        # res2:
        # [
        #    {'categ': categ(1)},
        #    {'line': invoice_line(2)},
        #    {'line': invoice_line(3)},
        #    {'subtotal': 8932.23},
        # ]
        # pprint(res2)
        return res2