feat: base 优化permission缓存处理

This commit is contained in:
caoqianming 2024-10-09 13:57:35 +08:00
parent bc77137093
commit f347630981
1 changed files with 38 additions and 26 deletions

View File

@ -5,20 +5,32 @@ from apps.system.models import DataFilter, Dept, Permission, PostRole, UserPost,
from django.db.models.query import QuerySet
from typing import List
# 后端代码里有的权限标识
ALL_PERMS = [
]
# 数据库里定义的权限标识
def get_alld_perms(update_cache=False) -> List[str]:
key = "perms_alld_list"
perms_alld_list = cache.get(key)
if perms_alld_list is None or update_cache:
nested_list = Permission.objects.all().values_list('codes', flat=True)
perms_alld_list = list(set([item for sublist in nested_list for item in sublist]))
perms_alld_list.sort()
cache.set(key, perms_alld_list, timeout=60*5)
return perms_alld_list
def get_user_perms_map(user):
def get_user_perms_map(user, update_cache=False):
"""
获取权限字典,可用redis存取(包括功能和数据权限)
"""
key = f'perms_{str(user.id)}'
if cache.get(key) is None or update_cache:
user_perms_map = {}
if user.is_superuser:
for perm in Permission.objects.all():
if perm.codes:
for code in perm.codes:
codes = get_alld_perms()
for code in codes:
user_perms_map[code] = {}
else:
objs = UserPost.objects.filter(user=user).exclude(post=None)
@ -38,7 +50,7 @@ def get_user_perms_map(user):
user_perms_map[code][dept_id] = pr.data_range
else:
user_perms_map[code] = {dept_id: pr.data_range}
cache.set('perms_' + str(user.id), user_perms_map, timeout=300)
cache.set(key, user_perms_map, timeout=300)
return user_perms_map