import json
from django_redis import get_redis_connection
from erp_system.models import PermissionsModel


def cache_permissions_by_user(user):
    """
    把当前用户的权限信息，缓存到Redis数据库中
    Redis 中存放用户权限的结构为： user_用户ID---> 字典{key:path, value[列表] }
    """
    permission_ids = user.roles.values_list('permissions', flat=True).distinct()
    # permission_list 返回是列表和字典的嵌套，如下：[{'id':1, 'path': '/xxx/xxx'}, {'id':2, 'path': '/xxx/aaa'}]
    permission_list = PermissionsModel.objects.filter(id__in=permission_ids, is_menu=False).values('id', 'path', 'method', 'name')

    if not permission_list.exists():
        # 该用户根本就没有分配权限, 直接返回
        return None

    permission_dict = {}  # 存放权限的字典
    for permission in permission_list:  # permission是字典
        # 去除不可见字符， ’\u200b’是Unicode中的零宽度字符，可以理解为不可见字符。
        method = str(permission.get('method')).replace('\u200b', '')
        path = str(permission.get('path')).replace('\u200b', '')
        _name = str(permission.get('name')).replace('\u200b', '')
        _id = permission.get('id')


        if permission_dict.get(path):
            # 如果字典中已经存在一个相同path的权限，把新权限追加
            permission_dict[path].append({
                'method': method,
                'sign': _name,
                'id': _id
            })
        else:
            permission_dict[path] = [{
                'method': method,
                'sign': _name,
                'id': _id
            }]

    # 因为redis中的值，只能存放字节， 由字符串转换字节 。 json格式的字符串
    for key in permission_dict:
        permission_dict[key] = json.dumps(permission_dict[key])

    redis_conn = get_redis_connection('default')
    redis_conn.hmset(f'user_{user.id}', permission_dict)