fiuai-s3 0.3.6__tar.gz → 0.3.8__tar.gz

{fiuai_s3-0.3.6 → fiuai_s3-0.3.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fiuai-s3
-Version: 0.3.6
+Version: 0.3.8
 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.6 → fiuai_s3-0.3.8}/pyproject.toml

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

{fiuai_s3-0.3.6 → fiuai_s3-0.3.8}/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.2"
+__version__ = "0.3.8"
 __all__ = ["ObjectStorage", "ObjectStorageFactory", "StorageConfig", "DocFileObject", "DocSourceFrom", "DocFileType"] 

{fiuai_s3-0.3.6 → fiuai_s3-0.3.8}/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}/{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}/{object_key}"
+        return self.download_file(_path)
         
     def upload_file(self, object_key: str, data: bytes) -> bool:
         """上传文件到MinIO

{fiuai_s3-0.3.6 → fiuai_s3-0.3.8}/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:
         """上传文件到对象存储

{fiuai_s3-0.3.6 → fiuai_s3-0.3.8}/src/fiuai_s3/type.py

@@ -31,6 +31,7 @@ class DocSourceFrom(Enum):
     """
     USER = "user"
     AI = "ai"
+    IDP = "idp"
     INTEGRATION = "integration"
     ETAX = "etax"
     BANK = "bank"