audrius/contract
1
0
Fork 1
mirror of https://github.com/OCA/contract.git synced 2025-02-13 17:57:24 +02:00
Code Issues Packages Projects Releases Wiki Activity

[IMP] contract: Allows to get several functions to create recurring documents

Browse Source
This commit is contained in:
Denis Roussel
2021-11-04 13:27:41 +01:00
parent f38f94b532
commit c4e240f157
1 changed files with 19 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
contract/models/contract.py
View File

@@ -6,6 +6,7 @@
# Copyright 2018 ACSONE SA/NV # Copyright 2018 ACSONE SA/NV
# Copyright 2021 Tecnativa - Víctor Martínez # Copyright 2021 Tecnativa - Víctor Martínez
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl). # License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
import logging
from odoo import api, fields, models from odoo import api, fields, models
from odoo.exceptions import UserError, ValidationError from odoo.exceptions import UserError, ValidationError
@@ -13,6 +14,8 @@ from odoo.osv import expression
from odoo.tests import Form from odoo.tests import Form
from odoo.tools.translate import _ from odoo.tools.translate import _
_logger = logging.getLogger(__name__)
class ContractContract(models.Model): class ContractContract(models.Model):
_name = "contract.contract" _name = "contract.contract"
@@ -597,6 +600,20 @@ class ContractContract(models.Model):
@api.model @api.model
def _cron_recurring_create(self, date_ref=False, create_type="invoice"): def _cron_recurring_create(self, date_ref=False, create_type="invoice"):
"""
The cron function in order to create recurrent documents
from contracts.
"""
_recurring_create_func = f"_recurring_create_{create_type}"
if not hasattr(self, _recurring_create_func):
_logger.info(
_(
"No function to create %s documents automatically is "
"declared in contract.contract model. Passing."
),
create_type,
)
return False
if not date_ref: if not date_ref:
date_ref = fields.Date.context_today(self) date_ref = fields.Date.context_today(self)
domain = self._get_contracts_to_invoice_domain(date_ref) domain = self._get_contracts_to_invoice_domain(date_ref)
@@ -606,22 +623,16 @@ class ContractContract(models.Model):
[("generation_type", "=", create_type)], [("generation_type", "=", create_type)],
] ]
) )
invoice_obj = self.env["account.move"]
contracts = self.search(domain) contracts = self.search(domain)
companies = set(contracts.mapped("company_id")) companies = set(contracts.mapped("company_id"))
invoice_ids = []
# Invoice by companies, so assignation emails get correct context # Invoice by companies, so assignation emails get correct context
# Use ids instead of recordset for perfomances reasons
for company in companies: for company in companies:
contracts_to_invoice = contracts.filtered( contracts_to_invoice = contracts.filtered(
lambda c: c.company_id == company lambda c: c.company_id == company
and (not c.date_end or c.recurring_next_date <= c.date_end) and (not c.date_end or c.recurring_next_date <= c.date_end)
).with_company(company) ).with_company(company)
invoice_ids.extend( getattr(contracts_to_invoice, _recurring_create_func)(date_ref)
contracts_to_invoice._recurring_create_invoice(date_ref).ids return True
)
return invoice_obj.browse(invoice_ids)
@api.model @api.model
def cron_recurring_create_invoice(self, date_ref=None): def cron_recurring_create_invoice(self, date_ref=None):