zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: add validation checks in page and list methods of KnowledgeFolder

Browse Source
This commit is contained in:
CaptainB 2025-08-04 11:10:13 +08:00
parent 0e684f2946
commit d635cc24b4
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
apps/knowledge/serializers/knowledge.py
View File

@ -171,6 +171,7 @@ class KnowledgeSerializer(serializers.Serializer):
return query_set_dict return query_set_dict
def page(self, current_page: int, page_size: int): def page(self, current_page: int, page_size: int):
self.is_valid(raise_exception=True)
folder_id = self.data.get('folder_id', self.data.get("workspace_id")) folder_id = self.data.get('folder_id', self.data.get("workspace_id"))
root = KnowledgeFolder.objects.filter(id=folder_id).first() root = KnowledgeFolder.objects.filter(id=folder_id).first()
if not root: if not root:
@ -195,6 +196,7 @@ class KnowledgeSerializer(serializers.Serializer):
) )
def list(self): def list(self):
self.is_valid(raise_exception=True)
folder_id = self.data.get('folder_id') folder_id = self.data.get('folder_id')
if not folder_id: if not folder_id:
folder_id = self.data.get('workspace_id') folder_id = self.data.get('workspace_id')