diff --git a/core/storage.py b/core/storage.py index a459797..707ff91 100644 --- a/core/storage.py +++ b/core/storage.py @@ -147,7 +147,7 @@ class SystemFileStorage(FileStorageInterface): return FileResponse( file_path, media_type="application/octet-stream", - headers={"Content-Disposition": content_disposition}, + headers={"Content-Disposition": content_disposition, "Content-Length": str(file_code.size)}, filename=filename # 保留原始文件名以备某些场景使用 ) @@ -329,7 +329,8 @@ class S3FileStorage(FileStorageInterface): from fastapi.responses import StreamingResponse headers = { - "Content-Disposition": f'attachment; filename="{filename.encode("utf-8").decode("latin-1")}"' + "Content-Disposition": f'attachment; filename="{filename.encode("utf-8").decode("latin-1")}"', + "Content-Length": str(file_code.size) } return StreamingResponse( stream_generator(), @@ -633,7 +634,8 @@ class OneDriveFileStorage(FileStorageInterface): yield chunk headers = { - "Content-Disposition": f'attachment; filename="{filename.encode("utf-8").decode("latin-1")}"' + "Content-Disposition": f'attachment; filename="{filename.encode("utf-8").decode("latin-1")}"', + "Content-Length": str(file_code.size) } return StreamingResponse( stream_generator(), @@ -810,7 +812,8 @@ class OpenDALFileStorage(FileStorageInterface): # 如果 reader 方法不存在,回退到全量读取(兼容旧版本) content = await self.operator.read(await file_code.get_file_path()) headers = { - "Content-Disposition": f'attachment; filename="{filename}"' + "Content-Disposition": f'attachment; filename="{filename}"', + "Content-Length": str(file_code.size) } return Response( content, headers=headers, media_type="application/octet-stream" @@ -825,7 +828,8 @@ class OpenDALFileStorage(FileStorageInterface): yield chunk headers = { - "Content-Disposition": f'attachment; filename="{filename}"' + "Content-Disposition": f'attachment; filename="{filename}"', + "Content-Length": str(file_code.size) } return StreamingResponse( stream_generator(), @@ -1038,7 +1042,8 @@ class WebDAVFileStorage(FileStorageInterface): yield chunk headers = { - "Content-Disposition": f'attachment; filename="{filename.encode("utf-8").decode()}"' + "Content-Disposition": f'attachment; filename="{filename.encode("utf-8").decode()}"', + "Content-Length": str(file_code.size) } return StreamingResponse( stream_generator(),