# -*- coding: utf-8 -*- from openerp import api, models class ShiftLeaveWizard(models.TransientModel): _inherit = 'shift.leave.wizard' @api.multi def button_confirm(self): tmpl_name = False for wiz in self: tmpl_name = wiz.partner_id.in_ftop_team and \ 'louve_custom_email.louve_ftop_leave_email' or \ 'louve_custom_email.louve_abcd_leave_email' return super(ShiftLeaveWizard, self.with_context( leave_mail_tmpl=tmpl_name)).button_confirm()