from odoo import fields, models


class MembershipInvoice(models.TransientModel):
    _inherit = "membership.invoice"

    payment_mode_id = fields.Many2one(
        comodel_name="account.payment.mode",
        string="Payment Mode",
        domain=[("payment_type", "=", "inbound")],
    )

    date_invoice = fields.Date(string="Invoice Date")

    def membership_invoice(self):
        res = super().membership_invoice()
        invoice_ids = None
        for d in res["domain"]:
            if d[0] == "id" and d[1] == "in":
                invoice_ids = d[2]
                if invoice_ids:
                    self.env["account.move"].browse(invoice_ids).write(
                        {
                            "payment_mode_id": self.payment_mode_id.id,
                            "invoice_date": self.date_invoice,
                        },
                    )
        return res