diff options
Diffstat (limited to 'addons/note/models/res_users.py')
| -rw-r--r-- | addons/note/models/res_users.py | 42 |
1 files changed, 35 insertions, 7 deletions
diff --git a/addons/note/models/res_users.py b/addons/note/models/res_users.py index be218c27b28..f4ae07e86a4 100644 --- a/addons/note/models/res_users.py +++ b/addons/note/models/res_users.py @@ -1,11 +1,14 @@ # -*- coding: utf-8 -*- # Part of Odoo. See LICENSE file for full copyright and licensing details. +import logging + from odoo import api, models, modules, _ +_logger = logging.getLogger(__name__) -class Users(models.Model): +class Users(models.Model): _name = 'res.users' _inherit = ['res.users'] @@ -14,23 +17,48 @@ class Users(models.Model): user = super(Users, self).create(values) # for new employee, create his own 5 base note stages if user.has_group('base.group_user'): - for num in range(4): - stage = self.env.ref('note.note_stage_%02d' % (num,), raise_if_not_found=False) - if stage: - stage.sudo().copy(default={'user_id': user.id}) + user._create_note_stages() return user @api.model + def _init_data_user_note_stages(self): + emp_group_id = self.env.ref('base.group_user').id + query = """ +SELECT res_users.id +FROM res_users +WHERE res_users.active IS TRUE AND EXISTS ( + SELECT 1 FROM res_groups_users_rel WHERE res_groups_users_rel.gid = %s AND res_groups_users_rel.uid = res_users.id +) AND NOT EXISTS ( + SELECT 1 FROM note_stage stage WHERE stage.user_id = res_users.id +) +GROUP BY id""" + self.env.cr.execute(query, (emp_group_id,)) + uids = [res[0] for res in self.env.cr.fetchall()] + self.browse(uids)._create_note_stages() + + def _create_note_stages(self): + data_found = False + for num in range(4): + stage = self.env.ref('note.note_stage_%02d' % (num,), raise_if_not_found=False) + data_found = True + if stage: + for user in self: + stage.sudo().copy(default={'user_id': user.id}) + if data_found: + for user in self: + _logger.info('Note default columns created for user id %s', user.id) + + @api.model def activity_user_count(self): """ If user have not scheduled any note, it will not appear in activity menu. Making note activity always visible with number of notes on label. If there is no notes, activity menu not visible for note. """ activities = super(Users, self).activity_user_count() - notes_count = self.env['note.note'].search_count([('user_id', '=', self.env.uid), ('open', '=', True)]) + notes_count = self.env['note.note'].search_count([('user_id', '=', self.env.uid)]) if notes_count: note_index = next((index for (index, a) in enumerate(activities) if a["model"] == "note.note"), None) - note_label = ("%s (%d)") % (_('Notes'), notes_count) + note_label = _('Notes') if note_index is not None: activities[note_index]['name'] = note_label else: |