addons/email_template/wizard/email_template_preview.py

   1 # -*- coding: utf-8 -*-
   2 ##############################################################################
   3 #
   4 #    OpenERP, Open Source Management Solution
   5 #    Copyright (C) 2009 Sharoon Thomas
   6 #    Copyright (C) 2010-Today OpenERP SA (<http://www.openerp.com>)
   7 #
   8 #    This program is free software: you can redistribute it and/or modify
   9 #    it under the terms of the GNU General Public License as published by
  10 #    the Free Software Foundation, either version 3 of the License, or
  11 #    (at your option) any later version.
  12 #
  13 #    This program is distributed in the hope that it will be useful,
  14 #    but WITHOUT ANY WARRANTY; without even the implied warranty of
  15 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16 #    GNU General Public License for more details.
  17 #
  18 #    You should have received a copy of the GNU General Public License
  19 #    along with this program.  If not, see <http://www.gnu.org/licenses/>
  20 #
  21 ##############################################################################
  22
  23 from osv import osv, fields
  24
  25 class email_template_preview(osv.osv_memory):
  26     _inherit = "email.template"
  27     _name = "email_template.preview"
  28     _description = "Email Template Preview"
  29
  30     def _get_records(self, cr, uid, context=None):
  31         """
  32         Return Records of particular Email Template's Model
  33         """
  34         if context is None:
  35             context = {}
  36
  37         template_id = context.get('template_id', False)
  38         if not template_id:
  39             return []
  40         email_template = self.pool.get('email.template')
  41         template = email_template.browse(cr, uid, int(template_id), context=context)
  42         template_object = template.model_id
  43         model =  self.pool.get(template_object.model)
  44         record_ids = model.search(cr, uid, [], 0, 10, 'id', context=context)
  45         default_id = context.get('default_res_id')
  46
  47         if default_id and default_id not in record_ids:
  48             record_ids.insert(0, default_id)
  49
  50         return model.name_get(cr, uid, record_ids, context)
  51
  52
  53     def default_get(self, cr, uid, fields, context=None):
  54         if context is None:
  55             context = {}
  56         result = super(email_template_preview, self).default_get(cr, uid, fields, context=context)
  57
  58         email_template = self.pool.get('email.template')
  59         template_id = context.get('template_id')
  60         if 'res_id' in fields and not result.get('res_id'):
  61             records = self._get_records(cr, uid, context=context)
  62             result['res_id'] = records and records[0][0] or False # select first record as a Default
  63         if template_id and 'model_id' in fields and not result.get('model_id'):
  64             result['model_id'] = email_template.read(cr, uid, int(template_id), ['model_id'], context).get('model_id', False)
  65         return result
  66
  67     _columns = {
  68         'res_id': fields.selection(_get_records, 'Sample Document'),
  69     }
  70
  71     def on_change_res_id(self, cr, uid, ids, res_id, context=None):
  72         if not res_id: return {}
  73         vals = {}
  74         email_template = self.pool.get('email.template')
  75         template_id = context and context.get('template_id')
  76         template = email_template.browse(cr, uid, template_id, context=context)
  77         vals['name'] = template.name
  78         mail_values = email_template.generate_email(cr, uid, template_id, res_id, context=context)
  79         for k in ('email_from','email_to','email_cc','reply_to','subject','body_html'):
  80             vals[k] = mail_values[k]
  81         return {'value': vals}
  82
  83 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: