99from common .exception .app_exception import AppApiException
1010from common .utils .common import get_file_content
1111from maxkb .conf import PROJECT_DIR
12- from system_manage .models import ResourceChatUserGroupAuthorize , UserGroup
12+ from system_manage .models import ResourceChatUserAuthorize , ResourceChatUserGroupAuthorize , UserGroup
1313
1414
1515class ResourceChatUserGroupQuerySerializer (serializers .Serializer ):
@@ -29,20 +29,30 @@ def get_query_set(self):
2929 resource_chat_user_group_authorize_query_set = QuerySet (ResourceChatUserGroupAuthorize ).filter (
3030 resource_id = resource_id , resource_type = resource_type
3131 )
32+ resource_chat_user_authorize_query_set = QuerySet (ResourceChatUserAuthorize ).filter (
33+ resource_id = resource_id , resource_type = resource_type
34+ )
3235 query_set = QuerySet (UserGroup )
3336 if user_group_name is not None :
3437 query_set = query_set .filter (name__contains = user_group_name )
3538 if workspace_id is not None :
3639 resource_chat_user_group_authorize_query_set = resource_chat_user_group_authorize_query_set .filter (
3740 workspace_id = workspace_id
3841 )
42+ resource_chat_user_authorize_query_set = resource_chat_user_authorize_query_set .filter (
43+ workspace_id = workspace_id
44+ )
3945 else :
4046 resource_chat_user_group_authorize_query_set = resource_chat_user_group_authorize_query_set .filter (
4147 workspace_id__isnull = True
4248 )
49+ resource_chat_user_authorize_query_set = resource_chat_user_authorize_query_set .filter (
50+ workspace_id__isnull = True
51+ )
4352 return {
4453 "default_query_set" : query_set ,
4554 "resource_chat_user_group_authorize_query_set" : resource_chat_user_group_authorize_query_set ,
55+ "resource_chat_user_authorize_query_set" : resource_chat_user_authorize_query_set ,
4656 }
4757
4858 def list (self ):
0 commit comments