You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
331 lines
12 KiB
331 lines
12 KiB
# -*- coding: utf-8 -*-
|
|
#
|
|
#
|
|
# Authors: Guewen Baconnier
|
|
# Copyright 2015 Camptocamp SA
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as
|
|
# published by the Free Software Foundation, either version 3 of the
|
|
# License, or (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
#
|
|
|
|
from lxml import etree
|
|
|
|
from openerp import models, fields, api, exceptions, _
|
|
from openerp.osv.orm import setup_modifiers
|
|
|
|
|
|
class ResPartnerRevision(models.Model):
|
|
_name = 'res.partner.revision'
|
|
_description = 'Partner Revision'
|
|
_order = 'date desc'
|
|
_rec_name = 'date'
|
|
|
|
partner_id = fields.Many2one(comodel_name='res.partner',
|
|
string='Partner',
|
|
required=True)
|
|
change_ids = fields.One2many(comodel_name='res.partner.revision.change',
|
|
inverse_name='revision_id',
|
|
string='Changes')
|
|
date = fields.Datetime(default=fields.Datetime.now)
|
|
# TODO: add a revision state, done when all lines are done or
|
|
# canceled
|
|
note = fields.Text()
|
|
|
|
@api.multi
|
|
def apply(self):
|
|
self.mapped('change_ids').apply()
|
|
|
|
@api.multi
|
|
def cancel(self):
|
|
self.mapped('change_ids').cancel()
|
|
|
|
@api.multi
|
|
def add_revision(self, record, values):
|
|
""" Add a revision on a partner
|
|
|
|
By default, when a partner is modified by a user or by the
|
|
system, the changes are applied and a validated revision is
|
|
created. Callers which want to delegate the write of some
|
|
fields to the revision must explicitly ask for it by providing a
|
|
key ``__revision_rules`` in the environment's context.
|
|
|
|
Should be called before the execution of ``write`` on the record
|
|
so we can keep track of the existing value and also because the
|
|
returned values should be used for ``write`` as some of the
|
|
values may have been removed.
|
|
|
|
:param values: the values being written on the partner
|
|
:type values: dict
|
|
|
|
:returns: dict of values that should be wrote on the partner
|
|
(fields with a 'Validate' or 'Never' rule are excluded)
|
|
|
|
"""
|
|
record.ensure_one()
|
|
change_model = self.env['res.partner.revision.change']
|
|
write_values = values.copy()
|
|
changes = []
|
|
rules = self.env['revision.behavior'].get_rules(record._model._name)
|
|
for field in values:
|
|
rule = rules.get(field)
|
|
if not rule:
|
|
continue
|
|
if field in values:
|
|
if not change_model._has_field_changed(record, field,
|
|
values[field]):
|
|
continue
|
|
change, pop_value = change_model._prepare_revision_change(
|
|
record, rule, field, values[field]
|
|
)
|
|
if pop_value:
|
|
write_values.pop(field)
|
|
changes.append(change)
|
|
if changes:
|
|
self.env['res.partner.revision'].create({
|
|
'partner_id': record.id,
|
|
'change_ids': [(0, 0, vals) for vals in changes],
|
|
'date': fields.Datetime.now(),
|
|
})
|
|
return write_values
|
|
|
|
|
|
class ResPartnerRevisionChange(models.Model):
|
|
_name = 'res.partner.revision.change'
|
|
_description = 'Partner Revision Change'
|
|
_rec_name = 'field_id'
|
|
|
|
revision_id = fields.Many2one(comodel_name='res.partner.revision',
|
|
required=True,
|
|
string='Revision',
|
|
ondelete='cascade')
|
|
field_id = fields.Many2one(comodel_name='ir.model.fields',
|
|
string='Field',
|
|
required=True)
|
|
field_type = fields.Selection(related='field_id.ttype',
|
|
string='Field Type',
|
|
readonly=True)
|
|
|
|
current_value_display = fields.Char(
|
|
string='Current',
|
|
compute='_compute_value_display',
|
|
)
|
|
new_value_display = fields.Char(
|
|
string='New',
|
|
compute='_compute_value_display',
|
|
)
|
|
|
|
current_value_char = fields.Char(string='Current')
|
|
current_value_date = fields.Date(string='Current')
|
|
current_value_datetime = fields.Datetime(string='Current')
|
|
current_value_float = fields.Float(string='Current')
|
|
current_value_integer = fields.Integer(string='Current')
|
|
current_value_text = fields.Text(string='Current')
|
|
current_value_boolean = fields.Boolean(string='Current')
|
|
current_value_reference = fields.Reference(string='Current',
|
|
selection='_reference_models')
|
|
|
|
new_value_char = fields.Char(string='New')
|
|
new_value_date = fields.Date(string='New')
|
|
new_value_datetime = fields.Datetime(string='New')
|
|
new_value_float = fields.Float(string='New')
|
|
new_value_integer = fields.Integer(string='New')
|
|
new_value_text = fields.Text(string='New')
|
|
new_value_boolean = fields.Boolean(string='New')
|
|
new_value_reference = fields.Reference(string='New',
|
|
selection='_reference_models')
|
|
|
|
state = fields.Selection(
|
|
selection=[('draft', 'Waiting'),
|
|
('done', 'Accepted'),
|
|
('cancel', 'Refused'),
|
|
],
|
|
required=True,
|
|
default='draft',
|
|
)
|
|
|
|
@api.model
|
|
def _reference_models(self):
|
|
models = self.env['ir.model'].search([])
|
|
return [(model.model, model.name) for model in models]
|
|
|
|
_suffix_to_types = {
|
|
'char': ('char', 'selection'),
|
|
'date': ('date',),
|
|
'datetime': ('datetime',),
|
|
'float': ('float',),
|
|
'integer': ('integer',),
|
|
'text': ('text',),
|
|
'boolean': ('boolean',),
|
|
'reference': ('many2one',),
|
|
}
|
|
|
|
_type_to_suffix = {ftype: suffix
|
|
for suffix, ftypes in _suffix_to_types.iteritems()
|
|
for ftype in ftypes}
|
|
|
|
_current_value_fields = ['current_value_%s' % suffix
|
|
for suffix in _suffix_to_types]
|
|
_new_value_fields = ['new_value_%s' % suffix
|
|
for suffix in _suffix_to_types]
|
|
_value_fields = _current_value_fields + _new_value_fields
|
|
|
|
@api.one
|
|
@api.depends(lambda self: self._value_fields)
|
|
def _compute_value_display(self):
|
|
for prefix in ('current', 'new'):
|
|
value = getattr(self, 'get_%s_value' % prefix)()
|
|
if self.field_id.ttype == 'many2one' and value:
|
|
value = value.display_name
|
|
setattr(self, '%s_value_display' % prefix, value)
|
|
|
|
@api.model
|
|
def create(self, vals):
|
|
vals = vals.copy()
|
|
field = self.env['ir.model.fields'].browse(vals.get('field_id'))
|
|
if 'current_value' in vals:
|
|
current_value = vals.pop('current_value')
|
|
if field:
|
|
current_field_name = self.get_field_for_type(field, 'current')
|
|
vals[current_field_name] = current_value
|
|
if 'new_value' in vals:
|
|
new_value = vals.pop('new_value')
|
|
if field:
|
|
new_field_name = self.get_field_for_type(field, 'new')
|
|
vals[new_field_name] = new_value
|
|
return super(ResPartnerRevisionChange, self).create(vals)
|
|
|
|
@api.model
|
|
def get_field_for_type(self, field, current_or_new):
|
|
assert current_or_new in ('new', 'current')
|
|
field_type = self._type_to_suffix.get(field.ttype)
|
|
if not field_type:
|
|
raise NotImplementedError(
|
|
'field type %s is not supported' % field_type
|
|
)
|
|
return '%s_value_%s' % (current_or_new, field_type)
|
|
|
|
@api.multi
|
|
def get_current_value(self):
|
|
self.ensure_one()
|
|
field_name = self.get_field_for_type(self.field_id, 'current')
|
|
return self[field_name]
|
|
|
|
@api.multi
|
|
def get_new_value(self):
|
|
self.ensure_one()
|
|
field_name = self.get_field_for_type(self.field_id, 'new')
|
|
return self[field_name]
|
|
|
|
@api.multi
|
|
def apply(self):
|
|
# TODO: optimize with 1 write for all fields, group by revision
|
|
for change in self:
|
|
if change.state in ('cancel', 'done'):
|
|
continue
|
|
partner = change.revision_id.partner_id
|
|
value_for_write = change._convert_value_for_write(
|
|
change.get_new_value()
|
|
)
|
|
partner.write({change.field_id.name: value_for_write})
|
|
change.write({'state': 'done'})
|
|
|
|
@api.multi
|
|
def cancel(self):
|
|
if any(change.state == 'done' for change in self):
|
|
raise exceptions.Warning(
|
|
_('This change has already be applied.')
|
|
)
|
|
self.write({'state': 'cancel'})
|
|
|
|
@api.model
|
|
def _has_field_changed(self, record, field, value):
|
|
field_def = record._fields[field]
|
|
return field_def.convert_to_write(record[field]) != value
|
|
|
|
@api.multi
|
|
def _convert_value_for_write(self, value):
|
|
model = self.env[self.field_id.model_id.model]
|
|
model_field_def = model._fields[self.field_id.name]
|
|
return model_field_def.convert_to_write(value)
|
|
|
|
@api.model
|
|
def _convert_value_for_revision(self, record, field, value):
|
|
field_def = record._fields[field]
|
|
if field_def.type == 'many2one':
|
|
# store as 'reference'
|
|
comodel = field_def.comodel_name
|
|
return "%s,%s" % (comodel, value) if value else False
|
|
else:
|
|
return value
|
|
|
|
@api.multi
|
|
def _prepare_revision_change(self, record, rule, field, value):
|
|
""" Prepare data for a revision change
|
|
|
|
It returns a dict of the values to write on the revision change
|
|
and a boolean that indicates if the value should be popped out
|
|
of the values to write on the model.
|
|
|
|
:returns: dict of values, boolean
|
|
"""
|
|
field_def = record._fields[field]
|
|
# get a ready to write value for the type of the field,
|
|
# for instance takes '.id' from a many2one's record (the
|
|
# new value is already a value as expected for the
|
|
# write)
|
|
current_value = field_def.convert_to_write(record[field])
|
|
# get values ready to write as expected by the revision
|
|
# (for instance, a many2one is written in a reference
|
|
# field)
|
|
current_value = self._convert_value_for_revision(record, field,
|
|
current_value)
|
|
new_value = self._convert_value_for_revision(record, field, value)
|
|
change = {
|
|
'current_value': current_value,
|
|
'new_value': new_value,
|
|
'field_id': rule.field_id.id,
|
|
}
|
|
pop_value = False
|
|
if not self.env.context.get('__revision_rules'):
|
|
# by default always write on partner
|
|
change['state'] = 'done'
|
|
elif rule.default_behavior == 'auto':
|
|
change['state'] = 'done'
|
|
elif rule.default_behavior == 'validate':
|
|
change['state'] = 'draft'
|
|
pop_value = True # change to apply manually
|
|
elif rule.default_behavior == 'never':
|
|
change['state'] = 'cancel'
|
|
pop_value = True # change never applied
|
|
return change, pop_value
|
|
|
|
def fields_view_get(self, *args, **kwargs):
|
|
_super = super(ResPartnerRevisionChange, self)
|
|
result = _super.fields_view_get(*args, **kwargs)
|
|
if result['type'] != 'form':
|
|
return
|
|
doc = etree.XML(result['arch'])
|
|
for suffix, ftypes in self._suffix_to_types.iteritems():
|
|
for prefix in ('current', 'new'):
|
|
field_name = '%s_value_%s' % (prefix, suffix)
|
|
field_nodes = doc.xpath("//field[@name='%s']" % field_name)
|
|
for node in field_nodes:
|
|
node.set(
|
|
'attrs',
|
|
"{'invisible': "
|
|
"[('field_type', 'not in', %s)]}" % (ftypes,)
|
|
)
|
|
setup_modifiers(node)
|
|
result['arch'] = etree.tostring(doc)
|
|
return result
|