131 lines
5.7 KiB
Python
131 lines
5.7 KiB
Python
# -*- encoding: utf-8 -*-
|
|
##############################################################################
|
|
#
|
|
# OpenERP, Open Source Management Solution
|
|
# Copyright (C) 2011 Smile (<http://www.smile.fr>). All Rights Reserved
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
##############################################################################
|
|
|
|
from openerp import api, fields, models, SUPERUSER_ID, _
|
|
from openerp.exceptions import Warning
|
|
|
|
|
|
class ResUsers(models.Model):
|
|
_inherit = 'res.users'
|
|
|
|
@api.one
|
|
def _is_share(self, name, args):
|
|
return (self.id, self.user_profile or not self.has_group('base.group_user'))
|
|
|
|
@api.model
|
|
def _setup_fields(self):
|
|
super(ResUsers, self)._setup_fields()
|
|
self._fields['share'].column._fnct = ResUsers._is_share
|
|
|
|
@api.model
|
|
def _get_default_field_ids(self):
|
|
return self.env['ir.model.fields'].search([
|
|
('model', 'in', ('res.users', 'res.partner')),
|
|
('name', 'in', ('action_id', 'menu_id', 'groups_id')),
|
|
]).ids
|
|
|
|
user_profile = fields.Boolean('Is User Profile')
|
|
user_profile_id = fields.Many2one('res.users', 'User Profile', domain=[('id', '!=', SUPERUSER_ID), ('user_profile', '=', True)],
|
|
context={'active_test': False})
|
|
user_ids = fields.One2many('res.users', 'user_profile_id', 'Users', domain=[('user_profile', '=', False)])
|
|
field_ids = fields.Many2many('ir.model.fields', 'res_users_fields_rel', 'user_id', 'field_id', 'Fields to update',
|
|
domain=[('model', 'in', ('res.users', 'res.partner')),
|
|
('ttype', 'not in', ('one2many',)),
|
|
('name', 'not in', ('user_profile', 'user_profile_id', 'user_ids', 'field_ids', 'view'))],
|
|
default=_get_default_field_ids)
|
|
is_update_users = fields.Boolean(string="Update users", default=lambda *a: True,
|
|
help="If false, users associated to this profile will be not updated after creation")
|
|
|
|
_sql_constraints = [
|
|
('active_admin_check', 'CHECK (id = 1 AND active = TRUE OR id <> 1)', 'The user with id = 1 must be always active!'),
|
|
('profile_without_profile_id', 'CHECK( (user_profile = TRUE AND user_profile_id IS NULL) OR user_profile = FALSE )',
|
|
'Profile users cannot be linked to a profile!'),
|
|
]
|
|
|
|
@api.one
|
|
@api.constrains('user_profile_id')
|
|
def _check_user_profile_id(self):
|
|
admin = self.env.ref('base.user_root')
|
|
if self.user_profile_id == admin:
|
|
raise Warning(_("You can't use %s as user profile !") % admin.name)
|
|
|
|
@api.onchange('user_profile')
|
|
def onchange_user_profile(self):
|
|
if self.user_profile:
|
|
self.active = self.id == SUPERUSER_ID
|
|
self.user_profile_id = False
|
|
|
|
@api.multi
|
|
def _update_from_profile(self, fields=None):
|
|
if not self:
|
|
return
|
|
if len(self.mapped('user_profile_id')) != 1:
|
|
raise Warning(_("_update_from_profile accepts only users linked to a same profile"))
|
|
user_profile = self[0].user_profile_id
|
|
if not fields:
|
|
fields = user_profile.field_ids.mapped('name')
|
|
else:
|
|
fields = set(fields) & set(user_profile.field_ids.mapped('name'))
|
|
if user_profile:
|
|
vals = {}
|
|
for field in fields:
|
|
value = getattr(user_profile, field)
|
|
field_type = self._fields[field].type
|
|
if field_type == 'many2one':
|
|
vals[field] = value.id
|
|
elif field_type == 'many2many':
|
|
vals[field] = [(6, 0, value.ids)]
|
|
elif field_type == 'one2many':
|
|
raise Warning(_("_update_from_profile doesn't manage fields.One2many"))
|
|
else:
|
|
vals[field] = value
|
|
if vals:
|
|
self.write(vals)
|
|
|
|
@api.multi
|
|
def _update_users_linked_to_profile(self, fields=None):
|
|
for user_profile in self.filtered(lambda user: user.user_profile and user.is_update_users):
|
|
user_profile.with_context(active_test=False).mapped('user_ids')._update_from_profile(fields)
|
|
|
|
@api.model
|
|
def create(self, vals):
|
|
record = super(ResUsers, self).create(vals)
|
|
if record.user_profile_id:
|
|
record._update_from_profile()
|
|
return record
|
|
|
|
@api.multi
|
|
def write(self, vals):
|
|
if vals.get('user_profile_id'):
|
|
users_to_update = self.filtered(lambda user: user.user_profile_id.id != vals['user_profile_id'])
|
|
vals = self._remove_reified_groups(vals)
|
|
res = super(ResUsers, self).write(vals)
|
|
if vals.get('user_profile_id'):
|
|
users_to_update._update_from_profile()
|
|
else:
|
|
self._update_users_linked_to_profile(vals.keys())
|
|
return res
|
|
|
|
def copy_data(self, cr, uid, user_id, default=None, context=None):
|
|
default = default.copy() if default else {}
|
|
default['user_ids'] = []
|
|
return super(ResUsers, self).copy_data(cr, uid, user_id, default, context)
|