login get_all_user接口优化 about bk-paas HOT 2 CLOSED

    def _get_user_info(self, user):
        # if fill the _role, do not select from db
        if hasattr(user, '_role'):
            role = str(user._role)
        else:
            role = str(user.role_code)

        if hasattr(user, '_wx_userid'):
            wx_userid = user._wx_userid
        else:
            wx_userid = user.wx_userid

        if hasattr(user, '_language'):
            language = user._language
        else:
            language = user.language

        if hasattr(user, '_time_zone'):
            time_zone = user._time_zone
        else:
            time_zone = user.time_zone

        return {
            'username': user.username,
            'chname': user.chname,
            'qq': '',
            'phone': user.phone,
            'email': user.email,
            'role': role,
            'wx_userid': wx_userid,
            'language': language,
            'time_zone': time_zone
        }

    def _get_user_info_v2(self, user):
        # if fill the _role, do not select from db
        if hasattr(user, '_role'):
            role = user._role
        else:
            role = user.role_code

        if hasattr(user, '_wx_userid'):
            wx_userid = user._wx_userid
        else:
            wx_userid = user.wx_userid

        if hasattr(user, '_language'):
            language = user._language
        else:
            language = user.language

        if hasattr(user, '_time_zone'):
            time_zone = user._time_zone
        else:
            time_zone = user.time_zone

        return {
            'bk_username': user.username,
            'chname': user.chname,
            'qq': '',
            'phone': user.phone,
            'email': user.email,
            'bk_role': role,
            'wx_userid': wx_userid,
            'language': language,
            'time_zone': time_zone
        }

    def choose_role_code(self, role_list):
        # 多个角色，则已最高角色为主（superuser > developer > staff）
        if RoleCodeEnum.SUPERUSER in role_list:
            return RoleCodeEnum.SUPERUSER
        if RoleCodeEnum.DEVELOPER in role_list:
            return RoleCodeEnum.DEVELOPER
        if RoleCodeEnum.OPERATOR in role_list:
            return RoleCodeEnum.OPERATOR
        if RoleCodeEnum.AUDITOR in role_list:
            return RoleCodeEnum.AUDITOR
        return RoleCodeEnum.STAFF

    def filter_by_role(self, users, role):
        from collections import defaultdict
        from bkaccount.models import BkUserRole, BkRole, UserInfo

        # make {user_id: userinfo}
        userinfos = UserInfo.objects.all()
        userinfo_dict = {u.user_id: u for u in userinfos}

        # make {role_id:role_code}
        roles = BkRole.objects.all()
        role_dict = {r.id: r.code for r in roles}

        # make { user_id:[role_id1, role_id2] }
        user_role_code_list_dict = defaultdict(list)
        for user_role in BkUserRole.objects.all():
            role_code = role_dict[user_role.role_id]
            user_role_code_list_dict[user_role.user_id].append(role_code)

        # make {user_id:role_code}
        user_role_code_dict = {}
        for user_id, role_list in user_role_code_list_dict.iteritems():
            r = self.choose_role_code(role_list)
            user_role_code_dict[user_id] = r

        # filter or not
        shouldFilter = role.isdigit() and int(role) in ROLECODE_LIST

        result = []
        for user in users:
            user._role = user_role_code_dict.get(user.id, RoleCodeEnum.STAFF)
            userinfo = userinfo_dict.get(user.id)
            if userinfo:
                user._wx_userid = (userinfo.wx_userid or '')
                user._language = (userinfo.language or '')
                user._time_zone = (userinfo.time_zone or '')
            else:
                user._wx_userid = ""
                user._language = ""
                user._time_zone = ""

            # will be filter by the role
            if shouldFilter and user._role != int(role):
                continue
            result.append(user)

        return result

    def get_all_user(self, role):
        """
        获取所有用户
        """
        users = self.all()
        users = self.filter_by_role(users, role)
        # if role.isdigit() and int(role) in ROLECODE_LIST:
        #     users = [user for user in users if user.role_code == int(role)]
        return [self._get_user_info(user) for user in users]

    def get_all_users_v2(self, role):
        """
        获取所有用户
        """
        users = self.all()
        users = self.filter_by_role(users, role)
        # if role.isdigit() and int(role) in ROLECODE_LIST:
        #     users = [user for user in users if user.role_code == int(role)]
        return [self._get_user_info_v2(user) for user in users]