367 lines
16 KiB
Python
367 lines
16 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Part of Odoo, Flectra. See LICENSE file for full copyright and licensing details.
|
|
|
|
import datetime
|
|
import logging
|
|
import time
|
|
import traceback
|
|
from collections import defaultdict
|
|
|
|
import dateutil
|
|
from dateutil.relativedelta import relativedelta
|
|
|
|
from flectra import api, fields, models, SUPERUSER_ID
|
|
from flectra.modules.registry import Registry
|
|
from flectra.tools import DEFAULT_SERVER_DATETIME_FORMAT
|
|
from flectra.tools.safe_eval import safe_eval
|
|
|
|
_logger = logging.getLogger(__name__)
|
|
|
|
DATE_RANGE_FUNCTION = {
|
|
'minutes': lambda interval: relativedelta(minutes=interval),
|
|
'hour': lambda interval: relativedelta(hours=interval),
|
|
'day': lambda interval: relativedelta(days=interval),
|
|
'month': lambda interval: relativedelta(months=interval),
|
|
False: lambda interval: relativedelta(0),
|
|
}
|
|
|
|
|
|
class BaseAutomation(models.Model):
|
|
_name = 'base.automation'
|
|
_description = 'Automated Action'
|
|
_order = 'sequence'
|
|
|
|
action_server_id = fields.Many2one(
|
|
'ir.actions.server', 'Server Actions',
|
|
domain="[('model_id', '=', model_id)]",
|
|
delegate=True, required=True, ondelete='restrict')
|
|
active = fields.Boolean(default=True, help="When unchecked, the rule is hidden and will not be executed.")
|
|
trigger = fields.Selection([
|
|
('on_create', 'On Creation'),
|
|
('on_write', 'On Update'),
|
|
('on_create_or_write', 'On Creation & Update'),
|
|
('on_unlink', 'On Deletion'),
|
|
('on_change', 'Based on Form Modification'),
|
|
('on_time', 'Based on Timed Condition')
|
|
], string='Trigger Condition', required=True, oldname="kind")
|
|
trg_date_id = fields.Many2one('ir.model.fields', string='Trigger Date',
|
|
help="""When should the condition be triggered.
|
|
If present, will be checked by the scheduler. If empty, will be checked at creation and update.""",
|
|
domain="[('model_id', '=', model_id), ('ttype', 'in', ('date', 'datetime'))]")
|
|
trg_date_range = fields.Integer(string='Delay after trigger date',
|
|
help="""Delay after the trigger date.
|
|
You can put a negative number if you need a delay before the
|
|
trigger date, like sending a reminder 15 minutes before a meeting.""")
|
|
trg_date_range_type = fields.Selection([('minutes', 'Minutes'), ('hour', 'Hours'), ('day', 'Days'), ('month', 'Months')],
|
|
string='Delay type', default='day')
|
|
trg_date_calendar_id = fields.Many2one("resource.calendar", string='Use Calendar',
|
|
help="When calculating a day-based timed condition, it is possible to use a calendar to compute the date based on working days.")
|
|
filter_pre_domain = fields.Char(string='Before Update Domain',
|
|
help="If present, this condition must be satisfied before the update of the record.")
|
|
filter_domain = fields.Char(string='Apply on', help="If present, this condition must be satisfied before executing the action rule.")
|
|
last_run = fields.Datetime(readonly=True, copy=False)
|
|
on_change_fields = fields.Char(string="On Change Fields Trigger", help="Comma-separated list of field names that triggers the onchange.")
|
|
|
|
# which fields have an impact on the registry
|
|
CRITICAL_FIELDS = ['model_id', 'active', 'trigger', 'on_change_fields']
|
|
|
|
@api.onchange('model_id')
|
|
def onchange_model_id(self):
|
|
self.model_name = self.model_id.model
|
|
|
|
@api.onchange('trigger')
|
|
def onchange_trigger(self):
|
|
if self.trigger in ['on_create', 'on_create_or_write', 'on_unlink']:
|
|
self.filter_pre_domain = self.trg_date_id = self.trg_date_range = self.trg_date_range_type = False
|
|
elif self.trigger in ['on_write', 'on_create_or_write']:
|
|
self.trg_date_id = self.trg_date_range = self.trg_date_range_type = False
|
|
elif self.trigger == 'on_time':
|
|
self.filter_pre_domain = False
|
|
|
|
@api.model
|
|
def create(self, vals):
|
|
vals['usage'] = 'base_automation'
|
|
base_automation = super(BaseAutomation, self).create(vals)
|
|
self._update_cron()
|
|
self._update_registry()
|
|
return base_automation
|
|
|
|
@api.multi
|
|
def write(self, vals):
|
|
res = super(BaseAutomation, self).write(vals)
|
|
if set(vals).intersection(self.CRITICAL_FIELDS):
|
|
self._update_cron()
|
|
self._update_registry()
|
|
return res
|
|
|
|
@api.multi
|
|
def unlink(self):
|
|
res = super(BaseAutomation, self).unlink()
|
|
self._update_cron()
|
|
self._update_registry()
|
|
return res
|
|
|
|
def _update_cron(self):
|
|
""" Activate the cron job depending on whether there exists action rules
|
|
based on time conditions.
|
|
"""
|
|
cron = self.env.ref('base_automation.ir_cron_data_base_automation_check', raise_if_not_found=False)
|
|
return cron and cron.toggle(model=self._name, domain=[('trigger', '=', 'on_time')])
|
|
|
|
def _update_registry(self):
|
|
""" Update the registry after a modification on action rules. """
|
|
if self.env.registry.ready and not self.env.context.get('import_file'):
|
|
# for the sake of simplicity, simply force the registry to reload
|
|
self._cr.commit()
|
|
self.env.reset()
|
|
registry = Registry.new(self._cr.dbname)
|
|
registry.registry_invalidated = True
|
|
|
|
def _get_actions(self, records, triggers):
|
|
""" Return the actions of the given triggers for records' model. The
|
|
returned actions' context contain an object to manage processing.
|
|
"""
|
|
if '__action_done' not in self._context:
|
|
self = self.with_context(__action_done={})
|
|
domain = [('model_name', '=', records._name), ('trigger', 'in', triggers)]
|
|
actions = self.with_context(active_test=True).search(domain)
|
|
return actions.with_env(self.env)
|
|
|
|
def _get_eval_context(self):
|
|
""" Prepare the context used when evaluating python code
|
|
:returns: dict -- evaluation context given to safe_eval
|
|
"""
|
|
return {
|
|
'datetime': datetime,
|
|
'dateutil': dateutil,
|
|
'time': time,
|
|
'uid': self.env.uid,
|
|
'user': self.env.user,
|
|
}
|
|
|
|
def _filter_pre(self, records):
|
|
""" Filter the records that satisfy the precondition of action ``self``. """
|
|
if self.filter_pre_domain and records:
|
|
domain = [('id', 'in', records.ids)] + safe_eval(self.filter_pre_domain, self._get_eval_context())
|
|
return records.search(domain)
|
|
else:
|
|
return records
|
|
|
|
def _filter_post(self, records):
|
|
""" Filter the records that satisfy the postcondition of action ``self``. """
|
|
if self.filter_domain and records:
|
|
domain = [('id', 'in', records.ids)] + safe_eval(self.filter_domain, self._get_eval_context())
|
|
return records.search(domain)
|
|
else:
|
|
return records
|
|
|
|
def _process(self, records):
|
|
""" Process action ``self`` on the ``records`` that have not been done yet. """
|
|
# filter out the records on which self has already been done
|
|
action_done = self._context['__action_done']
|
|
records_done = action_done.get(self, records.browse())
|
|
records -= records_done
|
|
if not records:
|
|
return
|
|
|
|
# mark the remaining records as done (to avoid recursive processing)
|
|
action_done = dict(action_done)
|
|
action_done[self] = records_done + records
|
|
self = self.with_context(__action_done=action_done)
|
|
records = records.with_context(__action_done=action_done)
|
|
|
|
# modify records
|
|
values = {}
|
|
if 'date_action_last' in records._fields:
|
|
values['date_action_last'] = fields.Datetime.now()
|
|
if values:
|
|
records.write(values)
|
|
|
|
# execute server actions
|
|
if self.action_server_id:
|
|
for record in records:
|
|
ctx = {'active_model': record._name, 'active_ids': record.ids, 'active_id': record.id}
|
|
self.action_server_id.with_context(**ctx).run()
|
|
|
|
@api.model_cr
|
|
def _register_hook(self):
|
|
""" Patch models that should trigger action rules based on creation,
|
|
modification, deletion of records and form onchanges.
|
|
"""
|
|
#
|
|
# Note: the patched methods must be defined inside another function,
|
|
# otherwise their closure may be wrong. For instance, the function
|
|
# create refers to the outer variable 'create', which you expect to be
|
|
# bound to create itself. But that expectation is wrong if create is
|
|
# defined inside a loop; in that case, the variable 'create' is bound to
|
|
# the last function defined by the loop.
|
|
#
|
|
|
|
def make_create():
|
|
""" Instanciate a create method that processes action rules. """
|
|
@api.model
|
|
def create(self, vals, **kw):
|
|
# retrieve the action rules to possibly execute
|
|
actions = self.env['base.automation']._get_actions(self, ['on_create', 'on_create_or_write'])
|
|
# call original method
|
|
record = create.origin(self.with_env(actions.env), vals, **kw)
|
|
# check postconditions, and execute actions on the records that satisfy them
|
|
for action in actions.with_context(old_values=None):
|
|
action._process(action._filter_post(record))
|
|
return record.with_env(self.env)
|
|
|
|
return create
|
|
|
|
def make_write():
|
|
""" Instanciate a _write method that processes action rules. """
|
|
#
|
|
# Note: we patch method _write() instead of write() in order to
|
|
# catch updates made by field recomputations.
|
|
#
|
|
@api.multi
|
|
def _write(self, vals, **kw):
|
|
# retrieve the action rules to possibly execute
|
|
actions = self.env['base.automation']._get_actions(self, ['on_write', 'on_create_or_write'])
|
|
records = self.with_env(actions.env)
|
|
# check preconditions on records
|
|
pre = {action: action._filter_pre(records) for action in actions}
|
|
# read old values before the update
|
|
old_values = {
|
|
old_vals.pop('id'): old_vals
|
|
for old_vals in records.read(list(vals))
|
|
}
|
|
# call original method
|
|
_write.origin(records, vals, **kw)
|
|
# check postconditions, and execute actions on the records that satisfy them
|
|
for action in actions.with_context(old_values=old_values):
|
|
action._process(action._filter_post(pre[action]))
|
|
return True
|
|
|
|
return _write
|
|
|
|
def make_unlink():
|
|
""" Instanciate an unlink method that processes action rules. """
|
|
@api.multi
|
|
def unlink(self, **kwargs):
|
|
# retrieve the action rules to possibly execute
|
|
actions = self.env['base.automation']._get_actions(self, ['on_unlink'])
|
|
records = self.with_env(actions.env)
|
|
# check conditions, and execute actions on the records that satisfy them
|
|
for action in actions:
|
|
action._process(action._filter_post(records))
|
|
# call original method
|
|
return unlink.origin(self, **kwargs)
|
|
|
|
return unlink
|
|
|
|
def make_onchange(action_rule_id):
|
|
""" Instanciate an onchange method for the given action rule. """
|
|
def base_automation_onchange(self):
|
|
action_rule = self.env['base.automation'].browse(action_rule_id)
|
|
result = {}
|
|
server_action = action_rule.action_server_id.with_context(active_model=self._name, onchange_self=self)
|
|
res = server_action.run()
|
|
if res:
|
|
if 'value' in res:
|
|
res['value'].pop('id', None)
|
|
self.update({key: val for key, val in res['value'].items() if key in self._fields})
|
|
if 'domain' in res:
|
|
result.setdefault('domain', {}).update(res['domain'])
|
|
if 'warning' in res:
|
|
result['warning'] = res['warning']
|
|
return result
|
|
|
|
return base_automation_onchange
|
|
|
|
patched_models = defaultdict(set)
|
|
def patch(model, name, method):
|
|
""" Patch method `name` on `model`, unless it has been patched already. """
|
|
if model not in patched_models[name]:
|
|
patched_models[name].add(model)
|
|
model._patch_method(name, method)
|
|
|
|
# retrieve all actions, and patch their corresponding model
|
|
for action_rule in self.with_context({}).search([]):
|
|
Model = self.env.get(action_rule.model_name)
|
|
|
|
# Do not crash if the model of the base_action_rule was uninstalled
|
|
if Model is None:
|
|
_logger.warning("Action rule with ID %d depends on model %s" %
|
|
(action_rule.id,
|
|
action_rule.model_name))
|
|
continue
|
|
|
|
if action_rule.trigger == 'on_create':
|
|
patch(Model, 'create', make_create())
|
|
|
|
elif action_rule.trigger == 'on_create_or_write':
|
|
patch(Model, 'create', make_create())
|
|
patch(Model, '_write', make_write())
|
|
|
|
elif action_rule.trigger == 'on_write':
|
|
patch(Model, '_write', make_write())
|
|
|
|
elif action_rule.trigger == 'on_unlink':
|
|
patch(Model, 'unlink', make_unlink())
|
|
|
|
elif action_rule.trigger == 'on_change':
|
|
# register an onchange method for the action_rule
|
|
method = make_onchange(action_rule.id)
|
|
for field_name in action_rule.on_change_fields.split(","):
|
|
Model._onchange_methods[field_name.strip()].append(method)
|
|
|
|
@api.model
|
|
def _check_delay(self, action, record, record_dt):
|
|
if action.trg_date_calendar_id and action.trg_date_range_type == 'day':
|
|
return action.trg_date_calendar_id.plan_days(
|
|
action.trg_date_range,
|
|
fields.Datetime.from_string(record_dt),
|
|
compute_leaves=True,
|
|
)
|
|
else:
|
|
delay = DATE_RANGE_FUNCTION[action.trg_date_range_type](action.trg_date_range)
|
|
return fields.Datetime.from_string(record_dt) + delay
|
|
|
|
@api.model
|
|
def _check(self, automatic=False, use_new_cursor=False):
|
|
""" This Function is called by scheduler. """
|
|
if '__action_done' not in self._context:
|
|
self = self.with_context(__action_done={})
|
|
|
|
# retrieve all the action rules to run based on a timed condition
|
|
eval_context = self._get_eval_context()
|
|
for action in self.with_context(active_test=True).search([('trigger', '=', 'on_time')]):
|
|
last_run = fields.Datetime.from_string(action.last_run) or datetime.datetime.utcfromtimestamp(0)
|
|
|
|
# retrieve all the records that satisfy the action's condition
|
|
domain = []
|
|
context = dict(self._context)
|
|
if action.filter_domain:
|
|
domain = safe_eval(action.filter_domain, eval_context)
|
|
records = self.env[action.model_name].with_context(context).search(domain)
|
|
|
|
# determine when action should occur for the records
|
|
if action.trg_date_id.name == 'date_action_last' and 'create_date' in records._fields:
|
|
get_record_dt = lambda record: record[action.trg_date_id.name] or record.create_date
|
|
else:
|
|
get_record_dt = lambda record: record[action.trg_date_id.name]
|
|
|
|
# process action on the records that should be executed
|
|
now = datetime.datetime.now()
|
|
for record in records:
|
|
record_dt = get_record_dt(record)
|
|
if not record_dt:
|
|
continue
|
|
action_dt = self._check_delay(action, record, record_dt)
|
|
if last_run <= action_dt < now:
|
|
try:
|
|
action._process(record)
|
|
except Exception:
|
|
_logger.error(traceback.format_exc())
|
|
|
|
action.write({'last_run': now.strftime(DEFAULT_SERVER_DATETIME_FORMAT)})
|
|
|
|
if automatic:
|
|
# auto-commit for batch processing
|
|
self._cr.commit()
|