103 lines
3.1 KiB
Python
103 lines
3.1 KiB
Python
# Copyright 2014 ABF OSIELL <http://osiell.com>
|
|
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
|
|
import datetime
|
|
import logging
|
|
|
|
from odoo import api, fields, models
|
|
|
|
|
|
_logger = logging.getLogger(__name__)
|
|
|
|
|
|
class ResUsersRole(models.Model):
|
|
_name = 'res.users.role'
|
|
_inherits = {'res.groups': 'group_id'}
|
|
_description = "User role"
|
|
|
|
group_id = fields.Many2one(
|
|
comodel_name='res.groups', required=True, ondelete='cascade',
|
|
readonly=True, string="Associated group")
|
|
line_ids = fields.One2many(
|
|
comodel_name='res.users.role.line',
|
|
inverse_name='role_id', string="Users")
|
|
user_ids = fields.One2many(
|
|
comodel_name='res.users', string="Users",
|
|
compute='_compute_user_ids')
|
|
group_category_id = fields.Many2one(
|
|
related='group_id.category_id',
|
|
default=lambda cls: cls.env.ref(
|
|
'base_user_role.ir_module_category_role').id)
|
|
|
|
@api.multi
|
|
@api.depends('line_ids.user_id')
|
|
def _compute_user_ids(self):
|
|
for role in self:
|
|
role.user_ids = role.line_ids.mapped('user_id')
|
|
|
|
@api.model
|
|
def create(self, vals):
|
|
new_record = super(ResUsersRole, self).create(vals)
|
|
new_record.update_users()
|
|
return new_record
|
|
|
|
@api.multi
|
|
def write(self, vals):
|
|
res = super(ResUsersRole, self).write(vals)
|
|
self.update_users()
|
|
return res
|
|
|
|
@api.multi
|
|
def unlink(self):
|
|
users = self.mapped('user_ids')
|
|
res = super(ResUsersRole, self).unlink()
|
|
users.set_groups_from_roles(force=True)
|
|
return res
|
|
|
|
@api.multi
|
|
def update_users(self):
|
|
"""Update all the users concerned by the roles identified by `ids`."""
|
|
users = self.mapped('user_ids')
|
|
users.set_groups_from_roles()
|
|
return True
|
|
|
|
@api.model
|
|
def cron_update_users(self):
|
|
logging.info("Update user roles")
|
|
self.search([]).update_users()
|
|
|
|
|
|
class ResUsersRoleLine(models.Model):
|
|
_name = 'res.users.role.line'
|
|
_description = 'Users associated to a role'
|
|
|
|
role_id = fields.Many2one(
|
|
comodel_name='res.users.role', string="Role",
|
|
ondelete='cascade')
|
|
user_id = fields.Many2one(
|
|
comodel_name='res.users', string="User")
|
|
date_from = fields.Date("From")
|
|
date_to = fields.Date("To")
|
|
is_enabled = fields.Boolean("Enabled", compute='_compute_is_enabled')
|
|
|
|
@api.multi
|
|
@api.depends('date_from', 'date_to')
|
|
def _compute_is_enabled(self):
|
|
today = datetime.date.today()
|
|
for role_line in self:
|
|
role_line.is_enabled = True
|
|
if role_line.date_from:
|
|
date_from = fields.Date.from_string(role_line.date_from)
|
|
if date_from > today:
|
|
role_line.is_enabled = False
|
|
if role_line.date_to:
|
|
date_to = fields.Date.from_string(role_line.date_to)
|
|
if today > date_to:
|
|
role_line.is_enabled = False
|
|
|
|
@api.multi
|
|
def unlink(self):
|
|
users = self.mapped('user_id')
|
|
res = super(ResUsersRoleLine, self).unlink()
|
|
users.set_groups_from_roles(force=True)
|
|
return res
|