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

fix: Fix some knowledge base import errors (#2739)

Browse Source
This commit is contained in:
shaohuzhang1 2025-03-31 10:29:54 +08:00 committed by GitHub
parent 4fa3fec103
commit a07df46f9d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
apps/dataset/serializers/common_serializers.py
View File

@ -40,6 +40,14 @@ def zip_dir(zip_path, output=None):
zip.close() zip.close()
def is_valid_uuid(s):
try:
uuid.UUID(s)
return True
except ValueError:
return False
def write_image(zip_path: str, image_list: List[str]): def write_image(zip_path: str, image_list: List[str]):
for image in image_list: for image in image_list:
search = re.search("\(.*\)", image) search = re.search("\(.*\)", image)
@ -47,6 +55,9 @@ def write_image(zip_path: str, image_list: List[str]):
text = search.group() text = search.group()
if text.startswith('(/api/file/'): if text.startswith('(/api/file/'):
r = text.replace('(/api/file/', '').replace(')', '') r = text.replace('(/api/file/', '').replace(')', '')
r = r.strip().split(" ")[0]
if not is_valid_uuid(r):
break
file = QuerySet(File).filter(id=r).first() file = QuerySet(File).filter(id=r).first()
if file is None: if file is None:
break break
@ -58,6 +69,9 @@ def write_image(zip_path: str, image_list: List[str]):
f.write(file.get_byte()) f.write(file.get_byte())
else: else:
r = text.replace('(/api/image/', '').replace(')', '') r = text.replace('(/api/image/', '').replace(')', '')
r = r.strip().split(" ")[0]
if not is_valid_uuid(r):
break
image_model = QuerySet(Image).filter(id=r).first() image_model = QuerySet(Image).filter(id=r).first()
if image_model is None: if image_model is None:
break break