addons/crm/wizard/crm_merge_opportunities.py

   1 # -*- coding: utf-8 -*-
   2 ##############################################################################
   3 #
   4 #    OpenERP, Open Source Management Solution
   5 #    Copyright (C) 2004-2010 Tiny SPRL (<http://tiny.be>).
   6 #
   7 #    This program is free software: you can redistribute it and/or modify
   8 #    it under the terms of the GNU Affero General Public License as
   9 #    published by the Free Software Foundation, either version 3 of the
  10 #    License, or (at your option) any later version.
  11 #
  12 #    This program is distributed in the hope that it will be useful,
  13 #    but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 #    GNU Affero General Public License for more details.
  16 #
  17 #    You should have received a copy of the GNU Affero General Public License
  18 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19 #
  20 ##############################################################################
  21 from osv import osv, fields
  22 from tools.translate import _
  23
  24 class crm_merge_opportunity(osv.osv_memory):
  25     """Merge two Opportunities"""
  26
  27     _name = 'crm.merge.opportunity'
  28     _description = 'Merge two Opportunities'
  29
  30     def _get_first_not_null_id(self, attr, ops):
  31         for op in ops:
  32             if hasattr(op, attr) and getattr(op, attr):
  33                 return getattr(op, attr).id
  34         return False
  35
  36     def _get_first_not_null(self, attr, ops):
  37         for op in ops:
  38             if hasattr(op, attr) and getattr(op, attr):
  39                 return getattr(op, attr)
  40         return False
  41
  42     def _concat_all(self, attr, ops):
  43         return ', '.join([getattr(op, attr) for op in ops if hasattr(op, attr) and getattr(op, attr)])
  44
  45
  46     def merge(self, cr, uid, op_ids, context=None):
  47         """
  48             @param opp_ids : list of opportunities ids to merge
  49         """
  50         opp_obj = self.pool.get('crm.lead')
  51         message_obj = self.pool.get('mailgate.message')
  52
  53         if len(op_ids) <= 1:
  54             raise osv.except_osv(_('Warning !'),_('Please select more than one opportunities.'))
  55
  56         first_opp = op_ids[0]
  57         data = {
  58                 'partner_id': self._get_first_not_null_id('partner_id', op_ids),  # !!
  59                 'title': self._get_first_not_null_id('title', op_ids),
  60                 'name' : self._concat_all('name', op_ids),  #not lost
  61                 'categ_id' : self._get_first_not_null_id('categ_id', op_ids), # !!
  62                 'channel_id' : self._get_first_not_null_id('channel_id', op_ids), # !!
  63                 'city' : self._get_first_not_null('city', op_ids),  # !!
  64                 'company_id' : self._get_first_not_null_id('company_id', op_ids), #!!
  65                 'contact_name' : self._concat_all('contact_name', op_ids), #not lost
  66                 'country_id' : self._get_first_not_null_id('country_id', op_ids), #!!
  67                 'partner_address_id' : self._get_first_not_null_id('partner_address_id', op_ids), #!!
  68                 'partner_assigned_id' : hasattr(opp_obj,'partner_assigned_id') and self._get_first_not_null_id('partner_assigned_id', op_ids), #!!
  69                 'type_id' : self._get_first_not_null_id('type_id', op_ids), #!!
  70                 'user_id' : self._get_first_not_null_id('user_id', op_ids), #!!
  71                 'section_id' : self._get_first_not_null_id('section_id', op_ids), #!!
  72                 'state_id' : self._get_first_not_null_id('state_id', op_ids),
  73                 'description' : self._concat_all('description', op_ids),  #not lost
  74                 'email' : self._get_first_not_null('email', op_ids), # !!
  75                 'fax' : self._get_first_not_null('fax', op_ids),
  76                 'mobile' : self._get_first_not_null('mobile', op_ids),
  77                 'partner_latitude' : hasattr(opp_obj,'partner_latitude') and self._get_first_not_null('partner_latitude', op_ids),
  78                 'partner_longitude' : hasattr(opp_obj,'partner_longitude') and self._get_first_not_null('partner_longitude', op_ids),
  79                 'partner_name' : self._get_first_not_null('partner_name', op_ids),
  80                 'phone' : self._get_first_not_null('phone', op_ids),
  81                 'probability' : self._get_first_not_null('probability', op_ids),
  82                 'planned_revenue' : self._get_first_not_null('planned_revenue', op_ids),
  83                 'street' : self._get_first_not_null('street', op_ids),
  84                 'street2' : self._get_first_not_null('street2', op_ids),
  85                 'zip' : self._get_first_not_null('zip', op_ids),
  86
  87             }
  88
  89
  90
  91         #copy message into the first opportunity
  92         for opp in op_ids[1:]:
  93             for history in opp.message_ids:
  94                 new_history = message_obj.copy(cr, uid, history.id, default={'res_id': opp.id})
  95         #Notification about loss of information
  96         details = []
  97         subject = ['Merged opportunities :']
  98         for opp in op_ids:
  99             subject.append(opp.name)
 100             details.append(_('Merged Opportunity: %s\n  Partner: %s\n  Stage: %s\n  Section: %s\n   Salesman: %s\n   Category: %s\n   Channel: %s\n   City: %s\n   Company: %s\n   Country: %s\n   Email: %s\n   Phone number: %s\n   Contact name: %s')  % ( opp.name, opp.partner_id.name or '',
 101                         opp.stage_id.name or '',
 102                         opp.section_id.name or '',
 103                         opp.user_id.name or '',
 104                         opp.categ_id.name or '',
 105                         opp.channel_id.name or '',
 106                         opp.city or '',
 107                         opp.company_id.name or '',
 108                         opp.country_id.name or '',
 109                         opp.email_from or '',
 110                         opp.phone or '',
 111                         opp.contact_name or ''))
 112         subject = subject[0] + ", ".join(subject[1:])
 113         details = "\n\n".join(details)
 114
 115         opp_obj._history(cr, uid, [first_opp], subject, details=details)
 116
 117         #data.update({'message_ids' : [(6, 0 ,self._concat_o2m('message_ids', op_ids))]})
 118         opp_obj.write(cr, uid, [first_opp.id], data)
 119
 120         unlink_ids = map(lambda x: x.id, op_ids[1:])
 121         opp_obj.unlink(cr, uid, unlink_ids)
 122
 123         models_data = self.pool.get('ir.model.data')
 124
 125
 126
 127         # Get Opportunity views
 128         result = models_data._get_id(
 129             cr, uid, 'crm', 'view_crm_case_opportunities_filter')
 130         opportunity_view_form = models_data._get_id(
 131             cr, uid, 'crm', 'crm_case_form_view_oppor')
 132         opportunity_view_tree = models_data._get_id(
 133             cr, uid, 'crm', 'crm_case_tree_view_oppor')
 134         if opportunity_view_form:
 135             opportunity_view_form = models_data.browse(
 136                 cr, uid, opportunity_view_form, context=context).res_id
 137         if opportunity_view_tree:
 138             opportunity_view_tree = models_data.browse(
 139                 cr, uid, opportunity_view_tree, context=context).res_id
 140
 141         return {
 142                 'name': _('Opportunity'),
 143                 'view_type': 'form',
 144                 'view_mode': 'tree, form',
 145                 'res_model': 'crm.lead',
 146                 'domain': [('type', '=', 'opportunity')],
 147                 'res_id': int(first_opp.id),
 148                 'view_id': False,
 149                 'views': [(opportunity_view_form, 'form'),
 150                           (opportunity_view_tree, 'tree'),
 151                           (False, 'calendar'), (False, 'graph')],
 152                 'type': 'ir.actions.act_window',
 153         }
 154
 155
 156     def action_merge(self, cr, uid, ids, context=None):
 157         obj_opportunity = self.browse(cr, uid, ids[0], context=context)
 158         if hasattr(obj_opportunity, 'opportunity_ids'):
 159             op_ids = obj_opportunity.opportunity_ids
 160
 161
 162         return self.merge(cr, uid, op_ids, context)
 163
 164
 165     _columns = {
 166         'opportunity_ids' : fields.many2many('crm.lead',  'merge_opportunity_rel', 'merge_id', 'opportunity_id', 'Opportunities', domain=[('type', '=', 'opportunity')]),
 167     }
 168
 169     def default_get(self, cr, uid, fields, context=None):
 170         """
 171         This function gets default values
 172         @param self: The object pointer
 173         @param cr: the current row, from the database cursor,
 174         @param uid: the current user’s ID for security checks,
 175         @param fields: List of fields for default value
 176         @param context: A standard dictionary for contextual values
 177
 178         @return : default values of fields.
 179         """
 180         record_ids = context and context.get('active_ids', False) or False
 181         res = super(crm_merge_opportunity, self).default_get(cr, uid, fields, context=context)
 182
 183         if record_ids:
 184             if 'opportunity_ids' in fields:
 185                 res.update({'opportunity_ids': record_ids})
 186
 187         return res
 188
 189 crm_merge_opportunity()
 190
 191 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: