feat: base 优化permission缓存处理
This commit is contained in:
parent
bc77137093
commit
f347630981
|
@ -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
|
||||
|
||||
|
||||
|
|
Loading…
Reference in New Issue