fiuai-s3 0.3.9__py3-none-any.whl → 0.4.2__py3-none-any.whl

fiuai_s3/__init__.py

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

fiuai_s3/alicloud/alicloud_storage.py

@@ -149,4 +149,107 @@ class AliCloudStorage(ObjectStorage):
             return files
         except Exception as e:
             logger.error(f"列出单据文件失败: {str(e)}")
-            return [] 
+            return []
+
+    def generate_presigned_url(self, object_key: str, method: str = "GET", expires_in: int = 3600, 
+                              response_headers: Optional[Dict[str, str]] = None,
+                              auth_tenant_id: Optional[str] = None, 
+                              auth_company_id: Optional[str] = None, 
+                              doc_id: Optional[str] = None) -> Optional[str]:
+        """
+        生成预签名URL
+        
+        Args:
+            object_key: 对象存储中的key
+            method: HTTP方法，支持 GET、PUT、POST、DELETE
+            expires_in: 过期时间（秒），默认3600秒（1小时）
+            response_headers: 响应头设置
+            auth_tenant_id: 租户ID（可选，若不传则用实例属性）
+            auth_company_id: 公司ID（可选，若不传则用实例属性）
+            doc_id: 单据ID（可选，若不传则用实例属性）
+            
+        Returns:
+            Optional[str]: 预签名URL，失败时返回None
+        """
+        try:
+            # 验证参数
+            if not object_key:
+                raise ValueError("object_key 不能为空")
+            
+            if method.upper() not in ["GET", "PUT", "POST", "DELETE"]:
+                raise ValueError(f"不支持的HTTP方法: {method}")
+            
+            if expires_in <= 0 or expires_in > 604800:  # 最大7天
+                raise ValueError("过期时间必须在1秒到604800秒（7天）之间")
+            
+            # 生成预签名URL
+            from datetime import datetime, timedelta
+            
+            if method.upper() == "GET":
+                url = self.bucket.sign_url(
+                    method='GET',
+                    key=object_key,
+                    expires=expires_in,
+                    headers=response_headers
+                )
+            elif method.upper() == "PUT":
+                url = self.bucket.sign_url(
+                    method='PUT',
+                    key=object_key,
+                    expires=expires_in
+                )
+            elif method.upper() == "POST":
+                url = self.bucket.sign_url(
+                    method='POST',
+                    key=object_key,
+                    expires=expires_in
+                )
+            elif method.upper() == "DELETE":
+                url = self.bucket.sign_url(
+                    method='DELETE',
+                    key=object_key,
+                    expires=expires_in
+                )
+            
+            logger.info(f"生成预签名URL成功: {object_key}, method: {method}")
+            return url
+            
+        except Exception as e:
+            logger.error(f"生成预签名URL失败: {str(e)}")
+            return None
+
+    def generate_presigned_doc_url(self, filename: str, method: str = "GET", expires_in: int = 3600,
+                                   response_headers: Optional[Dict[str, str]] = None,
+                                   auth_tenant_id: Optional[str] = None, 
+                                   auth_company_id: Optional[str] = None, 
+                                   doc_id: Optional[str] = None) -> Optional[str]:
+        """
+        生成单据文件的预签名URL
+        
+        Args:
+            filename: 文件名
+            method: HTTP方法，支持 GET、PUT、POST、DELETE
+            expires_in: 过期时间（秒），默认3600秒（1小时）
+            response_headers: 响应头设置
+            auth_tenant_id: 租户ID（可选，若不传则用实例属性）
+            auth_company_id: 公司ID（可选，若不传则用实例属性）
+            doc_id: 单据ID（可选，若不传则用实例属性）
+            
+        Returns:
+            Optional[str]: 预签名URL，失败时返回None
+        """
+        try:
+            # 构建单据文件路径
+            object_key = self._build_doc_path(filename, auth_tenant_id, auth_company_id, doc_id)
+            
+            # 调用通用预签名URL生成方法
+            return self.generate_presigned_url(
+                object_key=object_key,
+                method=method,
+                expires_in=expires_in,
+                response_headers=response_headers
+            )
+            
+        except Exception as e:
+            logger.error(f"生成单据文件预签名URL失败: {str(e)}")
+            return None 

fiuai_s3/minio/minio_storage.py

@@ -12,7 +12,7 @@ from minio import Minio
 from minio.error import S3Error
 from minio.commonconfig import Tags
 from ..object_storage import ObjectStorage, StorageConfig
-from ..type import DocFileObject, DocFileType, DocSourceFrom
+from ..type import DocFileObject
 
 logger = logging.getLogger(__name__)
 
@@ -186,8 +186,6 @@ class MinioStorage(ObjectStorage):
     def upload_doc_file(self, 
                         filename: str, 
                         data: bytes, 
-                        doc_source_from: Optional[DocSourceFrom] = None, 
-                        doc_file_type: Optional[DocFileType] = None, 
                         tags: Optional[Dict[str, str]] = None,
                         auth_tenant_id: Optional[str] = None, 
                         auth_company_id: Optional[str] = None, 
@@ -199,13 +197,9 @@ class MinioStorage(ObjectStorage):
             object_key = self._build_doc_path(filename, auth_tenant_id, auth_company_id, doc_id)
             # 将tags转换为Tags对象
             tags_obj = Tags()
-            for k, v in tags.items():
-                tags_obj[k] = v
-
-            doc_source_from = doc_source_from if doc_source_from else DocSourceFrom.OTHER
-            doc_file_type = doc_file_type if doc_file_type else DocFileType.OTHER
-            tags_obj["doc_source_from"] = doc_source_from.value
-            tags_obj["doc_file_type"] = doc_file_type.value
+            if tags:
+                for k, v in tags.items():
+                    tags_obj[k] = v
             
             self.client.put_object(
                 bucket_name=self.bucket_name,
@@ -263,4 +257,110 @@ class MinioStorage(ObjectStorage):
             return files
         except S3Error as e:
             logger.error(f"list doc files failed: {str(e)}")
-            return []
+            return []
+
+    def generate_presigned_url(self, object_key: str, method: str = "GET", expires_in: int = 3600, 
+                              response_headers: Optional[Dict[str, str]] = None,
+                              auth_tenant_id: Optional[str] = None, 
+                              auth_company_id: Optional[str] = None, 
+                              doc_id: Optional[str] = None) -> Optional[str]:
+        """
+        生成预签名URL
+        
+        Args:
+            object_key: 对象存储中的key
+            method: HTTP方法，支持 GET、PUT、POST、DELETE
+            expires_in: 过期时间（秒），默认3600秒（1小时）
+            response_headers: 响应头设置
+            auth_tenant_id: 租户ID（可选，若不传则用实例属性）
+            auth_company_id: 公司ID（可选，若不传则用实例属性）
+            doc_id: 单据ID（可选，若不传则用实例属性）
+            
+        Returns:
+            Optional[str]: 预签名URL，失败时返回None
+        """
+        try:
+            # 验证参数
+            if not object_key:
+                raise ValueError("object_key 不能为空")
+            
+            if method.upper() not in ["GET", "PUT", "POST", "DELETE"]:
+                raise ValueError(f"不支持的HTTP方法: {method}")
+            
+            if expires_in <= 0 or expires_in > 604800:  # 最大7天
+                raise ValueError("过期时间必须在1秒到604800秒（7天）之间")
+            
+            # 生成预签名URL
+            from datetime import timedelta
+            
+            if method.upper() == "GET":
+                url = self.client.presigned_get_object(
+                    bucket_name=self.bucket_name,
+                    object_name=object_key,
+                    expires=timedelta(seconds=expires_in),
+                    response_headers=response_headers
+                )
+            elif method.upper() == "PUT":
+                url = self.client.presigned_put_object(
+                    bucket_name=self.bucket_name,
+                    object_name=object_key,
+                    expires=timedelta(seconds=expires_in)
+                )
+            elif method.upper() == "POST":
+                url = self.client.presigned_post_policy(
+                    bucket_name=self.bucket_name,
+                    object_name=object_key,
+                    expires=timedelta(seconds=expires_in)
+                )
+            elif method.upper() == "DELETE":
+                url = self.client.presigned_delete_object(
+                    bucket_name=self.bucket_name,
+                    object_name=object_key,
+                    expires=timedelta(seconds=expires_in)
+                )
+            
+            logger.info(f"生成预签名URL成功: {object_key}, method: {method}")
+            return url
+            
+        except S3Error as e:
+            logger.error(f"生成预签名URL失败: {str(e)}")
+            return None
+        except Exception as e:
+            logger.error(f"生成预签名URL异常: {str(e)}")
+            return None
+
+    def generate_presigned_doc_url(self, filename: str, method: str = "GET", expires_in: int = 3600,
+                                   response_headers: Optional[Dict[str, str]] = None,
+                                   auth_tenant_id: Optional[str] = None, 
+                                   auth_company_id: Optional[str] = None, 
+                                   doc_id: Optional[str] = None) -> Optional[str]:
+        """
+        生成单据文件的预签名URL
+        
+        Args:
+            filename: 文件名
+            method: HTTP方法，支持 GET、PUT、POST、DELETE
+            expires_in: 过期时间（秒），默认3600秒（1小时）
+            response_headers: 响应头设置
+            auth_tenant_id: 租户ID（可选，若不传则用实例属性）
+            auth_company_id: 公司ID（可选，若不传则用实例属性）
+            doc_id: 单据ID（可选，若不传则用实例属性）
+            
+        Returns:
+            Optional[str]: 预签名URL，失败时返回None
+        """
+        try:
+            # 构建单据文件路径
+            object_key = self._build_doc_path(filename, auth_tenant_id, auth_company_id, doc_id)
+            
+            # 调用通用预签名URL生成方法
+            return self.generate_presigned_url(
+                object_key=object_key,
+                method=method,
+                expires_in=expires_in,
+                response_headers=response_headers
+            )
+            
+        except Exception as e:
+            logger.error(f"生成单据文件预签名URL失败: {str(e)}")
+            return None

fiuai_s3/object_storage.py

@@ -120,8 +120,6 @@ class ObjectStorage(ABC):
     def upload_doc_file(self, 
                         filename: str, 
                         data: bytes, 
-                        doc_source_from: Optional[DocSourceFrom] = None, 
-                        doc_file_type: Optional[DocFileType] = None, 
                         tags: Optional[Dict[str, str]] = None,
                         auth_tenant_id: Optional[str] = None, auth_company_id: Optional[str] = None, doc_id: Optional[str] = None) -> bool:
         """
@@ -165,6 +163,52 @@ class ObjectStorage(ABC):
         """
         pass
 
+    @abstractmethod
+    def generate_presigned_url(self, object_key: str, method: str = "GET", expires_in: int = 3600, 
+                              response_headers: Optional[Dict[str, str]] = None,
+                              auth_tenant_id: Optional[str] = None, 
+                              auth_company_id: Optional[str] = None, 
+                              doc_id: Optional[str] = None) -> Optional[str]:
+        """
+        生成预签名URL
+        
+        Args:
+            object_key: 对象存储中的key，如果为空则使用单据路径
+            method: HTTP方法，支持 GET、PUT、POST、DELETE
+            expires_in: 过期时间（秒），默认3600秒（1小时）
+            response_headers: 响应头设置
+            auth_tenant_id: 租户ID（可选，若不传则用实例属性）
+            auth_company_id: 公司ID（可选，若不传则用实例属性）
+            doc_id: 单据ID（可选，若不传则用实例属性）
+            
+        Returns:
+            Optional[str]: 预签名URL，失败时返回None
+        """
+        pass
+
+    @abstractmethod
+    def generate_presigned_doc_url(self, filename: str, method: str = "GET", expires_in: int = 3600,
+                                   response_headers: Optional[Dict[str, str]] = None,
+                                   auth_tenant_id: Optional[str] = None, 
+                                   auth_company_id: Optional[str] = None, 
+                                   doc_id: Optional[str] = None) -> Optional[str]:
+        """
+        生成单据文件的预签名URL
+        
+        Args:
+            filename: 文件名
+            method: HTTP方法，支持 GET、PUT、POST、DELETE
+            expires_in: 过期时间（秒），默认3600秒（1小时）
+            response_headers: 响应头设置
+            auth_tenant_id: 租户ID（可选，若不传则用实例属性）
+            auth_company_id: 公司ID（可选，若不传则用实例属性）
+            doc_id: 单据ID（可选，若不传则用实例属性）
+            
+        Returns:
+            Optional[str]: 预签名URL，失败时返回None
+        """
+        pass
+
 class ObjectStorageFactory:
     """对象存储工厂类"""
     
@@ -255,6 +299,68 @@ class ObjectStorageFactory:
             return MinioStorage(config, auth_tenant_id, auth_company_id, doc_id)
         else:
             raise ValueError(f"不支持的存储提供商: {provider}")
+    
+    @classmethod
+    def generate_presigned_url(cls, object_key: str, method: str = "GET", expires_in: int = 3600, 
+                              response_headers: Optional[Dict[str, str]] = None,
+                              auth_tenant_id: Optional[str] = None, 
+                              auth_company_id: Optional[str] = None, 
+                              doc_id: Optional[str] = None) -> Optional[str]:
+        """通过工厂类生成预签名URL
+        
+        Args:
+            object_key: 对象存储中的key
+            method: HTTP方法，支持 GET、PUT、POST、DELETE
+            expires_in: 过期时间（秒），默认3600秒（1小时）
+            response_headers: 响应头设置
+            auth_tenant_id: 租户ID（可选）
+            auth_company_id: 公司ID（可选）
+            doc_id: 单据ID（可选）
+            
+        Returns:
+            Optional[str]: 预签名URL，失败时返回None
+        """
+        storage = cls.get_instance()
+        return storage.generate_presigned_url(
+            object_key=object_key,
+            method=method,
+            expires_in=expires_in,
+            response_headers=response_headers,
+            auth_tenant_id=auth_tenant_id,
+            auth_company_id=auth_company_id,
+            doc_id=doc_id
+        )
+    
+    @classmethod
+    def generate_presigned_doc_url(cls, filename: str, method: str = "GET", expires_in: int = 3600,
+                                   response_headers: Optional[Dict[str, str]] = None,
+                                   auth_tenant_id: Optional[str] = None, 
+                                   auth_company_id: Optional[str] = None, 
+                                   doc_id: Optional[str] = None) -> Optional[str]:
+        """通过工厂类生成单据文件的预签名URL
+        
+        Args:
+            filename: 文件名
+            method: HTTP方法，支持 GET、PUT、POST、DELETE
+            expires_in: 过期时间（秒），默认3600秒（1小时）
+            response_headers: 响应头设置
+            auth_tenant_id: 租户ID（可选）
+            auth_company_id: 公司ID（可选）
+            doc_id: 单据ID（可选）
+            
+        Returns:
+            Optional[str]: 预签名URL，失败时返回None
+        """
+        storage = cls.get_instance()
+        return storage.generate_presigned_doc_url(
+            filename=filename,
+            method=method,
+            expires_in=expires_in,
+            response_headers=response_headers,
+            auth_tenant_id=auth_tenant_id,
+            auth_company_id=auth_company_id,
+            doc_id=doc_id
+        )
 
 # 导出工厂类
 __all__ = ['ObjectStorage', 'ObjectStorageFactory', 'StorageConfig']

{fiuai_s3-0.3.9.dist-info → fiuai_s3-0.4.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fiuai-s3
-Version: 0.3.9
+Version: 0.4.2
 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.4.2.dist-info/RECORD

@@ -0,0 +1,11 @@
+fiuai_s3/__init__.py,sha256=BZvQEEefO33hqV8ZcT0HSLtW0XKsw8KpoxBinHTIlzo,421
+fiuai_s3/object_storage.py,sha256=1vPiFFK1eb9_tF-fSn2_CBqpWtbm5WsgSvGk3VEOvvo,14169
+fiuai_s3/type.py,sha256=L1h3SQh7_NhWfjWPa1MMgGoxM7g6wQF26qg5WsERt6U,981
+fiuai_s3/alicloud/__init__.py,sha256=mmrCrFp5DzRF5fViJDq7_LpsqCViwTPXOPz4qoaSscI,218
+fiuai_s3/alicloud/alicloud_storage.py,sha256=DieaILMmPXYU8-xbZIZbdkmju3PXEVDWXAQ7qAtuJdg,10417
+fiuai_s3/minio/__init__.py,sha256=hOmpUkTq8TPgYOxfeOMcWq1_YsJ2r8Zf9VTX3w_v1Lk,209
+fiuai_s3/minio/minio_storage.py,sha256=z_LvuL_CGr0bisCF9vYGwHdqJbohYi9qQFevGV4JhyM,14281
+fiuai_s3-0.4.2.dist-info/METADATA,sha256=0KxOv7GE5LS5N0zPxcTvx63oAR2dP0FjhSHpeMNc8XE,18727
+fiuai_s3-0.4.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+fiuai_s3-0.4.2.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+fiuai_s3-0.4.2.dist-info/RECORD,,

fiuai_s3-0.3.9.dist-info/RECORD

@@ -1,11 +0,0 @@
-fiuai_s3/__init__.py,sha256=tLlqnMEl4K7R4jg2GbvGZanRsis1vm7KFKSc5R_wRUE,421
-fiuai_s3/object_storage.py,sha256=nSXpB39Xt6RGg_QR4qU8rs7g8ZNwLjhqi9d5wOi4iTo,9479
-fiuai_s3/type.py,sha256=L1h3SQh7_NhWfjWPa1MMgGoxM7g6wQF26qg5WsERt6U,981
-fiuai_s3/alicloud/__init__.py,sha256=mmrCrFp5DzRF5fViJDq7_LpsqCViwTPXOPz4qoaSscI,218
-fiuai_s3/alicloud/alicloud_storage.py,sha256=d24AFYDs7RsTsfL9IiVOxYlGyLzsaUa_WUYD5VPLP4s,6035
-fiuai_s3/minio/__init__.py,sha256=hOmpUkTq8TPgYOxfeOMcWq1_YsJ2r8Zf9VTX3w_v1Lk,209
-fiuai_s3/minio/minio_storage.py,sha256=-bkxf6GSfPRpRIBbUzb1rHB8YAXqW1EuorLpjXrEV_c,10003
-fiuai_s3-0.3.9.dist-info/METADATA,sha256=y0Ca-vsvRWe9uVJ_Y4QV7sFaC-9xZ-OIg076h9eCKrk,18727
-fiuai_s3-0.3.9.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-fiuai_s3-0.3.9.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-fiuai_s3-0.3.9.dist-info/RECORD,,