fiuai-s3 0.3.7__tar.gz → 0.3.9__tar.gz

{fiuai_s3-0.3.7 → fiuai_s3-0.3.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fiuai-s3
-Version: 0.3.7
+Version: 0.3.9
 Summary: 一个支持阿里云OSS和MinIO的对象存储抽象包
 Project-URL: Homepage, https://github.com/fiuai-sz/fiuai-s3
 Project-URL: Repository, https://github.com/fiuai-sz/fiuai-s3.git

{fiuai_s3-0.3.7 → fiuai_s3-0.3.9}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "fiuai-s3"
-version = "0.3.7"
+version = "0.3.9"
 description = "一个支持阿里云OSS和MinIO的对象存储抽象包"
 readme = "README.md"
 authors = [

{fiuai_s3-0.3.7 → fiuai_s3-0.3.9}/src/fiuai_s3/__init__.py

@@ -8,5 +8,5 @@
 from .object_storage import ObjectStorage, ObjectStorageFactory, StorageConfig
 from .type import DocFileObject, DocSourceFrom, DocFileType
 
-__version__ = "0.3.7"
+__version__ = "0.3.9"
 __all__ = ["ObjectStorage", "ObjectStorageFactory", "StorageConfig", "DocFileObject", "DocSourceFrom", "DocFileType"] 

{fiuai_s3-0.3.7 → fiuai_s3-0.3.9}/src/fiuai_s3/minio/minio_storage.py

@@ -70,6 +70,26 @@ class MinioStorage(ObjectStorage):
             raise ValueError("MinIO endpoint can not contain path, format should be: host:port")
             
         return endpoint
+    
+    def upload_temp_file(self, object_key: str, data: bytes, tmppath: str = None) -> bool:
+        """上传临时文件
+        
+        Args:
+            object_key: 对象存储中的key
+            tmppath: 临时文件路径, 如果为空，则使用默认临时目录
+        """
+        _path = f"{self.config.temp_dir}/{object_key}" if not tmppath else f"{tmppath.rstrip('/')}/{object_key}"
+        return self.upload_file(_path, data)
+
+    def download_temp_file(self, object_key: str, tmppath: str = None) -> bool:
+        """下载临时文件
+        
+        Args:
+            object_key: 对象存储中的key
+            tmppath: 临时文件路径, 如果为空，则使用默认临时目录
+        """
+        _path = f"{self.config.temp_dir}/{object_key}" if not tmppath else f"{tmppath.rstrip('/')}/{object_key}"
+        return self.download_file(_path)
         
     def upload_file(self, object_key: str, data: bytes) -> bool:
         """上传文件到MinIO

{fiuai_s3-0.3.7 → fiuai_s3-0.3.9}/src/fiuai_s3/object_storage.py

@@ -47,6 +47,26 @@ class ObjectStorage(ABC):
         self.auth_company_id = auth_company_id
         self.doc_id = doc_id
     
+    @abstractmethod
+    def upload_temp_file(self, object_key: str, data: bytes, tmppath: str = None) -> bool:
+        """上传临时文件
+        
+        Args:
+            object_key: 对象存储中的key
+            tmppath: 临时文件路径, 如果为空，则使用默认临时目录
+        """
+        pass
+
+    @abstractmethod
+    def download_temp_file(self, object_key: str, tmppath: str = None) -> bool:
+        """下载临时文件
+        
+        Args:
+            object_key: 对象存储中的key
+            tmppath: 临时文件路径, 如果为空，则使用默认临时目录
+        """
+        pass
+
     @abstractmethod
     def upload_file(self, object_key: str, data: bytes) -> bool:
         """上传文件到对象存储
@@ -177,7 +197,7 @@ class ObjectStorageFactory:
             endpoint=endpoint,
             access_key=access_key,
             secret_key=secret_key,
-            temp_dir=temp_dir,
+            temp_dir=temp_dir.rstrip("/").lstrip("/"),
             use_https=use_https
         )
         cls._instance = None