# -*- coding:utf-8 -*- from openerp import api, fields, models, exceptions, osv from datetime import datetime class crm_lead(models.Model): _inherit="crm.lead" task_ids = fields.One2many('crm.task','opportunity_id',"Actividades") @api.multi def open_wizard(self): now = datetime.now() view = self.env.ref('eiru_crm.crm_task_wizard') return { 'name': 'Añadir Actividad', 'res_model': 'crm.task', 'view_type': 'form', 'view_mode': 'form', 'target': 'new', 'views': [(view.id, 'form')], 'view_id': view.id, 'context': { 'default_date_action_next': now.strftime("%Y-%m-%d %H:%M:00"), 'default_user_id': self.user_id.id, 'default_opportunity_id': self.id, 'default_partner_id': self.partner_id.id, }, 'type': 'ir.actions.act_window', } @api.one def _exist_task(self): now = datetime.now() for task in self: count1 = self.env['crm.task'].search_count([('opportunity_id', '=', task.id),('state','=','pending'), ('date_action_next','>=',now.strftime("%Y-%m-%d %H:%M:00"))]) count2 = self.env['crm.task'].search_count([('opportunity_id', '=', task.id),('date_action_next','<',now.strftime("%Y-%m-%d %H:%M:00")),('state','=','pending')]) if count1 > 0: self.exist_task = True if count2 > 0: self.expired_date = True exist_task = fields.Boolean(string="Tarea Pendiente", compute="_exist_task", store=False) expired_date = fields.Boolean(string="Fecha Vencida", compute="_exist_task", store=True) def on_change_partner_id(self, cr, uid, ids, partner_id, context=None): values = {} if partner_id: partner = self.pool.get('res.partner').browse(cr, uid, partner_id, context=context) partner_name = (partner.parent_id and partner.parent_id.name) or (partner.is_company and partner.name) or False values = { 'partner_name': partner_name, 'contact_name': (not partner.is_company and partner.name) or False, 'title': partner.title and partner.title.id or False, 'street': partner.street, 'street2': partner.street2, 'city': partner.city, 'state_id': partner.state_id and partner.state_id.id or False, 'country_id': partner.country_id and partner.country_id.id or False, 'email_from': partner.email, 'phone': partner.phone, 'mobile': partner.mobile, 'fax': partner.fax, 'zip': partner.zip, 'function': partner.function, 'facebook': partner.facebook, 'instagram': partner.instagram, } return {'value': values} def case_mark_won(self, cr, uid, ids, context=None): """ Mark the case as won: state=done and probability=100 """ for each in ids: crm = self.pool.get('crm.lead').browse(cr, uid, each) self.pool.get('res.partner').write(cr, uid, crm.partner_id.id, {'customer': True}, context=context) stages_leads = {} for lead in self.browse(cr, uid, ids, context=context): stage_id = self.stage_find(cr, uid, [lead], lead.section_id.id or False, [('probability', '=', 100.0), ('on_change', '=', True)], context=context) if stage_id: if stages_leads.get(stage_id): stages_leads[stage_id].append(lead.id) else: stages_leads[stage_id] = [lead.id] else: raise osv.except_osv(_('Warning!'), _('To relieve your sales pipe and group all Won opportunities, configure one of your sales stage as follow:\n' 'probability = 100 % and select "Change Probability Automatically".\n' 'Create a specific stage or edit an existing one by editing columns of your opportunity pipe.')) for stage_id, lead_ids in stages_leads.items(): self.write(cr, uid, lead_ids, {'stage_id': stage_id}, context=context) return True def create(self, cr, uid, vals, context=None): context = dict(context or {}) if vals.get('type') and not context.get('default_type'): context['default_type'] = vals.get('type') if vals.get('section_id') and not context.get('default_section_id'): context['default_section_id'] = vals.get('section_id') if vals.get('user_id'): vals['date_open'] = fields.datetime.now() partner_id = vals['partner_id'] values = { 'email': vals['email_from'], 'phone': vals['phone'], 'mobile': vals['mobile'], 'facebook': vals['facebook'], 'instagram': vals['instagram'] } self.pool.get('res.partner').write(cr,uid,partner_id,values, context=context) # context: no_log, because subtype already handle this create_context = dict(context, mail_create_nolog=True) return super(crm_lead, self).create(cr, uid, vals, context=create_context) def write(self, cr, uid, ids, vals, context=None): # stage change: update date_last_stage_update if 'stage_id' in vals: vals['date_last_stage_update'] = fields.datetime.now() if vals.get('user_id'): vals['date_open'] = fields.datetime.now() # stage change with new stage: update probability and date_closed if vals.get('stage_id') and not vals.get('probability'): onchange_stage_values = self.onchange_stage_id(cr, uid, ids, vals.get('stage_id'), context=context)['value'] vals.update(onchange_stage_values) self.update_partner(cr,uid,ids,vals) return super(crm_lead, self).write(cr, uid, ids, vals, context=context) def update_partner(self, cr, uid, ids, vals, context=None): context = dict(context or {}) crm = self.pool.get('crm.lead').browse(cr,uid, ids) partner_id = crm.partner_id.id values = {'email_from','phone','mobile','facebook','instagram'} vals2 = {} for each in vals: if each in values: if each == 'email_from': vals2['email'] = vals[each] else: vals2[each] = vals[each] self.pool.get('res.partner').write(cr,uid,partner_id,vals2, context=context) return True class CrmStage(models.Model): _inherit="crm.case.stage" active = fields.Boolean('Activo', default=True) type2 = fields.Selection([('opportunity', 'Oportunidad')], string='Tipo', required=True, default='opportunity') code = fields.Char('Codigo') @api.multi def unlink(self): for record in self: if record.code == 'LEAD_NO_DELETE': raise exceptions.Warning(('Aviso'), ('Este estado no puede ser eliminado. Pruebe desactivarlo.')) return super(CrmStage, self).unlink()