diff options
Diffstat (limited to 'addons/base_sparse_field/models/models.py')
| -rw-r--r-- | addons/base_sparse_field/models/models.py | 99 |
1 files changed, 99 insertions, 0 deletions
diff --git a/addons/base_sparse_field/models/models.py b/addons/base_sparse_field/models/models.py new file mode 100644 index 00000000..179f6fbd --- /dev/null +++ b/addons/base_sparse_field/models/models.py @@ -0,0 +1,99 @@ +# -*- coding: utf-8 -*- + +from collections import defaultdict + +from odoo import models, fields, api, _ +from odoo.exceptions import UserError + + +class Base(models.AbstractModel): + _inherit = 'base' + + def _valid_field_parameter(self, field, name): + return name == 'sparse' or super()._valid_field_parameter(field, name) + + +class IrModelFields(models.Model): + _inherit = 'ir.model.fields' + + ttype = fields.Selection(selection_add=[ + ('serialized', 'serialized'), + ], ondelete={'serialized': 'cascade'}) + serialization_field_id = fields.Many2one('ir.model.fields', string='Serialization Field', + ondelete='cascade', domain="[('ttype','=','serialized'), ('model_id', '=', model_id)]", + help="If set, this field will be stored in the sparse structure of the " + "serialization field, instead of having its own database column. " + "This cannot be changed after creation.", + ) + + def write(self, vals): + # Limitation: renaming a sparse field or changing the storing system is + # currently not allowed + if 'serialization_field_id' in vals or 'name' in vals: + for field in self: + if 'serialization_field_id' in vals and field.serialization_field_id.id != vals['serialization_field_id']: + raise UserError(_('Changing the storing system for field "%s" is not allowed.', field.name)) + if field.serialization_field_id and (field.name != vals['name']): + raise UserError(_('Renaming sparse field "%s" is not allowed', field.name)) + + return super(IrModelFields, self).write(vals) + + def _reflect_fields(self, model_names): + super()._reflect_fields(model_names) + + # set 'serialization_field_id' on sparse fields; it is done here to + # ensure that the serialized field is reflected already + cr = self._cr + + # retrieve existing values + query = """ + SELECT model, name, id, serialization_field_id + FROM ir_model_fields + WHERE model IN %s + """ + cr.execute(query, [tuple(model_names)]) + existing = {row[:2]: row[2:] for row in cr.fetchall()} + + # determine updates, grouped by value + updates = defaultdict(list) + for model_name in model_names: + for field_name, field in self.env[model_name]._fields.items(): + field_id, current_value = existing[(model_name, field_name)] + try: + value = existing[(model_name, field.sparse)][0] if field.sparse else None + except KeyError: + msg = _("Serialization field %r not found for sparse field %s!") + raise UserError(msg % (field.sparse, field)) + if current_value != value: + updates[value].append(field_id) + + if not updates: + return + + # update fields + query = "UPDATE ir_model_fields SET serialization_field_id=%s WHERE id IN %s" + for value, ids in updates.items(): + cr.execute(query, [value, tuple(ids)]) + + records = self.browse(id_ for ids in updates.values() for id_ in ids) + self.pool.post_init(records.modified, ['serialization_field_id']) + + def _instanciate_attrs(self, field_data): + attrs = super(IrModelFields, self)._instanciate_attrs(field_data) + if attrs and field_data.get('serialization_field_id'): + serialization_record = self.browse(field_data['serialization_field_id']) + attrs['sparse'] = serialization_record.name + return attrs + + +class TestSparse(models.TransientModel): + _name = 'sparse_fields.test' + _description = 'Sparse fields Test' + + data = fields.Serialized() + boolean = fields.Boolean(sparse='data') + integer = fields.Integer(sparse='data') + float = fields.Float(sparse='data') + char = fields.Char(sparse='data') + selection = fields.Selection([('one', 'One'), ('two', 'Two')], sparse='data') + partner = fields.Many2one('res.partner', sparse='data') |
