diff --git a/base_optional_quick_create/model.py b/base_optional_quick_create/model.py index 00862f0c70d..01fbee52790 100644 --- a/base_optional_quick_create/model.py +++ b/base_optional_quick_create/model.py @@ -23,7 +23,7 @@ from openerp.tools.translate import _ -class ir_model(orm.Model): +class IrModel(orm.Model): _inherit = 'ir.model' _columns = { @@ -37,9 +37,7 @@ def wrapper(cr, uid, name, context=None): _("Can't create quickly. Opening create form")) return wrapper - def _register_hook(self, cr, ids=None): - if ids is None: - ids = self.search(cr, SUPERUSER_ID, []) + def _patch_quick_create(self, cr, ids): for model in self.browse(cr, SUPERUSER_ID, ids): if model.avoid_quick_create: model_name = model.model @@ -50,14 +48,18 @@ def _register_hook(self, cr, ids=None): model_obj.check_quick_create = True return True + def _register_hook(self, cr): + self._patch_quick_create(cr, self.search(cr, SUPERUSER_ID, [])) + return super(IrModel, self)._register_hook(cr) + def create(self, cr, uid, vals, context=None): - res_id = super(ir_model, self).create(cr, uid, vals, context=context) - self._register_hook(cr, [res_id]) + res_id = super(IrModel, self).create(cr, uid, vals, context=context) + self._patch_quick_create(cr, [res_id]) return res_id def write(self, cr, uid, ids, vals, context=None): if isinstance(ids, (int, long)): ids = [ids] - res = super(ir_model, self).write(cr, uid, ids, vals, context=context) - self._register_hook(cr, ids) + res = super(IrModel, self).write(cr, uid, ids, vals, context=context) + self._patch_quick_create(cr, ids) return res