def get_needaction_data(self):
""" Return for each menu entry in ``self``:
- whether it uses the needaction mechanism (needaction_enabled)
- the needaction counter of the related action, taking into account
the action domain
"""
menu_ids = set()
for menu in self:
menu_ids.add(menu.id)
ctx = {}
if menu.action and menu.action.type in (
'ir.actions.act_window',
'ir.actions.client') and menu.action.context:
with tools.ignore(Exception):
# use magical UnquoteEvalContext to ignore undefined client-side variables such as `active_id`
eval_ctx = tools.UnquoteEvalContext(self._context)
ctx = eval(menu.action.context,
locals_dict=eval_ctx,
nocopy=True) or {}
menu_refs = ctx.get('needaction_menu_ref')
if menu_refs:
if not isinstance(menu_refs, list):
menu_refs = [menu_refs]
for menu_ref in menu_refs:
record = self.env.ref(menu_ref, False)
if record and record._name == 'ir.ui.menu':
menu_ids.add(record.id)
res = {}
for menu in self.browse(menu_ids):
res[menu.id] = {
'needaction_enabled': False,
'needaction_counter': False,
}
if menu.action and menu.action.type in (
'ir.actions.act_window',
'ir.actions.client') and menu.action.res_model:
if menu.action.res_model in self.env:
model = self.env[menu.action.res_model]
if model._needaction:
if menu.action.type == 'ir.actions.act_window':
eval_context = self.env[
'ir.actions.act_window']._get_eval_context()
dom = eval(menu.action.domain or '[]',
eval_context)
else:
dom = eval(menu.action.params_store or '{}', {
'uid': self._uid
}).get('domain')
res[menu.id]['needaction_enabled'] = model._needaction
res[menu.id][
'needaction_counter'] = model._needaction_count(
dom)
return res
def get_needaction_data(self, cr, uid, ids, context=None):
""" Return for each menu entry of ids :
- if it uses the needaction mechanism (needaction_enabled)
- the needaction counter of the related action, taking into account
the action domain
"""
if context is None:
context = {}
res = {}
menu_ids = set()
for menu in self.browse(cr, uid, ids, context=context):
menu_ids.add(menu.id)
ctx = None
if menu.action and menu.action.type in (
'ir.actions.act_window',
'ir.actions.client') and menu.action.context:
try:
# use magical UnquoteEvalContext to ignore undefined client-side variables such as `active_id`
eval_ctx = tools.UnquoteEvalContext(**context)
ctx = eval(menu.action.context,
locals_dict=eval_ctx,
nocopy=True) or None
except Exception:
# if the eval still fails for some reason, we'll simply skip this menu
pass
menu_ref = ctx and ctx.get('needaction_menu_ref')
if menu_ref:
if not isinstance(menu_ref, list):
menu_ref = [menu_ref]
model_data_obj = self.pool.get('ir.model.data')
for menu_data in menu_ref:
try:
model, id = model_data_obj.get_object_reference(
cr, uid,
menu_data.split('.')[0],
menu_data.split('.')[1])
if (model == 'ir.ui.menu'):
menu_ids.add(id)
except Exception:
pass
menu_ids = list(menu_ids)
for menu in self.browse(cr, uid, menu_ids, context=context):
res[menu.id] = {
'needaction_enabled': False,
'needaction_counter': False,
}
if menu.action and menu.action.type in (
'ir.actions.act_window',
'ir.actions.client') and menu.action.res_model:
action_ctx = {}
if menu.action.context:
try:
# use magical UnquoteEvalContext to ignore undefined client-side variables such as `active_id`
eval_ctx = tools.UnquoteEvalContext(**context)
action_ctx = eval(menu.action.context,
locals_dict=eval_ctx,
nocopy=True) or {}
except Exception:
# if the eval still fails for some reason, we'll simply skip this menu
raise
if menu.action.res_model in self.pool:
obj = self.pool[menu.action.res_model]
if obj._needaction:
if menu.action.type == 'ir.actions.act_window':
dom = menu.action.domain and eval(
menu.action.domain, {'uid': uid}) or []
else:
dom = eval(menu.action.params_store or '{}', {
'uid': uid
}).get('domain')
full_context = context.copy()
full_context.update(action_ctx)
res[menu.id]['needaction_enabled'] = obj._needaction
res[menu.
id]['needaction_counter'] = obj._needaction_count(
cr, uid, dom, context=full_context)
return res