partner-contact/base_partner_merge/base_partner_merge.py

#!/usr/bin/env python
from __future__ import absolute_import
from email.utils import parseaddr
import functools
import htmlentitydefs
import itertools
import logging
import operator
import re
from ast import literal_eval
from openerp.tools import mute_logger

# Validation Library https://pypi.python.org/pypi/validate_email/1.1
from .validate_email import validate_email

import openerp
from openerp.osv import osv, orm
from openerp.osv import fields
from openerp.osv.orm import browse_record
from openerp.tools.translate import _

pattern = re.compile("&(\w+?);")

_logger = logging.getLogger('base.partner.merge')


# http://www.php2python.com/wiki/function.html-entity-decode/
def html_entity_decode_char(m, defs=htmlentitydefs.entitydefs):
    try:
        return defs[m.group(1)]
    except KeyError:
        return m.group(0)


def html_entity_decode(string):
    return pattern.sub(html_entity_decode_char, string)


def sanitize_email(partner_email):
    assert isinstance(partner_email, basestring) and partner_email

    result = re.subn(r';|/|:', ',',
                     html_entity_decode(partner_email or ''))[0].split(',')

    emails = [parseaddr(email)[1]
              for item in result
              for email in item.split()]

    return [email.lower()
            for email in emails
            if validate_email(email)]


def is_integer_list(ids):
    return all(isinstance(i, (int, long)) for i in ids)


class ResPartner(osv.Model):
    _inherit = 'res.partner'

    _columns = {
        'id': fields.integer('Id', readonly=True),
        'create_date': fields.datetime('Create Date', readonly=True),
    }


class MergePartnerLine(osv.TransientModel):
    _name = 'base.partner.merge.line'

    _columns = {
        'wizard_id': fields.many2one('base.partner.merge.automatic.wizard',
                                     'Wizard'),
        'min_id': fields.integer('MinID'),
        'aggr_ids': fields.char('Ids', required=True),
    }

    _order = 'min_id asc'


class MergePartnerAutomatic(osv.TransientModel):
    """
    The idea behind this wizard is to create a list of potential partners to
    merge. We use two objects, the first one is the wizard for the end-user.
    And the second will contain the partner list to merge.

    """
    _name = 'base.partner.merge.automatic.wizard'

    _columns = {
        # Group by
        'group_by_email': fields.boolean('Email'),
        'group_by_name': fields.boolean('Name'),
        'group_by_is_company': fields.boolean('Is Company'),
        'group_by_vat': fields.boolean('VAT'),
        'group_by_parent_id': fields.boolean('Parent Company'),

        'state': fields.selection([('option', 'Option'),
                                   ('selection', 'Selection'),
                                   ('finished', 'Finished')],
                                  'State',
                                  readonly=True,
                                  required=True),
        'number_group': fields.integer("Group of Contacts", readonly=True),
        'current_line_id': fields.many2one('base.partner.merge.line',
                                           'Current Line'),
        'line_ids': fields.one2many('base.partner.merge.line',
                                    'wizard_id', 'Lines'),
        'partner_ids': fields.many2many('res.partner', string='Contacts'),
        'dst_partner_id': fields.many2one('res.partner',
                                          string='Destination Contact'),

        'exclude_contact': fields.boolean('A user associated to the contact'),
        'exclude_journal_item': fields.boolean('Journal Items associated'
                                               ' to the contact'),
        'maximum_group': fields.integer("Maximum of Group of Contacts"),
    }

    def default_get(self, cr, uid, fields, context=None):
        if context is None:
            context = {}
        res = super(MergePartnerAutomatic, self
                    ).default_get(cr, uid, fields, context)
        if (context.get('active_model') == 'res.partner'
                and context.get('active_ids')):
            partner_ids = context['active_ids']
            res['state'] = 'selection'
            res['partner_ids'] = partner_ids
            res['dst_partner_id'] = self._get_ordered_partner(cr, uid,
                                                              partner_ids,
                                                              context=context
                                                              )[-1].id
        return res

    _defaults = {
        'state': 'option'
    }

    def get_fk_on(self, cr, table):
        q = """  SELECT cl1.relname as table,
                        att1.attname as column
                   FROM pg_constraint as con, pg_class as cl1, pg_class as cl2,
                        pg_attribute as att1, pg_attribute as att2
                  WHERE con.conrelid = cl1.oid
                    AND con.confrelid = cl2.oid
                    AND array_lower(con.conkey, 1) = 1
                    AND con.conkey[1] = att1.attnum
                    AND att1.attrelid = cl1.oid
                    AND cl2.relname = %s
                    AND att2.attname = 'id'
                    AND array_lower(con.confkey, 1) = 1
                    AND con.confkey[1] = att2.attnum
                    AND att2.attrelid = cl2.oid
                    AND con.contype = 'f'
        """
        return cr.execute(q, (table,))

    def _update_foreign_keys(self, cr, uid, src_partners,
                             dst_partner, context=None):
        _logger.debug('_update_foreign_keys for dst_partner: %s for '
                      'src_partners: %r',
                      dst_partner.id,
                      list(map(operator.attrgetter('id'), src_partners)))

        # find the many2one relation to a partner
        proxy = self.pool.get('res.partner')
        self.get_fk_on(cr, 'res_partner')

        # ignore two tables

        for table, column in cr.fetchall():
            if 'base_partner_merge_' in table:
                continue
            partner_ids = tuple(map(int, src_partners))

            query = ("SELECT column_name FROM information_schema.columns"
                     "  WHERE table_name LIKE '%s'") % (table)
            cr.execute(query, ())
            columns = []
            for data in cr.fetchall():
                if data[0] != column:
                    columns.append(data[0])

            query_dic = {
                'table': table,
                'column': column,
                'value': columns[0],
            }
            if len(columns) <= 1:
                # unique key treated
                query = """
                    UPDATE "%(table)s" as ___tu
                    SET %(column)s = %%s
                    WHERE
                        %(column)s = %%s AND
                        NOT EXISTS (
                            SELECT 1
                            FROM "%(table)s" as ___tw
                            WHERE
                                %(column)s = %%s AND
                                ___tu.%(value)s = ___tw.%(value)s
                        )""" % query_dic
                for partner_id in partner_ids:
                    cr.execute(query, (dst_partner.id, partner_id,
                                       dst_partner.id))
            else:
                cr.execute("SAVEPOINT recursive_partner_savepoint")
                try:
                    query = ('UPDATE "%(table)s" SET %(column)s = %%s WHERE '
                             '%(column)s IN %%s') % query_dic
                    cr.execute(query, (dst_partner.id, partner_ids,))

                    if (column == proxy._parent_name
                            and table == 'res_partner'):
                        query = """
                            WITH RECURSIVE cycle(id, parent_id) AS (
                                    SELECT id, parent_id FROM res_partner
                                UNION
                                    SELECT cycle.id, res_partner.parent_id
                                    FROM   res_partner, cycle
                                    WHERE  res_partner.id = cycle.parent_id
                                      AND  cycle.id != cycle.parent_id
                            )
                            SELECT id FROM cycle
                                WHERE id = parent_id AND id = %s
                        """
                        cr.execute(query, (dst_partner.id,))
                        if cr.fetchall():
                            cr.execute("ROLLBACK TO SAVEPOINT "
                                       "recursive_partner_savepoint")
                finally:
                    cr.execute("RELEASE SAVEPOINT "
                               "recursive_partner_savepoint")

    def _update_reference_fields(self, cr, uid, src_partners, dst_partner,
                                 context=None):
        _logger.debug('_update_reference_fields for dst_partner: %s for '
                      'src_partners: %r',
                      dst_partner.id,
                      list(map(operator.attrgetter('id'), src_partners)))

        def update_records(model, src, field_model='model', field_id='res_id',
                           context=None):
            proxy = self.pool.get(model)
            if proxy is None:
                return
            domain = [(field_model, '=', 'res.partner'),
                      (field_id, '=', src.id)]
            ids = proxy.search(cr, openerp.SUPERUSER_ID,
                               domain, context=context)
            return proxy.write(cr, openerp.SUPERUSER_ID, ids,
                               {field_id: dst_partner.id}, context=context)

        update_records = functools.partial(update_records, context=context)

        for partner in src_partners:
            update_records('base.calendar', src=partner,
                           field_model='model_id.model')
            update_records('ir.attachment', src=partner,
                           field_model='res_model')
            update_records('mail.followers', src=partner,
                           field_model='res_model')
            update_records('mail.message', src=partner)
            update_records('marketing.campaign.workitem', src=partner,
                           field_model='object_id.model')
            update_records('ir.model.data', src=partner)

        proxy = self.pool['ir.model.fields']
        domain = [('ttype', '=', 'reference')]
        record_ids = proxy.search(cr, openerp.SUPERUSER_ID, domain,
                                  context=context)

        for record in proxy.browse(cr, openerp.SUPERUSER_ID, record_ids,
                                   context=context):
            try:
                proxy_model = self.pool[record.model]
            except KeyError:
                # ignore old tables
                continue

            if record.model == 'ir.property':
                continue

            field_type = proxy_model._columns.get(record.name).__class__._type

            if field_type == 'function':
                continue

            for partner in src_partners:
                domain = [
                    (record.name, '=', 'res.partner,%d' % partner.id)
                ]
                model_ids = proxy_model.search(cr, openerp.SUPERUSER_ID,
                                               domain, context=context)
                values = {
                    record.name: 'res.partner,%d' % dst_partner.id,
                }
                proxy_model.write(cr, openerp.SUPERUSER_ID, model_ids, values,
                                  context=context)

    def _update_values(self, cr, uid, src_partners, dst_partner, context=None):
        _logger.debug('_update_values for dst_partner: %s for src_partners: '
                      '%r',
                      dst_partner.id,
                      list(map(operator.attrgetter('id'), src_partners)))

        columns = dst_partner._columns

        def write_serializer(column, item):
            if isinstance(item, browse_record):
                return item.id
            else:
                return item

        values = dict()
        for column, field in columns.iteritems():
            if (field._type not in ('many2many', 'one2many')
                    and not isinstance(field, fields.function)):
                for item in itertools.chain(src_partners, [dst_partner]):
                    if item[column]:
                        values[column] = write_serializer(column,
                                                          item[column])

        values.pop('id', None)
        parent_id = values.pop('parent_id', None)
        dst_partner.write(values)
        if parent_id and parent_id != dst_partner.id:
            try:
                dst_partner.write({'parent_id': parent_id})
            except (osv.except_osv, orm.except_orm):
                _logger.info('Skip recursive partner hierarchies for '
                             'parent_id %s of partner: %s',
                             parent_id, dst_partner.id)

    @mute_logger('openerp.osv.expression', 'openerp.osv.orm')
    def _merge(self, cr, uid, partner_ids, dst_partner=None, context=None):
        proxy = self.pool.get('res.partner')

        partner_ids = proxy.exists(cr, uid, list(partner_ids),
                                   context=context)
        if len(partner_ids) < 2:
            return

        if len(partner_ids) > 3:
            raise osv.except_osv(
                _('Error'),
                _("For safety reasons, you cannot merge more than 3 contacts "
                  "together. You can re-open the wizard several times if "
                  "needed."))

        if (openerp.SUPERUSER_ID != uid
                and len(set(partner.email for partner
                            in proxy.browse(cr, uid, partner_ids,
                                            context=context))) > 1):
            raise osv.except_osv(
                _('Error'),
                _("All contacts must have the same email. Only the "
                  "Administrator can merge contacts with different emails."))

        if dst_partner and dst_partner.id in partner_ids:
            src_partners = proxy.browse(cr, uid,
                                        [id for id in partner_ids
                                         if id != dst_partner.id],
                                        context=context)
        else:
            ordered_partners = self._get_ordered_partner(cr, uid, partner_ids,
                                                         context)
            dst_partner = ordered_partners[-1]
            src_partners = ordered_partners[:-1]
        _logger.info("dst_partner: %s", dst_partner.id)

        if (openerp.SUPERUSER_ID != uid
                and self._model_is_installed(cr, uid, 'account.move.line',
                                             context=context)
                and self.pool.get('account.move.line'
                                  ).search(cr, openerp.SUPERUSER_ID,
                                           [('partner_id',
                                             'in',
                                             [partner.id for partner
                                              in src_partners])],
                                           context=context)):
            raise osv.except_osv(
                _('Error'),
                _("Only the destination contact may be linked to existing "
                  "Journal Items. Please ask the Administrator if you need to"
                  " merge several contacts linked to existing Journal "
                  "Items."))

        call_it = lambda function: function(cr, uid, src_partners,
                                            dst_partner, context=context)

        call_it(self._update_foreign_keys)
        call_it(self._update_reference_fields)
        call_it(self._update_values)

        _logger.info('(uid = %s) merged the partners %r with %s',
                     uid,
                     list(map(operator.attrgetter('id'), src_partners)),
                     dst_partner.id)
        dst_partner.message_post(
            body='%s %s' % (
                _("Merged with the following partners:"),
                ", ".join('%s<%s>(ID %s)' % (p.name, p.email or 'n/a', p.id)
                                             for p in src_partners)))

        for partner in src_partners:
            partner.unlink()

    def clean_emails(self, cr, uid, context=None):
        """
        Clean the email address of the partner, if there is an email field
        with a minimum of two addresses, the system will create a new partner,
        with the information of the previous one and will copy the new cleaned
        email into the email field.
        """
        if context is None:
            context = {}

        proxy_model = self.pool['ir.model.fields']
        field_ids = proxy_model.search(cr, uid,
                                       [('model', '=', 'res.partner'),
                                        ('ttype', 'like', '%2many')],
                                       context=context)
        fields = proxy_model.read(cr, uid, field_ids, context=context)
        reset_fields = dict((field['name'], []) for field in fields)

        proxy_partner = self.pool['res.partner']
        context['active_test'] = False
        ids = proxy_partner.search(cr, uid, [], context=context)

        fields = ['name', 'var' 'partner_id' 'is_company', 'email']
        partners = proxy_partner.read(cr, uid, ids, fields, context=context)

        partners.sort(key=operator.itemgetter('id'))
        partners_len = len(partners)

        _logger.info('partner_len: %r', partners_len)

        for idx, partner in enumerate(partners):
            if not partner['email']:
                continue

            percent = (idx / float(partners_len)) * 100.0
            _logger.info('idx: %r', idx)
            _logger.info('percent: %r', percent)
            try:
                emails = sanitize_email(partner['email'])
                head, tail = emails[:1], emails[1:]
                email = head[0] if head else False

                proxy_partner.write(cr, uid, [partner['id']],
                                    {'email': email}, context=context)

                for email in tail:
                    values = dict(reset_fields, email=email)
                    proxy_partner.copy(cr, uid, partner['id'], values,
                                       context=context)

            except Exception:
                _logger.exception("There is a problem with this partner: %r",
                                  partner)
                raise
        return True

    def close_cb(self, cr, uid, ids, context=None):
        return {'type': 'ir.actions.act_window_close'}

    def _generate_query(self, fields, maximum_group=100):
        group_fields = ', '.join(fields)

        filters = []
        for field in fields:
            if field in ['email', 'name']:
                filters.append((field, 'IS NOT', 'NULL'))

        criteria = ' AND '.join('%s %s %s' % (field, operator, value)
                                for field, operator, value in filters)

        text = [
            "SELECT min(id), array_agg(id)",
            "FROM res_partner",
        ]

        if criteria:
            text.append('WHERE %s' % criteria)

        text.extend([
            "GROUP BY %s" % group_fields,
            "HAVING COUNT(*) >= 2",
            "ORDER BY min(id)",
        ])

        if maximum_group:
            text.extend([
                "LIMIT %s" % maximum_group,
            ])

        return ' '.join(text)

    def _compute_selected_groupby(self, this):
        group_by_str = 'group_by_'
        group_by_len = len(group_by_str)

        fields = [
            key[group_by_len:]
            for key in self._columns.keys()
            if key.startswith(group_by_str)
        ]

        groups = [
            field
            for field in fields
            if getattr(this, '%s%s' % (group_by_str, field), False)
        ]

        if not groups:
            raise osv.except_osv(_('Error'),
                                 _("You have to specify a filter for your "
                                   "selection"))

        return groups

    def next_cb(self, cr, uid, ids, context=None):
        """
        Don't compute any thing
        """
        context = dict(context or {}, active_test=False)
        this = self.browse(cr, uid, ids[0], context=context)
        if this.current_line_id:
            this.current_line_id.unlink()
        return self._next_screen(cr, uid, this, context)

    def _get_ordered_partner(self, cr, uid, partner_ids, context=None):
        partners = self.pool.get('res.partner'
                                 ).browse(cr, uid,
                                          list(partner_ids),
                                          context=context)
        ordered_partners = sorted(sorted(partners,
                                         key=operator.attrgetter('create_date'),
                                         reverse=True),
                                  key=operator.attrgetter('active'),
                                  reverse=True)
        return ordered_partners

    def _next_screen(self, cr, uid, this, context=None):
        this.refresh()
        values = {}
        if this.line_ids:
            # in this case, we try to find the next record.
            current_line = this.line_ids[0]
            current_partner_ids = literal_eval(current_line.aggr_ids)
            values.update({
                'current_line_id': current_line.id,
                'partner_ids': [(6, 0, current_partner_ids)],
                'dst_partner_id': self._get_ordered_partner(
                    cr, uid,
                    current_partner_ids,
                    context
                    )[-1].id,
                'state': 'selection',
            })
        else:
            values.update({
                'current_line_id': False,
                'partner_ids': [],
                'state': 'finished',
            })

        this.write(values)

        return {
            'type': 'ir.actions.act_window',
            'res_model': this._name,
            'res_id': this.id,
            'view_mode': 'form',
            'target': 'new',
        }

    def _model_is_installed(self, cr, uid, model, context=None):
        proxy = self.pool.get('ir.model')
        domain = [('model', '=', model)]
        return proxy.search_count(cr, uid, domain, context=context) > 0

    def _partner_use_in(self, cr, uid, aggr_ids, models, context=None):
        """
        Check if there is no occurence of this group of partner in the selected
        model
        """
        for model, field in models.iteritems():
            proxy = self.pool.get(model)
            domain = [(field, 'in', aggr_ids)]
            if proxy.search_count(cr, uid, domain, context=context):
                return True
        return False

    def compute_models(self, cr, uid, ids, context=None):
        """
        Compute the different models needed by the system if you want to
        exclude some partners.
        """
        assert is_integer_list(ids)

        this = self.browse(cr, uid, ids[0], context=context)

        models = {}
        if this.exclude_contact:
            models['res.users'] = 'partner_id'

        if (self._model_is_installed(cr, uid, 'account.move.line',
                                     context=context)
                and this.exclude_journal_item):
            models['account.move.line'] = 'partner_id'

        return models

    def _process_query(self, cr, uid, ids, query, context=None):
        """
        Execute the select request and write the result in this wizard
        """
        proxy = self.pool.get('base.partner.merge.line')
        this = self.browse(cr, uid, ids[0], context=context)
        models = self.compute_models(cr, uid, ids, context=context)
        cr.execute(query)

        counter = 0
        for min_id, aggr_ids in cr.fetchall():
            if models and self._partner_use_in(cr, uid, aggr_ids, models,
                                               context=context):
                continue
            values = {
                'wizard_id': this.id,
                'min_id': min_id,
                'aggr_ids': aggr_ids,
            }

            proxy.create(cr, uid, values, context=context)
            counter += 1

        values = {
            'state': 'selection',
            'number_group': counter,
        }

        this.write(values)

        _logger.info("counter: %s", counter)

    def start_process_cb(self, cr, uid, ids, context=None):
        """
        Start the process.
        * Compute the selected groups (with duplication)
        * If the user has selected the 'exclude_XXX' fields, avoid the
          partners.
        """
        assert is_integer_list(ids)

        context = dict(context or {}, active_test=False)
        this = self.browse(cr, uid, ids[0], context=context)
        groups = self._compute_selected_groupby(this)
        query = self._generate_query(groups, this.maximum_group)
        self._process_query(cr, uid, ids, query, context=context)

        return self._next_screen(cr, uid, this, context)

    def automatic_process_cb(self, cr, uid, ids, context=None):
        assert is_integer_list(ids)
        this = self.browse(cr, uid, ids[0], context=context)
        this.start_process_cb()
        this.refresh()

        for line in this.line_ids:
            partner_ids = literal_eval(line.aggr_ids)
            self._merge(cr, uid, partner_ids, context=context)
            line.unlink()
            cr.commit()

        this.write({'state': 'finished'})
        return {
            'type': 'ir.actions.act_window',
            'res_model': this._name,
            'res_id': this.id,
            'view_mode': 'form',
            'target': 'new',
        }

    def parent_migration_process_cb(self, cr, uid, ids, context=None):
        assert is_integer_list(ids)

        context = dict(context or {}, active_test=False)
        this = self.browse(cr, uid, ids[0], context=context)

        query = """
            SELECT
                min(p1.id),
                array_agg(DISTINCT p1.id)
            FROM
                res_partner as p1
            INNER join
                res_partner as p2
            ON
                p1.email = p2.email AND
                p1.name = p2.name AND
                (p1.parent_id = p2.id OR p1.id = p2.parent_id)
            WHERE
                p2.id IS NOT NULL
            GROUP BY
                p1.email,
                p1.name,
                CASE WHEN p1.parent_id = p2.id THEN p2.id
                    ELSE p1.id
                END
            HAVING COUNT(*) >= 2
            ORDER BY
                min(p1.id)
        """

        self._process_query(cr, uid, ids, query, context=context)

        for line in this.line_ids:
            partner_ids = literal_eval(line.aggr_ids)
            self._merge(cr, uid, partner_ids, context=context)
            line.unlink()
            cr.commit()

        this.write({'state': 'finished'})

        cr.execute("""
            UPDATE
                res_partner
            SET
                is_company = NULL,
                parent_id = NULL
            WHERE
                parent_id = id
        """)

        return {
            'type': 'ir.actions.act_window',
            'res_model': this._name,
            'res_id': this.id,
            'view_mode': 'form',
            'target': 'new',
        }

    def update_all_process_cb(self, cr, uid, ids, context=None):
        assert is_integer_list(ids)

        # WITH RECURSIVE cycle(id, parent_id) AS (
        #     SELECT id, parent_id FROM res_partner
        #   UNION
        #     SELECT  cycle.id, res_partner.parent_id
        #     FROM    res_partner, cycle
        #     WHERE   res_partner.id = cycle.parent_id AND
        #             cycle.id != cycle.parent_id
        # )
        # UPDATE  res_partner
        # SET     parent_id = NULL
        # WHERE   id in (SELECT id FROM cycle WHERE id = parent_id);

        this = self.browse(cr, uid, ids[0], context=context)

        self.parent_migration_process_cb(cr, uid, ids, context=None)

        list_merge = [
            {'group_by_vat': True,
             'group_by_email': True,
             'group_by_name': True},
            # {'group_by_name': True,
            #  'group_by_is_company': True,
            #  'group_by_parent_id': True},
            # {'group_by_email': True,
            #  'group_by_is_company': True,
            #  'group_by_parent_id': True},
            # {'group_by_name': True,
            #  'group_by_vat': True,
            #  'group_by_is_company': True,
            #  'exclude_journal_item': True},
            # {'group_by_email': True,
            #  'group_by_vat': True,
            #  'group_by_is_company': True,
            #  'exclude_journal_item': True},
            # {'group_by_email': True,
            #  'group_by_is_company': True,
            #  'exclude_contact': True,
            #  'exclude_journal_item': True},
            # {'group_by_name': True,
            #  'group_by_is_company': True,
            #  'exclude_contact': True,
            #  'exclude_journal_item': True}
        ]

        for merge_value in list_merge:
            id = self.create(cr, uid, merge_value, context=context)
            self.automatic_process_cb(cr, uid, [id], context=context)

        cr.execute("""
            UPDATE
                res_partner
            SET
                is_company = NULL
            WHERE
                parent_id IS NOT NULL AND
                is_company IS NOT NULL
        """)

        # cr.execute("""
        #     UPDATE
        #         res_partner as p1
        #     SET
        #         is_company = NULL,
        #         parent_id = (
        #             SELECT  p2.id
        #             FROM    res_partner as p2
        #             WHERE   p2.email = p1.email AND
        #                     p2.parent_id != p2.id
        #             LIMIT 1
        #         )
        #     WHERE
        #         p1.parent_id = p1.id
        # """)

        return self._next_screen(cr, uid, this, context)

    def merge_cb(self, cr, uid, ids, context=None):
        assert is_integer_list(ids)

        context = dict(context or {}, active_test=False)
        this = self.browse(cr, uid, ids[0], context=context)

        partner_ids = set(map(int, this.partner_ids))
        if not partner_ids:
            this.write({'state': 'finished'})
            return {
                'type': 'ir.actions.act_window',
                'res_model': this._name,
                'res_id': this.id,
                'view_mode': 'form',
                'target': 'new',
            }

        self._merge(cr, uid, partner_ids, this.dst_partner_id,
                    context=context)

        if this.current_line_id:
            this.current_line_id.unlink()

        return self._next_screen(cr, uid, this, context)

    def auto_set_parent_id(self, cr, uid, ids, context=None):
        assert is_integer_list(ids)

        # select partner who have one least invoice
        partner_treated = ['@gmail.com']
        cr.execute("""  SELECT p.id, p.email
                        FROM res_partner as p
                        LEFT JOIN account_invoice as a
                        ON p.id = a.partner_id AND a.state in ('open','paid')
                        WHERE p.grade_id is NOT NULL
                        GROUP BY p.id
                        ORDER BY COUNT(a.id) DESC
                """)
        re_email = re.compile(r".*@")
        for id, email in cr.fetchall():
            # check email domain
            email = re_email.sub("@", email or "")
            if not email or email in partner_treated:
                continue
            partner_treated.append(email)

            # don't update the partners if they are more of one who have
            # invoice
            cr.execute("""
                SELECT *
                    FROM res_partner as p
                        WHERE p.id != %s AND p.email LIKE '%%%s' AND
                           EXISTS (SELECT * FROM account_invoice as a
                                   WHERE p.id = a.partner_id
                                       AND a.state in ('open','paid'))
                    """ % (id, email))

            if len(cr.fetchall()) > 1:
                _logger.info("%s MORE OF ONE COMPANY", email)
                continue

            # to display changed values
            cr.execute("""  SELECT id,email
                            FROM res_partner
                            WHERE parent_id != %s
                                AND id != %s AND email LIKE '%%%s'
                    """ % (id, id, email))
            _logger.info("%r", cr.fetchall())

            # upgrade
            cr.execute("""  UPDATE res_partner
                            SET parent_id = %s
                            WHERE id != %s AND email LIKE '%%%s'
                    """ % (id, id, email))
        return False