You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
88 lines
3.2 KiB
88 lines
3.2 KiB
# Copyright 2014 ABF OSIELL <http://osiell.com>
|
|
# License LGPL-3.0 or later (http://www.gnu.org/licenses/lgpl).
|
|
from odoo import api, fields, models
|
|
|
|
|
|
class ResUsers(models.Model):
|
|
_inherit = 'res.users'
|
|
|
|
role_line_ids = fields.One2many(
|
|
comodel_name='res.users.role.line',
|
|
inverse_name='user_id',
|
|
string="Role lines",
|
|
default=lambda self: self._default_role_lines()
|
|
)
|
|
role_ids = fields.One2many(
|
|
comodel_name='res.users.role', string="Roles",
|
|
compute='_compute_role_ids')
|
|
|
|
@api.model
|
|
def _default_role_lines(self):
|
|
default_user = self.env.ref(
|
|
'base.default_user', raise_if_not_found=False)
|
|
default_values = []
|
|
if default_user:
|
|
for role_line in default_user.role_line_ids:
|
|
default_values.append({
|
|
'role_id': role_line.role_id.id,
|
|
'date_from': role_line.date_from,
|
|
'date_to': role_line.date_to,
|
|
'is_enabled': role_line.is_enabled,
|
|
})
|
|
return default_values
|
|
|
|
@api.multi
|
|
@api.depends('role_line_ids.role_id')
|
|
def _compute_role_ids(self):
|
|
for user in self:
|
|
user.role_ids = user.role_line_ids.mapped('role_id')
|
|
|
|
@api.model
|
|
def create(self, vals):
|
|
new_record = super(ResUsers, self).create(vals)
|
|
new_record.set_groups_from_roles()
|
|
return new_record
|
|
|
|
@api.multi
|
|
def write(self, vals):
|
|
res = super(ResUsers, self).write(vals)
|
|
self.sudo().set_groups_from_roles()
|
|
return res
|
|
|
|
@api.multi
|
|
def _get_enabled_roles(self):
|
|
return self.role_line_ids.filtered(
|
|
lambda rec: rec.is_enabled and
|
|
(not rec.company_id or rec.company_id == rec.user_id.company_id))
|
|
|
|
def set_groups_from_roles(self, force=False):
|
|
"""Set (replace) the groups following the roles defined on users.
|
|
If no role is defined on the user, its groups are let untouched unless
|
|
the `force` parameter is `True`.
|
|
"""
|
|
role_groups = {}
|
|
# We obtain all the groups associated to each role first, so that
|
|
# it is faster to compare later with each user's groups.
|
|
for role in self.mapped('role_line_ids.role_id'):
|
|
role_groups[role] = list(set(
|
|
role.group_id.ids + role.implied_ids.ids +
|
|
role.trans_implied_ids.ids))
|
|
for user in self:
|
|
if not user.role_line_ids and not force:
|
|
continue
|
|
group_ids = []
|
|
for role_line in user._get_enabled_roles():
|
|
role = role_line.role_id
|
|
group_ids += role_groups[role]
|
|
group_ids = list(set(group_ids)) # Remove duplicates IDs
|
|
groups_to_add = list(set(group_ids) - set(user.groups_id.ids))
|
|
groups_to_remove = list(set(user.groups_id.ids) - set(group_ids))
|
|
to_add = [(4, gr) for gr in groups_to_add]
|
|
to_remove = [(3, gr) for gr in groups_to_remove]
|
|
groups = to_remove + to_add
|
|
if groups:
|
|
vals = {
|
|
'groups_id': groups,
|
|
}
|
|
super(ResUsers, user).write(vals)
|
|
return True
|