# Copyright 2019 Komit # License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl). import logging from flectra import _, api, models from flectra.exceptions import UserError, ValidationError _logger = logging.getLogger(__name__) try: from email_validator import ( EmailSyntaxError, EmailUndeliverableError, validate_email, ) except ImportError: _logger.debug('Cannot import "email_validator".') validate_email = None class ResPartner(models.Model): _inherit = "res.partner" def copy_data(self, default=None): res = super(ResPartner, self).copy_data(default=default) if self._should_filter_duplicates(): for copy_vals in res: copy_vals.pop("email", None) return res @api.model def email_check(self, emails): return ",".join( self._normalize_email(email.strip()) for email in emails.split(",") if email.strip() ) @api.constrains("email") def _check_email_unique(self): if self._should_filter_duplicates(): for rec in self.filtered("email"): if "," in rec.email: raise UserError( _( "Field contains multiple email addresses. This is " "not supported when duplicate email addresses are " "not allowed." ) ) if self.search_count([("email", "=", rec.email), ("id", "!=", rec.id)]): raise UserError( _("Email '%s' is already in use.") % rec.email.strip() ) def _normalize_email(self, email): if not self._should_check_syntax(): return email if validate_email is None: _logger.warning( "Can not validate email, " 'python dependency required "email_validator"' ) return email try: result = validate_email( email, check_deliverability=self._should_check_deliverability(), ) except EmailSyntaxError: raise ValidationError(_("%s is an invalid email") % email.strip()) except EmailUndeliverableError: raise ValidationError( _("Cannot deliver to email address %s") % email.strip() ) return result["local"].lower() + "@" + result["domain_i18n"] def _should_check_syntax(self): return self.env.company.partner_email_check_syntax def _should_filter_duplicates(self): return self.env.company.partner_email_check_filter_duplicates def _should_check_deliverability(self): return self.env.company.partner_email_check_check_deliverability @api.model def create(self, vals): if vals.get("email"): vals["email"] = self.email_check(vals["email"]) return super(ResPartner, self).create(vals) def write(self, vals): if vals.get("email"): vals["email"] = self.email_check(vals["email"]) return super(ResPartner, self).write(vals)