Class S3FileStorage

core/storage.py:260–563

class S3FileStorage(FileStorageInterface):
    def __init__(self):
        self.access_key_id = settings.s3_access_key_id
        self.secret_access_key = settings.s3_secret_access_key
        self.bucket_name = settings.s3_bucket_name
        self.s3_hostname = settings.s3_hostname
        self.region_name = settings.s3_region_name
        self.signature_version = settings.s3_signature_version
        self.endpoint_url = settings.s3_endpoint_url or f"https://{self.s3_hostname}"
        self.aws_session_token = settings.aws_session_token
        self.addressing_style = str(settings.s3_addressing_style or "auto").lower()
        self.proxy = settings.s3_proxy
        self.session = aioboto3.Session(
            aws_access_key_id=self.access_key_id,
            aws_secret_access_key=self.secret_access_key,
        )
        if not settings.s3_endpoint_url:
            self.endpoint_url = f"https://{self.s3_hostname}"
        else:
            # 如果提供了 s3_endpoint_url，则优先使用它
            self.endpoint_url = settings.s3_endpoint_url

    def _client_config(self) -> Config:
        config = {"signature_version": self.signature_version}
        s3_config = {}
        if self.addressing_style in {"path", "virtual", "auto"}:
            s3_config["addressing_style"] = self.addressing_style
        if s3_config:
            config["s3"] = s3_config
        return Config(**config)

    def _client(self):
        return self.session.client(
            "s3",
            endpoint_url=self.endpoint_url,
            aws_session_token=self.aws_session_token,
            region_name=self.region_name,
            config=self._client_config(),
        )

    async def save_file(self, file: UploadFile, save_path: str):
        async with self._client() as s3:
            # 使用 upload_fileobj 流式上传，避免将整个文件加载到内存
            await s3.upload_fileobj(
                file.file,
                self.bucket_name,
                save_path,
                ExtraArgs={"ContentType": file.content_type or "application/octet-stream"},
            )

    async def delete_file(self, file_code: FileCodes):
        async with self._client() as s3:
            await s3.delete_object(
                Bucket=self.bucket_name, Key=await file_code.get_file_path()
            )

    async def get_file_response(self, file_code: FileCodes):
        try: