|
|
# -*- coding: utf-8 -*-# © 2013 Nicolas Bessi (Camptocamp SA)# © 2014 Agile Business Group (<http://www.agilebg.com>)# © 2015 Grupo ESOC (<http://www.grupoesoc.es>)# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).import loggingfrom openerp import api, fields, modelsfrom .. import exceptions
_logger = logging.getLogger(__name__)
class ResPartner(models.Model): """Adds last name and first name; name becomes a stored function field.""" _inherit = 'res.partner'
firstname = fields.Char("First name") lastname = fields.Char("Last name") name = fields.Char( compute="_compute_name", inverse="_inverse_name_after_cleaning_whitespace", required=False, store=True)
@api.model def create(self, vals): """Add inverted names at creation if unavailable.""" context = dict(self.env.context) name = vals.get("name", context.get("default_name"))
if name is not None: # Calculate the splitted fields inverted = self._get_inverse_name( self._get_whitespace_cleaned_name(name), vals.get("is_company", self.default_get(["is_company"])["is_company"]))
for key, value in inverted.iteritems(): if not vals.get(key) or context.get("copy"): vals[key] = value
# Remove the combined fields if "name" in vals: del vals["name"] if "default_name" in context: del context["default_name"]
return super(ResPartner, self.with_context(context)).create(vals)
@api.multi def copy(self, default=None): """Ensure partners are copied right.
Odoo adds ``(copy)`` to the end of :attr:`~.name`, but that would get ignored in :meth:`~.create` because it also copies explicitly firstname and lastname fields. """
return super(ResPartner, self.with_context(copy=True)).copy(default)
@api.model def default_get(self, fields_list): """Invert name when getting default values.""" result = super(ResPartner, self).default_get(fields_list)
inverted = self._get_inverse_name( self._get_whitespace_cleaned_name(result.get("name", "")), result.get("is_company", False))
for field in inverted.keys(): if field in fields_list: result[field] = inverted.get(field)
return result
@api.model def _get_computed_name(self, lastname, firstname): """Compute the 'name' field according to splitted data.
You can override this method to change the order of lastname and firstname the computed name"""
return u" ".join((p for p in (lastname, firstname) if p))
@api.one @api.depends("firstname", "lastname") def _compute_name(self): """Write the 'name' field according to splitted data.""" self.name = self._get_computed_name(self.lastname, self.firstname)
@api.one def _inverse_name_after_cleaning_whitespace(self): """Clean whitespace in :attr:`~.name` and split it.
The splitting logic is stored separately in :meth:`~._inverse_name`, so submodules can extend that method and get whitespace cleaning for free. """
# Remove unneeded whitespace clean = self._get_whitespace_cleaned_name(self.name)
# Clean name avoiding infinite recursion if self.name != clean: self.name = clean
# Save name in the real fields else: self._inverse_name()
@api.model def _get_whitespace_cleaned_name(self, name): """Remove redundant whitespace from :param:`name`.
Removes leading, trailing and duplicated whitespace. """
return u" ".join(name.split(None)) if name else name
@api.model def _get_inverse_name(self, name, is_company=False): """Compute the inverted name.
- If the partner is a company, save it in the lastname. - Otherwise, make a guess.
This method can be easily overriden by other submodules. You can also override this method to change the order of name's attributes
When this method is called, :attr:`~.name` already has unified and trimmed whitespace. """
# Company name goes to the lastname if is_company or not name: parts = [name or False, False] # Guess name splitting else: parts = name.strip().split(" ", 1) while len(parts) < 2: parts.append(False) return {"lastname": parts[0], "firstname": parts[1]}
@api.one def _inverse_name(self): """Try to revert the effect of :meth:`._compute_name`.""" parts = self._get_inverse_name(self.name, self.is_company) self.lastname, self.firstname = parts["lastname"], parts["firstname"]
@api.one @api.constrains("firstname", "lastname") def _check_name(self): """Ensure at least one name is set.""" if not (self.firstname or self.lastname): raise exceptions.EmptyNamesError(self)
@api.one @api.onchange("firstname", "lastname") def _onchange_subnames(self): """Avoid recursion when the user changes one of these fields.
This forces to skip the :attr:`~.name` inversion when the user is setting it in a not-inverted way. """
# Modify self's context without creating a new Environment. # See https://github.com/odoo/odoo/issues/7472#issuecomment-119503916. self.env.context = self.with_context(skip_onchange=True).env.context
@api.one @api.onchange("name") def _onchange_name(self): """Ensure :attr:`~.name` is inverted in the UI.""" if self.env.context.get("skip_onchange"): # Do not skip next onchange self.env.context = ( self.with_context(skip_onchange=False).env.context) else: self._inverse_name_after_cleaning_whitespace()
@api.model def _install_partner_firstname(self): """Save names correctly in the database.
Before installing the module, field ``name`` contains all full names. When installing it, this method parses those names and saves them correctly into the database. This can be called later too if needed. """
# Find records with empty firstname and lastname records = self.search([("firstname", "=", False), ("lastname", "=", False)])
# Force calculations there records._inverse_name() _logger.info("%d partners updated installing module.", len(records))
# As name is computed after sql_constraint check, we modify it to accept # to write when at least one name field is filled _sql_constraints = [ ('check_name', "CHECK( (type='contact' AND (name IS NOT NULL OR" " firstname IS NOT NULL OR lastname IS NOT NULL)) or" " (type!='contact') )", 'Contacts require a name.'), ]
|
