o i,@sdZddlmZddlmZddlmZmZddlZddl m Z ddl Z ddl m Z eeZGdd d Zdad efd d ZdS) uH MinIO 对象存储服务 用于存储和管理税务文档原始文件 )Minio)S3Error)OptionalBinaryION) timedelta) StorageErrorc @sLeZdZdZ    d/dedededefd d Zd ed d fddZ d0d ededede ded ef ddZ d0d edededed ef ddZ d ededed efddZ d eded e efddZd eded efddZd eded efd d!Zed"d#fd eded$ed e efd%d&Z ' (d1d ed)ed*ed efd+d,Zd ed)ed e fd-d.Zd S)2 MinioServiceuMinIO 对象存储服务类localhost:9000 minioadminFendpoint access_key secret_keysecurec Csl||_||_||_||_zt||||d|_td|WdSty5}z t d|d}~ww)u 初始化 MinIO 客户端 Args: endpoint: MinIO 服务器地址 access_key: 访问密钥 secret_key: 密钥 secure: 是否使用 HTTPS r r r rz!Successfully connected to MinIO: zFailed to connect to MinIO: N) r r r rrclientloggerinfo Exceptionerror)selfr r r rerB/lsinfo/ai/hellotax_ai/base_platform/app/services/storage/minio.py__init__s"zMinioService.__init__ bucket_namereturnNc Csz"|j|s|j|td|WdStd|WdStyD}ztd|d|tdt||dd}~ww)u 确保存储桶存在，如果不存在则创建 Args: bucket_name: 存储桶名称 Raises: StorageError: 如果创建存储桶失败 zCreated bucket: zBucket already exists: zFailed to create bucket : zFailed to create bucket: bucketN) r bucket_exists make_bucketrrrrrstr)rrrrrr ensure_bucket4s  zMinioService.ensure_bucketapplication/octet-stream object_name file_data file_size content_typec Cstz|||jj|||||dtd|d|WdSty9}ztd|WYd}~dSd}~ww)u: 上传文件到 MinIO Args: bucket_name: 存储桶名称 object_name: 对象名称（文件路径） file_data: 文件数据流 file_size: 文件大小 content_type: 文件类型 Returns: 是否上传成功 )rr$datalengthr'File uploaded successfully: /TFile upload failed: NF)r"r put_objectrrrr)rrr$r%r&r'rrrr upload_fileHs  zMinioService.upload_file file_pathc Csrz|||jj||||dtd|d|WdSty8}ztd|WYd}~dSd}~ww)u( 从本地路径上传文件到 MinIO Args: bucket_name: 存储桶名称 object_name: 对象名称（文件路径） file_path: 本地文件路径 content_type: 文件类型 Returns: 是否上传成功 )rr$r/r'r*r+Tr,NF)r"r fput_objectrrrr)rrr$r/r'rrrrupload_file_from_pathos z"MinioService.upload_file_from_pathc Cslz|jj|||dtd|d|d|WdSty5}ztd|WYd}~dSd}~ww) u 从 MinIO 下载文件到本地 Args: bucket_name: 存储桶名称 object_name: 对象名称（文件路径） file_path: 本地保存路径 Returns: 是否下载成功 )rr$r/zFile downloaded successfully: r+z -> TzFile download failed: NF)r fget_objectrrrr)rrr$r/rrrr download_fileszMinioService.download_filec Cs|z"|jj||d}|}||td|d||WSty=}ztd|WYd}~dSd}~ww)u 获取文件数据（不保存到本地） Args: bucket_name: 存储桶名称 object_name: 对象名称（文件路径） Returns: 文件数据，失败返回 None rr$z"File data retrieved successfully: r+zFailed to get file data: N) r get_objectreadclose release_connrrrr)rrr$responser(rrrr get_file_dataszMinioService.get_file_datac Csdz|jj||dtd|d|WdSty1}ztd|WYd}~dSd}~ww)u 删除文件 Args: bucket_name: 存储桶名称 object_name: 对象名称（文件路径） Returns: 是否删除成功 r4zFile deleted successfully: r+TzFile deletion failed: NF)r remove_objectrrrr)rrr$rrrr delete_fileszMinioService.delete_filecCs,z |jj||dWdStyYdSw)u 检查文件是否存在 Args: bucket_name: 存储桶名称 object_name: 对象名称（文件路径） Returns: 文件是否存在 r4TF)r stat_objectr)rrr$rrr file_existss zMinioService.file_exists)hoursexpiresc Csfz|jj|||d}td|d||WSty2}ztd|WYd}~dSd}~ww)u 获取文件的预签名 URL（用于临时访问） Args: bucket_name: 存储桶名称 object_name: 对象名称（文件路径） expires: 过期时间 Returns: 预签名 URL，失败返回 None )rr$rAzGenerated presigned URL: r+z"Failed to generate presigned URL: N)rpresigned_get_objectrrrr)rrr$rAurlrrrrget_presigned_urlszMinioService.get_presigned_urlTprefix recursivec Cs|z!|jj|||d}g}|D]}||j|j|j|jdq|WSty=}zt d|gWYd}~Sd}~ww)u 列出存储桶中的对象 Args: bucket_name: 存储桶名称 prefix: 对象名称前缀（用于过滤） recursive: 是否递归列出 Returns: 对象列表 rrFrG)r$size last_modifiedetagzFailed to list objects: N) r list_objectsappendr$rIrJrKrrr)rrrFrGobjectsresultobjrrrrrLs( zMinioService.list_objectsc s\ddlmzd|jj||dd}fdd|D}|s)td|d|WdSt|j||}|rVd d|D}td t |d |t d t |d ||d t |}td|d|d||WSt yrt y}z0t |dddkrtd|d|WYd}~dStd|d|t dt ||d d}~ww)u 按前缀批量删除 MinIO 对象 Args: bucket_name: 存储桶名称 prefix: 对象名称前缀，如 "tenant_3/kb_11/" Returns: 删除的对象数量 r DeleteObjectTrHcsg|]}|jqSrr$).0rPrQrr Vsz1MinioService.delete_by_prefix..zNo objects found with prefix: r+cSsg|]}|jqSrrS)rTrrrrrU]szFailed to delete z objects: z objects under prefix rzDeleted z objects with prefix: coderE NoSuchBucketz-Bucket does not exist, skip prefix deletion: Nz#Failed to delete objects by prefix rz$Failed to delete objects by prefix: )minio.deleteobjectsrRrrLrrlistremove_objectsrlenrrgetattrr!) rrrFrN delete_listerrors error_names deleted_countrrrQrdelete_by_prefix@s@  zMinioService.delete_by_prefix)r r r F)r#)rET)__name__ __module__ __qualname____doc__r!boolrr"rintr.r1r3rbytesr:r<r>rrDrYrLrarrrrrs ! , $       $rrcCs8tdurddlm}t|j|j|j|jdkdatS)uG获取 MinIO 服务实例（单例模式，从 settings 读取配置）Nrsettingstruer) _minio_service app.configrjrMINIO_ENDPOINTMINIO_ACCESS_KEYMINIO_SECRET_KEY MINIO_SECURElowerrirrrget_minio_servicevs  rs)reminior minio.errorrtypingrrloggingdatetimerioapp.core.exceptionsr getLoggerrbrrrlrsrrrrs     e