|
@ -18,29 +18,42 @@ |
|
|
# |
|
|
# |
|
|
############################################################################## |
|
|
############################################################################## |
|
|
|
|
|
|
|
|
from openerp.osv import orm |
|
|
|
|
|
|
|
|
from openerp.osv import orm, fields |
|
|
|
|
|
from openerp import SUPERUSER_ID |
|
|
|
|
|
|
|
|
class ir_model(orm.Model): |
|
|
class ir_model(orm.Model): |
|
|
|
|
|
|
|
|
_inherit = 'ir.model' |
|
|
_inherit = 'ir.model' |
|
|
|
|
|
|
|
|
|
|
|
_columns = { |
|
|
|
|
|
'avoid_quick_create': fields.boolean('Avoid quick create'), |
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
def _wrap_name_create(self, old_create, model): |
|
|
def _wrap_name_create(self, old_create, model): |
|
|
|
|
|
|
|
|
def wrapper(cr, uid, name, context=None): |
|
|
def wrapper(cr, uid, name, context=None): |
|
|
import pdb; pdb.set_trace() |
|
|
|
|
|
return old_create(cr, uid, name, context=context) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
raise Exception("Can't create quickly. Opening create form") |
|
|
return wrapper |
|
|
return wrapper |
|
|
|
|
|
|
|
|
def _register_hook(self, cr, ids=None): |
|
|
def _register_hook(self, cr, ids=None): |
|
|
model = 'res.partner' |
|
|
|
|
|
model_obj = self.pool.get(model) |
|
|
|
|
|
if not hasattr(model_obj, 'check_quick_create'): |
|
|
|
|
|
model_obj.name_create = self._wrap_name_create(model_obj.name_create, model) |
|
|
|
|
|
model_obj.check_quick_create = True |
|
|
|
|
|
|
|
|
if ids is None: |
|
|
|
|
|
ids = self.search(cr, SUPERUSER_ID, []) |
|
|
|
|
|
for model in self.browse(cr, SUPERUSER_ID, ids): |
|
|
|
|
|
if model.avoid_quick_create: |
|
|
|
|
|
model_name = model.model |
|
|
|
|
|
model_obj = self.pool.get(model_name) |
|
|
|
|
|
if not hasattr(model_obj, 'check_quick_create'): |
|
|
|
|
|
model_obj.name_create = self._wrap_name_create(model_obj.name_create, model_name) |
|
|
|
|
|
model_obj.check_quick_create = True |
|
|
return True |
|
|
return True |
|
|
|
|
|
|
|
|
def name_create(self, cr, uid, name, context=None): |
|
|
|
|
|
res = super(ir_model, self).name_create(cr, uid, name, context=context) |
|
|
|
|
|
self._register_hook(cr, [res]) |
|
|
|
|
|
return res |
|
|
|
|
|
|
|
|
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]) |
|
|
|
|
|
return res_id |
|
|
|
|
|
|
|
|
|
|
|
def write(self, cr, uid, ids, vals, context=None): |
|
|
|
|
|
if isinstance(ids, (int, long)): |
|
|
|
|
|
ids = [ids] |
|
|
|
|
|
super(ir_model, self).write(cr, uid, ids, vals, context=context) |
|
|
|
|
|
self._register_hook(cr, ids) |
|
|
|
|
|
return True |