china_addons/oec_theme_backend/models/res_users.py

# -*- coding: utf-8 -*-

import json
from odoo import api, fields, models, tools, _
from odoo.addons.base.models.res_users import check_identity


class ResUsers(models.Model):
    _inherit = "res.users"

    lock_screen = fields.Boolean(string="Lock Screen", default=False)

    theme_id = fields.Many2one(
        "res.theme", string="Theme", store=True, domain="[('user_id', '=', id)]"
    )

    # ------------------------------------------------------------
    # main
    # ------------------------------------------------------------
    main_menu_mode = fields.Selection(related="theme_id.main_menu_mode", readonly=False)
    main_submenu_position = fields.Selection(
        related="theme_id.main_submenu_position",
        readonly=False,
    )

    # ------------------------------------------------------------
    # SideNavbar
    # ------------------------------------------------------------
    sidebar_display_number_of_submenus = fields.Boolean(
        related="theme_id.sidebar_display_number_of_submenus", readonly=False
    )
    sidebar_fixed = fields.Boolean(related="theme_id.sidebar_fixed", readonly=False)
    sidebar_show_minimize_button = fields.Boolean(related="theme_id.sidebar_show_minimize_button", readonly=False)
    sidebar_default_minimized = fields.Boolean(
        related="theme_id.sidebar_default_minimized", readonly=False
    )
    sidebar_hover_maximize = fields.Boolean(
        related="theme_id.sidebar_hover_maximize", readonly=False
    )

    @api.model
    def set_user_theme(self, uid, theme):
        """
        为当前用户设置主题。
        """
        result ={}
        try:
            user = self.browse(uid)
            user.theme_id.write(theme)
        except Exception as e:
            result = {
                "state":False,
                "title":_("Theme setting failed!"),
                "message":str(e)
            }
        else:
            result = {
                "state":True,
                "title":_("Theme set successfully!"),
                "message": _("The theme is set successfully, Click the 'Refresh' button to load the new theme.")
            }
        finally:
            return result

    @api.model_create_multi
    def create(self, vals_list):
        """
        创建新用户时，创建主题
        """
        users = super(ResUsers, self).create(vals_list)
        for new_user in users:
            new_user.theme_id = self.env["res.theme"].sudo()._get_or_create_theme(new_user.id, "user")
        return users