addons/mail/res_users.py

   1 # -*- coding: utf-8 -*-
   2 ##############################################################################
   3 #
   4 #    OpenERP, Open Source Management Solution
   5 #    Copyright (C) 2009-Today OpenERP SA (<http://www.openerp.com>)
   6 #
   7 #    This program is free software: you can redistribute it and/or modify
   8 #    it under the terms of the GNU Affero General Public License as
   9 #    published by the Free Software Foundation, either version 3 of the
  10 #    License, or (at your option) any later version
  11 #
  12 #    This program is distributed in the hope that it will be useful,
  13 #    but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 #    GNU Affero General Public License for more details
  16 #
  17 #    You should have received a copy of the GNU Affero General Public License
  18 #    along with this program.  If not, see <http://www.gnu.org/licenses/>
  19 #
  20 ##############################################################################
  21
  22 from openerp.osv import fields, osv
  23 from openerp import SUPERUSER_ID
  24 from openerp.tools.translate import _
  25
  26
  27 class res_users(osv.Model):
  28     """ Update of res.users class
  29         - add a preference about sending emails about notifications
  30         - make a new user follow itself
  31         - add a welcome message
  32         - add suggestion preference
  33     """
  34     _name = 'res.users'
  35     _inherit = ['res.users']
  36     _inherits = {'mail.alias': 'alias_id'}
  37
  38     _columns = {
  39         'alias_id': fields.many2one('mail.alias', 'Alias', ondelete="restrict", required=True,
  40             help="Email address internally associated with this user. Incoming "\
  41                  "emails will appear in the user's notifications."),
  42         'display_groups_suggestions': fields.boolean("Display Groups Suggestions"),
  43     }
  44
  45     _defaults = {
  46         'display_groups_suggestions': True,
  47     }
  48
  49     def __init__(self, pool, cr):
  50         """ Override of __init__ to add access rights on notification_email_send
  51             and alias fields. Access rights are disabled by default, but allowed
  52             on some specific fields defined in self.SELF_{READ/WRITE}ABLE_FIELDS.
  53         """
  54         init_res = super(res_users, self).__init__(pool, cr)
  55         # duplicate list to avoid modifying the original reference
  56         self.SELF_WRITEABLE_FIELDS = list(self.SELF_WRITEABLE_FIELDS)
  57         self.SELF_WRITEABLE_FIELDS.extend(['notification_email_send', 'display_groups_suggestions'])
  58         # duplicate list to avoid modifying the original reference
  59         self.SELF_READABLE_FIELDS = list(self.SELF_READABLE_FIELDS)
  60         self.SELF_READABLE_FIELDS.extend(['notification_email_send', 'alias_domain', 'alias_name', 'display_groups_suggestions'])
  61         return init_res
  62
  63     def _auto_init(self, cr, context=None):
  64         """ Installation hook: aliases, partner following themselves """
  65         # create aliases for all users and avoid constraint errors
  66         return self.pool.get('mail.alias').migrate_to_alias(cr, self._name, self._table, super(res_users, self)._auto_init,
  67             self._name, self._columns['alias_id'], 'login', alias_force_key='id', context=context)
  68
  69     def create(self, cr, uid, data, context=None):
  70         if not data.get('login', False):
  71             raise osv.except_osv(_('Invalid Action!'), _('You may not create a user. To create new users, you should use the "Settings > Users" menu.'))
  72         if context is None:
  73             context = {}
  74
  75         create_context = dict(context, alias_model_name=self._name, alias_parent_model_name=self._name)
  76         user_id = super(res_users, self).create(cr, uid, data, context=create_context)
  77         user = self.browse(cr, uid, user_id, context=context)
  78         self.pool.get('mail.alias').write(cr, SUPERUSER_ID, [user.alias_id.id], {"alias_force_thread_id": user_id, "alias_parent_thread_id": user_id}, context)
  79
  80         # create a welcome message
  81         self._create_welcome_message(cr, uid, user, context=context)
  82         return user_id
  83
  84     def copy_data(self, *args, **kwargs):
  85         data = super(res_users, self).copy_data(*args, **kwargs)
  86         if data and data.get('alias_name'):
  87             data['alias_name'] = data['login']
  88         return data
  89
  90     def _create_welcome_message(self, cr, uid, user, context=None):
  91         if not self.has_group(cr, uid, 'base.group_user'):
  92             return False
  93         company_name = user.company_id.name if user.company_id else ''
  94         body = _('%s has joined the %s network.') % (user.name, company_name)
  95         # TODO change SUPERUSER_ID into user.id but catch errors
  96         return self.pool.get('res.partner').message_post(cr, SUPERUSER_ID, [user.partner_id.id],
  97             body=body, context=context)
  98
  99     def unlink(self, cr, uid, ids, context=None):
 100         # Cascade-delete mail aliases as well, as they should not exist without the user.
 101         alias_pool = self.pool.get('mail.alias')
 102         alias_ids = [user.alias_id.id for user in self.browse(cr, uid, ids, context=context) if user.alias_id]
 103         res = super(res_users, self).unlink(cr, uid, ids, context=context)
 104         alias_pool.unlink(cr, uid, alias_ids, context=context)
 105         return res
 106
 107     def _message_post_get_pid(self, cr, uid, thread_id, context=None):
 108         assert thread_id, "res.users does not support posting global messages"
 109         if context and 'thread_model' in context:
 110             context['thread_model'] = 'res.users'
 111         if isinstance(thread_id, (list, tuple)):
 112             thread_id = thread_id[0]
 113         return self.browse(cr, SUPERUSER_ID, thread_id).partner_id.id
 114
 115     def message_post(self, cr, uid, thread_id, context=None, **kwargs):
 116         """ Redirect the posting of message on res.users to the related partner.
 117             This is done because when giving the context of Chatter on the
 118             various mailboxes, we do not have access to the current partner_id. """
 119         if isinstance(thread_id, (list, tuple)):
 120             thread_id = thread_id[0]
 121         partner_ids = kwargs.get('partner_ids', [])
 122         partner_id = self._message_post_get_pid(cr, uid, thread_id, context=context)
 123         if partner_id not in [command[1] for command in partner_ids]:
 124             partner_ids.append(partner_id)
 125         kwargs['partner_ids'] = partner_ids
 126         return self.pool.get('mail.thread').message_post(cr, uid, False, **kwargs)
 127
 128     def message_update(self, cr, uid, ids, msg_dict, update_vals=None, context=None):
 129         return True
 130
 131     def message_subscribe(self, cr, uid, ids, partner_ids, subtype_ids=None, context=None):
 132         return True
 133
 134     def message_get_partner_info_from_emails(self, cr, uid, emails, link_mail=False, context=None):
 135         return self.pool.get('mail.thread').message_get_partner_info_from_emails(cr, uid, emails, link_mail=link_mail, context=context)
 136
 137     def message_get_suggested_recipients(self, cr, uid, ids, context=None):
 138         return dict.fromkeys(ids, list())
 139
 140     def stop_showing_groups_suggestions(self, cr, uid, user_id, context=None):
 141         """Update display_groups_suggestions value to False"""
 142         if context is None:
 143             context = {}
 144         self.write(cr, uid, user_id, {"display_groups_suggestions": False}, context)
 145
 146
 147 class res_users_mail_group(osv.Model):
 148     """ Update of res.users class
 149         - if adding groups to an user, check mail.groups linked to this user
 150           group, and the user. This is done by overriding the write method.
 151     """
 152     _name = 'res.users'
 153     _inherit = ['res.users']
 154
 155     # FP Note: to improve, post processing may be better ?
 156     def write(self, cr, uid, ids, vals, context=None):
 157         write_res = super(res_users_mail_group, self).write(cr, uid, ids, vals, context=context)
 158         if vals.get('groups_id'):
 159             # form: {'group_ids': [(3, 10), (3, 3), (4, 10), (4, 3)]} or {'group_ids': [(6, 0, [ids]}
 160             user_group_ids = [command[1] for command in vals['groups_id'] if command[0] == 4]
 161             user_group_ids += [id for command in vals['groups_id'] if command[0] == 6 for id in command[2]]
 162             mail_group_obj = self.pool.get('mail.group')
 163             mail_group_ids = mail_group_obj.search(cr, uid, [('group_ids', 'in', user_group_ids)], context=context)
 164             mail_group_obj.message_subscribe_users(cr, uid, mail_group_ids, ids, context=context)
 165         return write_res
 166
 167 class res_groups_mail_group(osv.Model):
 168     """ Update of res.groups class
 169         - if adding users from a group, check mail.groups linked to this user
 170           group and subscribe them. This is done by overriding the write method.
 171     """
 172     _name = 'res.groups'
 173     _inherit = 'res.groups'
 174
 175     # FP Note: to improve, post processeing, after the super may be better
 176     def write(self, cr, uid, ids, vals, context=None):
 177         write_res = super(res_groups_mail_group, self).write(cr, uid, ids, vals, context=context)
 178         if vals.get('users'):
 179             # form: {'group_ids': [(3, 10), (3, 3), (4, 10), (4, 3)]} or {'group_ids': [(6, 0, [ids]}
 180             user_ids = [command[1] for command in vals['users'] if command[0] == 4]
 181             user_ids += [id for command in vals['users'] if command[0] == 6 for id in command[2]]
 182             mail_group_obj = self.pool.get('mail.group')
 183             mail_group_ids = mail_group_obj.search(cr, uid, [('group_ids', 'in', ids)], context=context)
 184             mail_group_obj.message_subscribe_users(cr, uid, mail_group_ids, user_ids, context=context)
 185         return write_res