1 # -*- coding: utf-8 -*-
2 ##############################################################################
4 # OpenERP, Open Source Management Solution
5 # Copyright (C) 2009-Today OpenERP SA (<http://www.openerp.com>)
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
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
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/>
20 ##############################################################################
22 from osv import osv, fields
23 from tools.translate import _
24 from lxml import etree
26 class res_users(osv.osv):
27 """ Update of res.users class
28 - add a preference about sending emails about notificatoins
29 - make a new user follow itself
32 _inherit = ['res.users', 'mail.thread']
33 _inherits = {'mail.alias': 'alias_id'}
36 'notification_email_pref': fields.selection([
38 ('comments', 'Only comments'),
39 ('to_me', 'Only when sent directly to me'),
41 ], 'Receive Feeds by Email', required=True,
42 help="Choose in which case you want to receive an email when you receive new feeds."),
43 'alias_id': fields.many2one('mail.alias', 'Mail Alias', ondelete="cascade", required=True),
47 'notification_email_pref': 'none',
50 def __init__(self, pool, cr):
51 """ Override of __init__ to add access rights on notification_email_pref
52 field. Access rights are disabled by default, but allowed on
53 fields defined in self.SELF_WRITEABLE_FIELDS.
55 init_res = super(res_users, self).__init__(pool, cr)
56 # duplicate list to avoid modifying the original reference
57 self.SELF_WRITEABLE_FIELDS = list(self.SELF_WRITEABLE_FIELDS)
58 self.SELF_WRITEABLE_FIELDS.append('notification_email_pref')
61 def fields_view_get(self, cr, uid, view_id=None, view_type='form', context=None, toolbar=False, submenu=False):
62 res = super(res_users,self).fields_view_get(cr, uid, view_id, view_type, context, toolbar=toolbar, submenu=submenu)
63 if view_type == 'form':
64 domain = self.pool.get("ir.config_parameter").get_param(cr, uid, "mail.catchall.domain", context=context)
66 doc = etree.XML(res['arch'])
67 alias_node = doc.xpath("//field[@name='alias_id']")[0]
68 parent = alias_node.getparent()
69 parent.remove(alias_node)
70 res['arch'] = etree.tostring(doc)
73 def create(self, cr, uid, data, context=None):
74 # create default alias same as the login
75 model_pool = self.pool.get('ir.model.data')
76 alias_pool = self.pool.get('mail.alias')
77 res_id = model_pool.get_object( cr, uid, "mail", "model_res_users")
78 data.update({'alias_name': data.get('login'),
79 'alias_model_id': res_id.id})
80 name = alias_pool.create_unique_alias(cr, uid, data, sequence=False ,context=context)
81 user_id = super(res_users, self).create(cr, uid, data, context=context)
82 user = self.browse(cr, uid, user_id, context=context)
83 alias_pool.write(cr, uid, [user.alias_id.id], {"alias_force_thread_id": user.id}, context)
84 # make user follow itself
85 self.message_subscribe(cr, uid, [user_id], [user_id], context=context)
86 # create a welcome message to broadcast
87 company_name = user.company_id.name if user.company_id else 'the company'
88 message = _('%s has joined %s! You may leave him/her a message to celebrate a new arrival in the company ! You can help him/her doing its first steps on OpenERP.') % (user.name, company_name)
89 # TODO: clean the broadcast feature. As this is not cleany specified, temporarily remove the message broadcasting that is not buggy but not very nice.
90 #self.message_broadcast(cr, uid, [user.id], 'Welcome notification', message, context=context)
93 def write(self, cr, uid, ids, vals, context=None):
94 # if login of user have been changed then change alias of user also.
95 if 'login' in vals.keys():
96 for user in self.browse(cr, uid, ids, context=context):
97 self.pool.get('mail.alias').write(cr, uid, [user.alias_id.id], {'alias_name': vals['login']}, context=context)
98 return super(res_users, self).write(cr, uid, ids, vals, context=context)
101 def message_load_ids(self, cr, uid, ids, limit=100, offset=0, domain=[], ascent=False, root_ids=[False], context=None):
102 """ Override of message_load_ids
103 User discussion page :
104 - messages posted on res.users, res_id = user.id
105 - messages directly sent to user with @user_login
109 msg_obj = self.pool.get('mail.message')
111 for user in self.browse(cr, uid, ids, context=context):
112 msg_ids += msg_obj.search(cr, uid, ['|', '|', ('body_text', 'like', '@%s' % (user.login)), ('body_html', 'like', '@%s' % (user.login)), '&', ('res_id', '=', user.id), ('model', '=', self._name)] + domain,
113 limit=limit, offset=offset, context=context)
114 if (ascent): msg_ids = self._message_add_ancestor_ids(cr, uid, ids, msg_ids, root_ids, context=context)