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

[14.0][IMP] contract: Improve function call for cron recurring creates

Browse Source
This commit is contained in:
Denis Roussel
2022-01-25 14:59:55 +01:00
parent c80f7e0e6b
commit eb40551989
1 changed files with 13 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
contract/models/contract.py
View File

@@ -598,22 +598,24 @@ class ContractContract(models.Model):
self._compute_recurring_next_date() self._compute_recurring_next_date()
return moves return moves
@api.model
def _get_recurring_create_func(self, create_type="invoice"):
"""
Allows to retrieve the recurring create function depending
on generate_type attribute
"""
if create_type == "invoice":
return self.__class__._recurring_create_invoice
@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 The cron function in order to create recurrent documents
from contracts. from contracts.
""" """
_recurring_create_func = f"_recurring_create_{create_type}" _recurring_create_func = self._get_recurring_create_func(
if not hasattr(self, _recurring_create_func): create_type=create_type
_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)
@@ -631,7 +633,7 @@ class ContractContract(models.Model):
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)
getattr(contracts_to_invoice, _recurring_create_func)(date_ref) _recurring_create_func(contracts_to_invoice, date_ref)
return True return True
@api.model @api.model