feat: base 优化permission缓存处理
This commit is contained in:
parent
e1bb844b99
commit
1e82f73422
|
@ -5,20 +5,32 @@ from apps.system.models import DataFilter, Dept, Permission, PostRole, UserPost,
|
||||||
from django.db.models.query import QuerySet
|
from django.db.models.query import QuerySet
|
||||||
from typing import List
|
from typing import List
|
||||||
|
|
||||||
|
# 后端代码里有的权限标识
|
||||||
ALL_PERMS = [
|
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存取(包括功能和数据权限)
|
获取权限字典,可用redis存取(包括功能和数据权限)
|
||||||
"""
|
"""
|
||||||
|
key = f'perms_{str(user.id)}'
|
||||||
|
if cache.get(key) is None or update_cache:
|
||||||
user_perms_map = {}
|
user_perms_map = {}
|
||||||
if user.is_superuser:
|
if user.is_superuser:
|
||||||
for perm in Permission.objects.all():
|
codes = get_alld_perms()
|
||||||
if perm.codes:
|
for code in codes:
|
||||||
for code in perm.codes:
|
|
||||||
user_perms_map[code] = {}
|
user_perms_map[code] = {}
|
||||||
else:
|
else:
|
||||||
objs = UserPost.objects.filter(user=user).exclude(post=None)
|
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
|
user_perms_map[code][dept_id] = pr.data_range
|
||||||
else:
|
else:
|
||||||
user_perms_map[code] = {dept_id: pr.data_range}
|
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
|
return user_perms_map
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue