initial commit 2

1 files changed, 107 insertions, 0 deletions

diff --git a/addons/mail/models/mail_blacklist.py b/addons/mail/models/mail_blacklist.py
new file mode 100644
index 00000000..d19a3290
--- /dev/null
+++ b/addons/mail/models/mail_blacklist.py
@@ -0,0 +1,107 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from odoo import api, fields, models, tools, _
+from odoo.exceptions import UserError
+
+
+class MailBlackList(models.Model):
+    """ Model of blacklisted email addresses to stop sending emails."""
+    _name = 'mail.blacklist'
+    _inherit = ['mail.thread']
+    _description = 'Mail Blacklist'
+    _rec_name = 'email'
+
+    email = fields.Char(string='Email Address', required=True, index=True, help='This field is case insensitive.',
+                        tracking=True)
+    active = fields.Boolean(default=True, tracking=True)
+
+    _sql_constraints = [
+        ('unique_email', 'unique (email)', 'Email address already exists!')
+    ]
+
+    @api.model_create_multi
+    def create(self, values):
+        # First of all, extract values to ensure emails are really unique (and don't modify values in place)
+        new_values = []
+        all_emails = []
+        for value in values:
+            email = tools.email_normalize(value.get('email'))
+            if not email:
+                raise UserError(_('Invalid email address %r', value['email']))
+            if email in all_emails:
+                continue
+            all_emails.append(email)
+            new_value = dict(value, email=email)
+            new_values.append(new_value)
+
+        """ To avoid crash during import due to unique email, return the existing records if any """
+        sql = '''SELECT email, id FROM mail_blacklist WHERE email = ANY(%s)'''
+        emails = [v['email'] for v in new_values]
+        self._cr.execute(sql, (emails,))
+        bl_entries = dict(self._cr.fetchall())
+        to_create = [v for v in new_values if v['email'] not in bl_entries]
+
+        # TODO DBE Fixme : reorder ids according to incoming ids.
+        results = super(MailBlackList, self).create(to_create)
+        return self.env['mail.blacklist'].browse(bl_entries.values()) | results
+
+    def write(self, values):
+        if 'email' in values:
+            values['email'] = tools.email_normalize(values['email'])
+        return super(MailBlackList, self).write(values)
+
+    def _search(self, args, offset=0, limit=None, order=None, count=False, access_rights_uid=None):
+        """ Override _search in order to grep search on email field and make it
+        lower-case and sanitized """
+        if args:
+            new_args = []
+            for arg in args:
+                if isinstance(arg, (list, tuple)) and arg[0] == 'email' and isinstance(arg[2], str):
+                    normalized = tools.email_normalize(arg[2])
+                    if normalized:
+                        new_args.append([arg[0], arg[1], normalized])
+                    else:
+                        new_args.append(arg)
+                else:
+                    new_args.append(arg)
+        else:
+            new_args = args
+        return super(MailBlackList, self)._search(new_args, offset=offset, limit=limit, order=order, count=count, access_rights_uid=access_rights_uid)
+
+    def _add(self, email):
+        normalized = tools.email_normalize(email)
+        record = self.env["mail.blacklist"].with_context(active_test=False).search([('email', '=', normalized)])
+        if len(record) > 0:
+            record.action_unarchive()
+        else:
+            record = self.create({'email': email})
+        return record
+
+    def action_remove_with_reason(self, email, reason=None):
+        record = self._remove(email)
+        if reason:
+            record.message_post(body=_("Unblacklisting Reason: %s", reason))
+        
+        return record
+
+    def _remove(self, email):
+        normalized = tools.email_normalize(email)
+        record = self.env["mail.blacklist"].with_context(active_test=False).search([('email', '=', normalized)])
+        if len(record) > 0:
+            record.action_archive()
+        else:
+            record = record.create({'email': email, 'active': False})
+        return record
+
+    def mail_action_blacklist_remove(self):
+        return {
+            'name': 'Are you sure you want to unblacklist this Email Address?',
+            'type': 'ir.actions.act_window',
+            'view_mode': 'form',
+            'res_model': 'mail.blacklist.remove',
+            'target': 'new',
+        }
+
+    def action_add(self):
+        self._add(self.email)