import base64

from odoo import _, models
from odoo.exceptions import UserError
from odoo.tools import float_is_zero


class SaleOrderBatch(models.Model):
    _inherit = "sale.order.batch"

    def export_wholesaler_csv(self):
        self.ensure_one()
        # Prepare CSV content
        csv_content = "Artnr,Menge\n"
        for product in self.product_ids.filtered(lambda p: p.product_id.default_code):
            if not float_is_zero(
                product.product_uom_qty,
                precision_rounding=product.product_id.uom_id.rounding,
            ):
                # We think 10g variance is okay
                if not float_is_zero(
                    product.open_packaging_qty, precision_rounding=0.01
                ):
                    raise UserError(
                        _(
                            f"Packaging is still open for product {product.product_id.name}"
                        )
                    )
                wholesaler_qty = int(
                    round(product.product_uom_qty / product.product_packaging_qty)
                )
                csv_content += f"{product.product_id.default_code},{wholesaler_qty}\n"  # noqa: E231

        # Encode CSV content to base64
        csv_base64 = base64.b64encode(csv_content.encode("utf-8"))

        # Create attachment
        attachment = self.env["ir.attachment"].create(
            {
                "name": f"wholesaler-{self.name}.csv",
                "type": "binary",
                "datas": csv_base64,
                "res_model": self._name,
                "res_id": self.id,
            }
        )

        return {
            "type": "ir.actions.act_url",
            "url": f"/web/content/{attachment.id}?download=true",
            "target": "self",
        }