diff --git a/contract/wizards/contract_manually_create_invoice.py b/contract/wizards/contract_manually_create_invoice.py index 3dad2271..78af2c91 100644 --- a/contract/wizards/contract_manually_create_invoice.py +++ b/contract/wizards/contract_manually_create_invoice.py @@ -11,56 +11,44 @@ class ContractManuallyCreateInvoice(models.TransientModel): invoice_date = fields.Date(string="Invoice Date", required=True) contract_to_invoice_count = fields.Integer( - compute="_compute_contract_to_invoice_count" + compute="_compute_contract_to_invoice_ids" + ) + contract_to_invoice_ids = fields.Many2many( + comodel_name="contract.contract", + compute="_compute_contract_to_invoice_ids", ) - @api.multi - def action_show_contract_to_invoice(self): - self.ensure_one() + @api.depends('invoice_date') + def _compute_contract_to_invoice_ids(self): if not self.invoice_date: contract_to_invoice_domain = [('id', '=', False)] else: contract_to_invoice_domain = self.env[ 'contract.contract' ]._get_contracts_to_invoice_domain(self.invoice_date) + self.contract_to_invoice_ids = self.env['contract.contract'].search( + contract_to_invoice_domain + ) + self.contract_to_invoice_count = len(self.contract_to_invoice_ids) + @api.multi + def action_show_contract_to_invoice(self): + self.ensure_one() return { "type": "ir.actions.act_window", "name": _("Contracts to invoice"), "res_model": "contract.contract", - "domain": contract_to_invoice_domain, + "domain": [('id', 'in', self.contract_to_invoice_ids.ids)], "view_mode": "tree,form", "context": self.env.context, } - @api.depends('invoice_date') - def _compute_contract_to_invoice_count(self): - self.ensure_one() - if not self.invoice_date: - self.contract_to_invoice_count = 0 - else: - contract_model = self.env['contract.contract'] - self.contract_to_invoice_count = contract_model.search_count( - contract_model._get_contracts_to_invoice_domain( - self.invoice_date - ) - ) - @api.multi def create_invoice(self): self.ensure_one() - contract_model = self.env['contract.contract'] - contracts = contract_model.search( - contract_model._get_contracts_to_invoice_domain(self.invoice_date) - ) - invoices = contract_model.cron_recurring_create_invoice( - self.invoice_date - ) - for contract in contracts: - contract.message_post( - body=_("Manually invoiced. Invoice date: %s") - % self.invoice_date - ) + invoices = self.env['account.invoice'] + for contract in self.contract_to_invoice_ids: + invoices |= contract.recurring_create_invoice() return { "type": "ir.actions.act_window", "name": _("Invoices"),