图片路径测试

Bin

2025-12-17 d616898802dfe7e5dd648bcf53c6d1f86b6d3642

 label_studio/data_export/mixins.py

@@ -349,27 +349,10 @@
            out_dir = pathlib.Path(tmp_dir) / OUT
            out_dir.mkdir(mode=0o700, parents=True, exist_ok=True)

            upload_dir = os.path.join(settings.MEDIA_ROOT, settings.UPLOAD_DIR)
            
            # 调试日志：打印关键参数
            logger.error(f'[COCO Export Debug] ========== Export Started ==========')
            logger.error(f'[COCO Export Debug] Project ID: {self.project.id}')
            logger.error(f'[COCO Export Debug] Export format: {to_format}')
            logger.error(f'[COCO Export Debug] download_resources: {download_resources}')
            logger.error(f'[COCO Export Debug] hostname: {hostname}')
            logger.error(f'[COCO Export Debug] upload_dir: {upload_dir}')
            logger.error(f'[COCO Export Debug] upload_dir exists: {os.path.exists(upload_dir)}')
            if os.path.exists(upload_dir):
                logger.error(f'[COCO Export Debug] upload_dir contents: {os.listdir(upload_dir)}')
            logger.error(f'[COCO Export Debug] tmp_dir: {tmp_dir}')
            logger.error(f'[COCO Export Debug] out_dir: {out_dir}')
            logger.error(f'[COCO Export Debug] settings.MEDIA_ROOT: {settings.MEDIA_ROOT}')
            logger.error(f'[COCO Export Debug] settings.UPLOAD_DIR: {settings.UPLOAD_DIR}')

            converter = Converter(
                config=self.project.get_parsed_config(),
                project_dir=None,
                upload_dir=upload_dir,
                upload_dir=os.getenv('LS_UPLOAD_DIR', '/data'),  # 优先使用环境变量
                download_resources=download_resources,
                # for downloading resource we need access to the API
                access_token=self.project.organization.created_by.auth_token.key,
@@ -378,47 +361,24 @@
            input_name = pathlib.Path(self.file.name).name
            input_file_path = pathlib.Path(tmp_dir) / input_name

            logger.error(f'[COCO Export Debug] input_name: {input_name}')
            logger.error(f'[COCO Export Debug] input_file_path: {input_file_path}')

            with open(input_file_path, 'wb') as file_:
                file_.write(self.file.open().read())

            logger.error(f'[COCO Export Debug] Starting converter.convert...')
            converter.convert(input_file_path, out_dir, to_format, is_dir=False)
            logger.error(f'[COCO Export Debug] Converter.convert completed')

            files = get_all_files_from_dir(out_dir)
            dirs = get_all_dirs_from_dir(out_dir)

            logger.error(f'[COCO Export Debug] Output files: {files}')
            logger.error(f'[COCO Export Debug] Output dirs: {dirs}')
            
            # 检查是否有 images 目录
            images_dir = out_dir / 'images'
            if images_dir.exists():
                image_files = list(images_dir.glob('*'))
                logger.error(f'[COCO Export Debug] images/ directory exists with {len(image_files)} files')
                logger.error(f'[COCO Export Debug] Image files: {[f.name for f in image_files]}')
            else:
                logger.error(f'[COCO Export Debug] images/ directory does NOT exist')

            if len(files) == 0 and len(dirs) == 0:
                logger.error(f'[COCO Export Debug] No files or dirs in output')
                return None
            elif len(files) == 1 and len(dirs) == 0:
                output_file = files[0]
                filename = pathlib.Path(input_name).stem + pathlib.Path(output_file).suffix
                logger.error(f'[COCO Export Debug] Single file output: {output_file}')
            else:
                logger.error(f'[COCO Export Debug] Creating zip archive...')
                shutil.make_archive(out_dir, 'zip', out_dir)
                output_file = pathlib.Path(tmp_dir) / (str(out_dir.stem) + '.zip')
                filename = pathlib.Path(input_name).stem + '.zip'
                logger.error(f'[COCO Export Debug] Zip archive created: {output_file}')

            logger.error(f'[COCO Export Debug] ========== Export Completed ===========')
            
            # TODO(jo): can we avoid the `f.read()` here?
            with open(output_file, mode='rb') as f:
                return File(