From 3a268f69d61667cd8c6b0d22e316369d4206d89f Mon Sep 17 00:00:00 2001 From: shaohuzhang1 Date: Mon, 30 Dec 2024 10:49:08 +0800 Subject: [PATCH] fix: Exporting zip cannot open the file --- apps/dataset/serializers/common_serializers.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/apps/dataset/serializers/common_serializers.py b/apps/dataset/serializers/common_serializers.py index 1cb33f88f67..f908ca6bf8e 100644 --- a/apps/dataset/serializers/common_serializers.py +++ b/apps/dataset/serializers/common_serializers.py @@ -47,6 +47,8 @@ def write_image(zip_path: str, image_list: List[str]): if text.startswith('(/api/file/'): r = text.replace('(/api/file/', '').replace(')', '') file = QuerySet(File).filter(id=r).first() + if file is None: + break zip_inner_path = os.path.join('api', 'file', r) file_path = os.path.join(zip_path, zip_inner_path) if not os.path.exists(os.path.dirname(file_path)): @@ -56,6 +58,8 @@ def write_image(zip_path: str, image_list: List[str]): else: r = text.replace('(/api/image/', '').replace(')', '') image_model = QuerySet(Image).filter(id=r).first() + if image_model is None: + break zip_inner_path = os.path.join('api', 'image', r) file_path = os.path.join(zip_path, zip_inner_path) if not os.path.exists(os.path.dirname(file_path)):