1 # -*- coding: utf-8 -*-
2 ##############################################################################
4 # OpenERP, Open Source Management Solution
5 # Copyright (C) 2004-2010 Tiny SPRL (<http://tiny.be>).
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.
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.
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/>.
20 ##############################################################################
21 from osv import osv, fields
22 from tools.translate import _
24 class crm_merge_opportunity(osv.osv_memory):
25 """Merge two Opportunities"""
27 _name = 'crm.merge.opportunity'
28 _description = 'Merge two Opportunities'
30 def _get_first_not_null_id(self, attr, ops):
32 if hasattr(op, attr) and getattr(op, attr):
33 return getattr(op, attr).id
36 def _get_first_not_null(self, attr, ops):
38 if hasattr(op, attr) and getattr(op, attr):
39 return getattr(op, attr)
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)])
46 def get_attachments(self, cr, uid, id, context=None):
47 attach_obj = self.pool.get('ir.attachment')
49 attach_ids = attach_obj.search(cr, uid, [('res_model' , '=', 'crm.lead'), ('res_id', '=', id)])
52 def set_attachements_res_id(self, cr, uid, op_id, attach_ids, context=None):
53 attach_obj = self.pool.get('ir.attachment')
54 attach_obj.write(cr, uid, attach_ids, {'res_id' : op_id})
58 def merge(self, cr, uid, op_ids, context=None):
60 @param opp_ids : list of opportunities ids to merge
62 opp_obj = self.pool.get('crm.lead')
63 message_obj = self.pool.get('mailgate.message')
65 lead_ids = context and context.pop('lead_ids', []) or []
69 raise osv.except_osv(_('Warning !'),_('Please select more than one opportunities.'))
71 opportunities = opp_obj.browse(cr, uid, lead_ids, context=context)
72 opportunities_list = list(set(op_ids) - set(opportunities))
75 first_opportunity = opportunities[0]
76 tail_opportunities = opportunities_list
78 first_opportunity = opportunities_list[0]
79 tail_opportunities = opportunities_list[1:]
83 'partner_id': self._get_first_not_null_id('partner_id', op_ids), # !!
84 'title': self._get_first_not_null_id('title', op_ids),
85 'name' : self._concat_all('name', op_ids), #not lost
86 'categ_id' : self._get_first_not_null_id('categ_id', op_ids), # !!
87 'channel_id' : self._get_first_not_null_id('channel_id', op_ids), # !!
88 'city' : self._get_first_not_null('city', op_ids), # !!
89 'company_id' : self._get_first_not_null_id('company_id', op_ids), #!!
90 'contact_name' : self._concat_all('contact_name', op_ids), #not lost
91 'country_id' : self._get_first_not_null_id('country_id', op_ids), #!!
92 'partner_address_id' : self._get_first_not_null_id('partner_address_id', op_ids), #!!
93 'partner_assigned_id' : hasattr(opp_obj,'partner_assigned_id') and self._get_first_not_null_id('partner_assigned_id', op_ids), #!!
94 'type_id' : self._get_first_not_null_id('type_id', op_ids), #!!
95 'user_id' : self._get_first_not_null_id('user_id', op_ids), #!!
96 'section_id' : self._get_first_not_null_id('section_id', op_ids), #!!
97 'state_id' : self._get_first_not_null_id('state_id', op_ids),
98 'description' : self._concat_all('description', op_ids), #not lost
99 'email' : self._get_first_not_null('email', op_ids), # !!
100 'fax' : self._get_first_not_null('fax', op_ids),
101 'mobile' : self._get_first_not_null('mobile', op_ids),
102 'partner_latitude' : hasattr(opp_obj,'partner_latitude') and self._get_first_not_null('partner_latitude', op_ids),
103 'partner_longitude' : hasattr(opp_obj,'partner_longitude') and self._get_first_not_null('partner_longitude', op_ids),
104 'partner_name' : self._get_first_not_null('partner_name', op_ids),
105 'phone' : self._get_first_not_null('phone', op_ids),
106 'probability' : self._get_first_not_null('probability', op_ids),
107 'planned_revenue' : self._get_first_not_null('planned_revenue', op_ids),
108 'street' : self._get_first_not_null('street', op_ids),
109 'street2' : self._get_first_not_null('street2', op_ids),
110 'zip' : self._get_first_not_null('zip', op_ids),
114 #copy message into the first opportunity + merge attachement
115 for opp in tail_opportunities:
116 attach_ids = self.get_attachments(cr, uid, opp, context=context)
117 self.set_attachements_res_id(cr, uid, first_opportunity.id, attach_ids)
118 for history in opp.message_ids:
119 new_history = message_obj.copy(cr, uid, history.id, default={'res_id': opp.id})
121 #Notification about loss of information
123 subject = ['Merged opportunities :']
125 subject.append(opp.name)
126 details.append(_('Merged Opportunity: %s\n Partner: %s\n Stage: %s\n Section: %s\n Salesman: %s\n Category: %s\n Channel: %s\n Company: %s\n Contact name: %s\n Email: %s\n Phone number: %s\n Fax: %s\n Mobile: %s\n State: %s\n Description: %s\n Probability: %s\n Planned revennue: %s\n Country: %s\n City: %s\n Street: %s\n Street 2: %s\n Zip 2: %s') % ( opp.name, opp.partner_id.name or '',
127 opp.stage_id.name or '',
128 opp.section_id.name or '',
129 opp.user_id.name or '',
130 opp.categ_id.name or '',
131 opp.channel_id.name or '',
132 opp.company_id.name or '',
133 opp.contact_name or '',
134 opp.email_from or '',
138 opp.state_id.name or '',
139 opp.description or '',
140 opp.probability or '',
141 opp.planned_revenue or '',
142 opp.country_id.name or '',
148 subject = subject[0] + ", ".join(subject[1:])
149 details = "\n\n".join(details)
151 opp_obj._history(cr, uid, [first_opportunity], subject, details=details)
152 #data.update({'message_ids' : [(6, 0 ,self._concat_o2m('message_ids', op_ids))]})
153 opp_obj.write(cr, uid, [first_opportunity.id], data)
154 unlink_ids = map(lambda x: x.id, tail_opportunities)
155 opp_obj.unlink(cr, uid, unlink_ids, context=context)
157 models_data = self.pool.get('ir.model.data')
161 # Get Opportunity views
162 result = models_data._get_id(
163 cr, uid, 'crm', 'view_crm_case_opportunities_filter')
164 opportunity_view_form = models_data._get_id(
165 cr, uid, 'crm', 'crm_case_form_view_oppor')
166 opportunity_view_tree = models_data._get_id(
167 cr, uid, 'crm', 'crm_case_tree_view_oppor')
168 if opportunity_view_form:
169 opportunity_view_form = models_data.browse(
170 cr, uid, opportunity_view_form, context=context).res_id
171 if opportunity_view_tree:
172 opportunity_view_tree = models_data.browse(
173 cr, uid, opportunity_view_tree, context=context).res_id
176 'name': _('Opportunity'),
178 'view_mode': 'tree, form',
179 'res_model': 'crm.lead',
180 'domain': [('type', '=', 'opportunity')],
181 'res_id': int(first_opportunity.id),
183 'views': [(opportunity_view_form, 'form'),
184 (opportunity_view_tree, 'tree'),
185 (False, 'calendar'), (False, 'graph')],
186 'type': 'ir.actions.act_window',
190 def action_merge(self, cr, uid, ids, context=None):
191 obj_opportunity = self.browse(cr, uid, ids[0], context=context)
192 op_ids = obj_opportunity.opportunity_ids
193 self.write(cr, uid, ids, {'opportunity_ids' : [(6,0, [op_ids[0].id])]}, context=context)
194 context['lead_ids'] = [op_ids[0].id]
195 return self.merge(cr, uid, op_ids, context)
199 'opportunity_ids' : fields.many2many('crm.lead', 'merge_opportunity_rel', 'merge_id', 'opportunity_id', 'Opportunities', domain=[('type', '=', 'opportunity')]),
202 def default_get(self, cr, uid, fields, context=None):
204 This function gets default values
205 @param self: The object pointer
206 @param cr: the current row, from the database cursor,
207 @param uid: the current user’s ID for security checks,
208 @param fields: List of fields for default value
209 @param context: A standard dictionary for contextual values
211 @return : default values of fields.
213 record_ids = context and context.get('active_ids', False) or False
214 res = super(crm_merge_opportunity, self).default_get(cr, uid, fields, context=context)
217 if 'opportunity_ids' in fields:
218 res.update({'opportunity_ids': record_ids})
222 crm_merge_opportunity()
224 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: