[IMP] use model._fields instead of model._all_columns to cover all fields
[odoo/odoo.git] / addons / mass_mailing / controllers / main.py
1
2 import werkzeug
3
4 from openerp import http, SUPERUSER_ID
5 from openerp.http import request
6
7
8 class MassMailController(http.Controller):
9
10     @http.route('/mail/track/<int:mail_id>/blank.gif', type='http', auth='none')
11     def track_mail_open(self, mail_id, **post):
12         """ Email tracking. """
13         mail_mail_stats = request.registry.get('mail.mail.statistics')
14         mail_mail_stats.set_opened(request.cr, SUPERUSER_ID, mail_mail_ids=[mail_id])
15         response = werkzeug.wrappers.Response()
16         response.mimetype = 'image/gif'
17         response.data = 'R0lGODlhAQABAIAAANvf7wAAACH5BAEAAAAALAAAAAABAAEAAAICRAEAOw=='.decode('base64')
18         return response
19
20     @http.route(['/mail/mailing/<int:mailing_id>/unsubscribe'], type='http', auth='none')
21     def mailing(self, mailing_id, email=None, res_id=None, **post):
22         cr, uid, context = request.cr, request.uid, request.context
23         MassMailing = request.registry['mail.mass_mailing']
24         mailing_ids = MassMailing.exists(cr, SUPERUSER_ID, [mailing_id], context=context)
25         if not mailing_ids:
26             return 'KO'
27         mailing = MassMailing.browse(cr, SUPERUSER_ID, mailing_ids[0], context=context)
28         if mailing.mailing_model == 'mail.mass_mailing.contact':
29             list_ids = [l.id for l in mailing.contact_list_ids]
30             record_ids = request.registry[mailing.mailing_model].search(cr, SUPERUSER_ID, [('list_id', 'in', list_ids), ('id', '=', res_id), ('email', 'ilike', email)], context=context)
31             request.registry[mailing.mailing_model].write(cr, SUPERUSER_ID, record_ids, {'opt_out': True}, context=context)
32         else:
33             email_fname = None
34             model = request.registry[mailing.mailing_model]
35             if 'email_from' in model._fields:
36                 email_fname = 'email_from'
37             elif 'email' in model._fields:
38                 email_fname = 'email'
39             if email_fname:
40                 record_ids = model.search(cr, SUPERUSER_ID, [('id', '=', res_id), (email_fname, 'ilike', email)], context=context)
41             if 'opt_out' in model._fields:
42                 model.write(cr, SUPERUSER_ID, record_ids, {'opt_out': True}, context=context)
43         return 'OK'
44
45     @http.route(['/website_mass_mailing/is_subscriber'], type='json', auth="public", website=True)
46     def is_subscriber(self, list_id, **post):
47         cr, uid, context = request.cr, request.uid, request.context
48         Contacts = request.registry['mail.mass_mailing.contact']
49         Users = request.registry['res.users']
50
51         is_subscriber = False
52         email = None
53         if uid != request.website.user_id.id:
54             email = Users.browse(cr, SUPERUSER_ID, uid, context).email
55         elif request.session.get('mass_mailing_email'):
56             email = request.session['mass_mailing_email']
57
58         if email:
59             contact_ids = Contacts.search(cr, SUPERUSER_ID, [('list_id', '=', int(list_id)), ('email', '=', email), ('opt_out', '=', False)], context=context)
60             is_subscriber = len(contact_ids) > 0
61
62         return {'is_subscriber': is_subscriber, 'email': email}
63
64     @http.route(['/website_mass_mailing/subscribe'], type='json', auth="public", website=True)
65     def subscribe(self, list_id, email, **post):
66         cr, uid, context = request.cr, request.uid, request.context
67         Contacts = request.registry['mail.mass_mailing.contact']
68
69         contact_ids = Contacts.search_read(cr, SUPERUSER_ID, [('list_id', '=', int(list_id)), ('email', '=', email)], ['opt_out'], context=context)
70         if not contact_ids:
71             Contacts.add_to_list(cr, SUPERUSER_ID, email, int(list_id), context=context)
72         else:
73             if contact_ids[0]['opt_out']:
74                 Contacts.write(cr, SUPERUSER_ID, [contact_ids[0]['id']], {'opt_out': False}, context=context)
75         # add email to session
76         request.session['mass_mailing_email'] = email
77         return True