mirror of
https://github.com/OCA/web.git
synced 2025-02-22 13:21:25 +02:00
Improve logic and add tests
This commit is contained in:
committed by
Jairo Llopis
parent
f916f2c9f1
commit
f2de5d8663
@@ -2,31 +2,45 @@
|
||||
# Copyright 2016 ACSONE SA/NV
|
||||
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
|
||||
|
||||
from openerp import api, models, _
|
||||
from openerp import api, fields, models, _
|
||||
|
||||
|
||||
class ResUsers(models.Model):
|
||||
|
||||
_inherit = 'res.users'
|
||||
|
||||
@api.multi
|
||||
@api.depends('create_date')
|
||||
def _compute_channel_names(self):
|
||||
for record in self:
|
||||
res_id = record.id
|
||||
record.notify_info_channel_name = 'notify_info_%s' % res_id
|
||||
record.notify_warning_channel_name = 'notify_warning_%s' % res_id
|
||||
|
||||
notify_info_channel_name = fields.Char(
|
||||
compute='_compute_channel_names', store=True, required=True)
|
||||
notify_warning_channel_name = fields.Char(
|
||||
compute='_compute_channel_names', store=True, requried=True)
|
||||
|
||||
@api.multi
|
||||
def notify_info(self, message, title=None, sticky=False):
|
||||
title = title or _('Information')
|
||||
self._notify_channel('notify_info', message, title, sticky)
|
||||
self._notify_channel(
|
||||
'notify_info_channel_name', message, title, sticky)
|
||||
|
||||
@api.multi
|
||||
def notify_warning(self, message, title=None, sticky=False):
|
||||
title = title or _('Warning')
|
||||
self._notify_channel('notify_warning', message, title, sticky)
|
||||
self._notify_channel(
|
||||
'notify_warning_channel_name', message, title, sticky)
|
||||
|
||||
@api.multi
|
||||
def _notify_channel(self, channel_name_prefix, message, title, sticky):
|
||||
notification = {
|
||||
def _notify_channel(self, channel_name_field, message, title, sticky):
|
||||
bus_message = {
|
||||
'message': message,
|
||||
'title': title,
|
||||
'sticky': sticky
|
||||
}
|
||||
bus_bus = self.env['bus.bus']
|
||||
for record in self:
|
||||
channel_name = channel_name_prefix + "_%s" % record.id
|
||||
bus_bus.sendone(channel_name, notification)
|
||||
notifications = [(getattr(record, channel_name_field), bus_message)
|
||||
for record in self]
|
||||
self.env['bus.bus'].sendmany(notifications)
|
||||
|
||||
Reference in New Issue
Block a user