odoo/addons/mail/mail_subscription.py

# -*- coding: utf-8 -*-
##############################################################################
#
#    OpenERP, Open Source Management Solution
#    Copyright (C) 2009-today OpenERP SA (<http://www.openerp.com>)
#
#    This program is free software: you can redistribute it and/or modify
#    it under the terms of the GNU Affero General Public License as
#    published by the Free Software Foundation, either version 3 of the
#    License, or (at your option) any later version
#
#    This program is distributed in the hope that it will be useful,
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#    GNU Affero General Public License for more details
#
#    You should have received a copy of the GNU Affero General Public License
#    along with this program.  If not, see <http://www.gnu.org/licenses/>
#
##############################################################################

from osv import osv
from osv import fields

class mail_subscription(osv.osv):
    """
    mail_subscription holds the data related to the follow mechanism inside OpenERP.
    A subscription is characterized by:
        :param: res_model: model of the followed objects
        :param: res_id: ID of resource (may be 0 for every objects)
        :param: user_id: user_id of the follower
    """
    _name = 'mail.subscription'
    _rec_name = 'id'
    _order = 'res_model asc'
    _description = 'Mail subscription'
    _columns = {
        'res_model': fields.char('Related Document Model', size=128,
                        required=True, select=1,
                        help='Model of the followed resource'),
        'res_id': fields.integer('Related Document ID', select=1,
                        help='Id of the followed resource'),
        'user_id': fields.many2one('res.users', string='Related User',
                        ondelete='cascade', required=True, select=1),
    }

class mail_notification(osv.osv):
    """
    mail_notification is a relational table modeling messages pushed to users.
        :param: read: not used currently
    """
    _name = 'mail.notification'
    _rec_name = 'id'
    _log_access = False
    _order = 'message_id desc'
    _description = 'Mail notification'
    _columns = {
        'user_id': fields.many2one('res.users', string='User',
                        ondelete='cascade', required=True, select=1),
        'message_id': fields.many2one('mail.message', string='Message',
                        ondelete='cascade', required=True, select=1),
        'read': fields.boolean('Read', help="Not used currently",),
        # TODO: add a timestamp ? or use message date ?
    }
    _defaults = {
        'read': False,
    }
    # Create notification in the wall of each user
    # Send by email the notification depending on the user preferences
    def notify(self, cr, uid, user_ids, msg_id, context=context):
        user_obj = self.pool.get('res.users')
        msg_obj = self.pool.get('mail.message')
        msg = msg_obj.browse(cr, uid, msg_id, context=context)

        towrite = {
            'email_to': msg.email_to,
            'subject': msg.subject
        }
        for user in user_obj.browse(cr, uid, user_ids, context=context):
            notification_obj.create(cr, uid, {
                'user_id': user.id,
                'message_id': msg_id
            }, context=context)
            if user.notification_email_pref=='none' or not user.user_email:
                continue

            if user.notification_email_pref=='comment' and msg.type in ('email','comment'):
                continue

            if not msg.email_from:
                current_user = res_users_obj.browse(cr, uid, uid, context=context)
                towrite['email_from'] = current_user.user_email

            towrite['state'] = 'outgoing'
            if not towrite.get('email_to', False):
                towrite['email_to'] = email_to
            else:
                if email_to not in towrite['email_to']:
                    towrite['email_to'] = towrite['email_to'] + ', ' + email_to

            if towrite.get('subject', False):
                towrite['subject'] = msg.name_get(cr, uid, [msg.id], context=context)[0][1]
        if towrite.get('state', False):
            msg_obj.write(cr, uid, [msg.id], towrite, context=context)
        return True