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.
36 lines
1.4 KiB
36 lines
1.4 KiB
# Copyright (C) 2021 Open Source Integrators
|
|
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
|
|
|
|
from odoo import api, models
|
|
|
|
|
|
class ResUsers(models.Model):
|
|
_inherit = "res.users"
|
|
|
|
@classmethod
|
|
def authenticate(cls, db, login, password, user_agent_env):
|
|
uid = super().authenticate(db, login, password, user_agent_env)
|
|
# On login, ensure the proper roles are applied
|
|
# The last Role applied may not be the correct one,
|
|
# sonce the new session current company can be different
|
|
with cls.pool.cursor() as cr:
|
|
env = api.Environment(cr, uid, {})
|
|
if env.user.role_line_ids:
|
|
env.user.set_groups_from_roles()
|
|
return uid
|
|
|
|
def _get_enabled_roles(self):
|
|
res = super()._get_enabled_roles()
|
|
# Enable only the Roles corresponing to the currently selected company
|
|
if self.env.user.role_line_ids:
|
|
curr_company = self.env.company
|
|
res = res.filtered(
|
|
lambda x: not x.company_id or x.company_id == curr_company
|
|
)
|
|
return res
|
|
|
|
def set_groups_from_roles(self, force=False, company_id=False):
|
|
# When using the Company Switcher widget, the self.env.company is not yet set
|
|
if company_id:
|
|
self = self.with_company(company_id)
|
|
return super().set_groups_from_roles(force=force)
|