tamar-file-hub-client 0.0.1__py3-none-any.whl

file_hub_client/services/file/sync_file_service.py

@@ -0,0 +1,255 @@
+"""
+同步文件服务
+"""
+import grpc
+from typing import Optional, Dict, List, Any
+
+from .base_file_service import BaseFileService
+from ...rpc.sync_client import SyncGrpcClient
+from ...schemas import (
+    File,
+    FileListResponse,
+)
+from ...errors import FileNotFoundError
+
+
+class SyncFileService(BaseFileService):
+    """同步文件服务"""
+
+    def __init__(self, client: SyncGrpcClient):
+        """
+        初始化文件服务
+        
+        Args:
+            client: 同步gRPC客户端
+        """
+        self.client = client
+
+    def generate_share_link(
+            self,
+            file_id: str,
+            *,
+            is_public: bool = True,
+            access_scope: str = "view",
+            expire_seconds: int = 86400,
+            max_access: Optional[int] = None,
+            share_password: Optional[str] = None,
+            **metadata
+    ) -> str:
+        """
+        生成分享链接
+        
+        Args:
+            file_id: 文件ID
+            is_public: 是否公开
+            access_scope: 访问范围
+            expire_seconds: 过期时间（秒）
+            max_access: 最大访问次数
+            share_password: 访问密码
+            **metadata: 额外的元数据（如 x-org-id, x-user-id 等）
+            
+        Returns:
+            分享ID
+        """
+        from ...rpc.gen import file_service_pb2, file_service_pb2_grpc
+
+        stub = self.client.get_stub(file_service_pb2_grpc.FileServiceStub)
+
+        request = file_service_pb2.ShareLinkRequest(
+            file_id=file_id,
+            is_public=is_public,
+            access_scope=access_scope,
+            expire_seconds=expire_seconds
+        )
+
+        if max_access is not None:
+            request.max_access = max_access
+        if share_password:
+            request.share_password = share_password
+
+        # 构建元数据
+        grpc_metadata = self.client.build_metadata(**metadata)
+
+        response = stub.GenerateShareLink(request, metadata=grpc_metadata)
+
+        return response.file_share_id
+
+    def visit_file(
+            self,
+            file_share_id: str,
+            access_type: str = "view",
+            access_duration: int = 0,
+            metadata: Optional[Dict[str, Any]] = None,
+            **extra_metadata
+    ) -> None:
+        """
+        访问文件（通过分享链接）
+        
+        Args:
+            file_share_id: 分享ID
+            access_type: 访问类型
+            access_duration: 访问时长
+            metadata: 元数据
+            **extra_metadata: 额外的元数据（如 x-org-id, x-user-id 等）
+        """
+        from ...rpc.gen import file_service_pb2, file_service_pb2_grpc
+        from google.protobuf import struct_pb2
+
+        stub = self.client.get_stub(file_service_pb2_grpc.FileServiceStub)
+
+        # 转换metadata为Struct
+        struct_metadata = struct_pb2.Struct()
+        if metadata:
+            for key, value in metadata.items():
+                struct_metadata[key] = value
+
+        request = file_service_pb2.FileVisitRequest(
+            file_share_id=file_share_id,
+            access_type=access_type,
+            access_duration=access_duration,
+            metadata=struct_metadata
+        )
+
+        # 构建元数据
+        grpc_metadata = self.client.build_metadata(**extra_metadata)
+
+        stub.VisitFile(request, metadata=grpc_metadata)
+
+    def get_file(self, file_id: str, **metadata) -> File:
+        """
+        获取文件信息
+        
+        Args:
+            file_id: 文件ID
+            **metadata: 额外的元数据（如 x-org-id, x-user-id 等）
+            
+        Returns:
+            文件信息
+        """
+        from ...rpc.gen import file_service_pb2, file_service_pb2_grpc
+
+        stub = self.client.get_stub(file_service_pb2_grpc.FileServiceStub)
+
+        request = file_service_pb2.GetFileRequest(file_id=file_id)
+
+        # 构建元数据
+        grpc_metadata = self.client.build_metadata(**metadata)
+
+        try:
+            response = stub.GetFile(request, metadata=grpc_metadata)
+            return self._convert_file_info(response)
+        except grpc.RpcError as e:
+            if e.code() == grpc.StatusCode.NOT_FOUND:
+                raise FileNotFoundError(file_id)
+            raise
+
+    def rename_file(self, file_id: str, new_name: str, **metadata) -> File:
+        """
+        重命名文件
+        
+        Args:
+            file_id: 文件ID
+            new_name: 新文件名
+            **metadata: 额外的元数据（如 x-org-id, x-user-id 等）
+            
+        Returns:
+            更新后的文件信息
+        """
+        from ...rpc.gen import file_service_pb2, file_service_pb2_grpc
+
+        stub = self.client.get_stub(file_service_pb2_grpc.FileServiceStub)
+
+        request = file_service_pb2.RenameFileRequest(
+            file_id=file_id,
+            new_name=new_name
+        )
+
+        # 构建元数据
+        grpc_metadata = self.client.build_metadata(**metadata)
+
+        try:
+            response = stub.RenameFile(request, metadata=grpc_metadata)
+            return self._convert_file_info(response)
+        except grpc.RpcError as e:
+            if e.code() == grpc.StatusCode.NOT_FOUND:
+                raise FileNotFoundError(file_id)
+            raise
+
+    def delete_file(self, file_id: str, **metadata) -> None:
+        """
+        删除文件
+        
+        Args:
+            file_id: 文件ID
+            **metadata: 额外的元数据（如 x-org-id, x-user-id 等）
+        """
+        from ...rpc.gen import file_service_pb2, file_service_pb2_grpc
+
+        stub = self.client.get_stub(file_service_pb2_grpc.FileServiceStub)
+
+        request = file_service_pb2.DeleteFileRequest(file_id=file_id)
+
+        # 构建元数据
+        grpc_metadata = self.client.build_metadata(**metadata)
+
+        try:
+            stub.DeleteFile(request, metadata=grpc_metadata)
+        except grpc.RpcError as e:
+            if e.code() == grpc.StatusCode.NOT_FOUND:
+                raise FileNotFoundError(file_id)
+            raise
+
+    def list_files(
+            self,
+            folder_id: Optional[str] = None,
+            file_name: Optional[str] = None,
+            file_type: Optional[List[str]] = None,
+            created_by_role: Optional[str] = None,
+            created_by: Optional[str] = None,
+            page_size: int = 20,
+            page: int = 1,
+            **metadata
+    ) -> FileListResponse:
+        """
+        列出文件
+        
+        Args:
+            folder_id: 文件夹ID
+            file_name: 文件名过滤
+            file_type: 文件类型过滤
+            created_by_role: 创建者角色过滤
+            created_by: 创建者过滤
+            page_size: 每页大小
+            page: 页码
+            **metadata: 额外的元数据（如 x-org-id, x-user-id 等）
+            
+        Returns:
+            文件列表响应
+        """
+        from ...rpc.gen import file_service_pb2, file_service_pb2_grpc
+
+        stub = self.client.get_stub(file_service_pb2_grpc.FileServiceStub)
+
+        request = file_service_pb2.ListFilesRequest(
+            folder_id=folder_id,
+            page_size=page_size,
+            page=page
+        )
+
+        if file_name:
+            request.file_name = file_name
+        if file_type:
+            request.file_type.extend(file_type)
+        if created_by_role:
+            request.created_by_role = created_by_role
+        if created_by:
+            request.created_by = created_by
+
+        # 构建元数据
+        grpc_metadata = self.client.build_metadata(**metadata)
+
+        response = stub.ListFiles(request, metadata=grpc_metadata)
+
+        files = [self._convert_file_info(f) for f in response.files]
+
+        return FileListResponse(files=files)

file_hub_client/services/folder/__init__.py

@@ -0,0 +1,10 @@
+"""
+文件夹服务模块
+"""
+from .async_folder_service import AsyncFolderService
+from .sync_folder_service import SyncFolderService
+
+__all__ = [
+    "AsyncFolderService",
+    "SyncFolderService",
+]

file_hub_client/services/folder/async_folder_service.py

@@ -0,0 +1,206 @@
+"""
+异步文件夹服务
+"""
+import grpc
+from typing import Optional, Any
+
+from ...rpc.async_client import AsyncGrpcClient
+from ...schemas import (
+    FolderInfo,
+    FolderListResponse,
+)
+from ...errors import FolderNotFoundError, ValidationError
+from ...utils.retry import retry_with_backoff
+
+
+class AsyncFolderService:
+    """异步文件夹服务"""
+
+    def __init__(self, client: AsyncGrpcClient):
+        """
+        初始化文件夹服务
+        
+        Args:
+            client: 异步gRPC客户端
+        """
+        self.client = client
+
+    @retry_with_backoff(max_retries=3)
+    async def create_folder(
+            self,
+            folder_name: str,
+            *,
+            parent_id: Optional[str] = None,
+            **metadata
+    ) -> FolderInfo:
+        """
+        创建文件夹
+        
+        Args:
+            folder_name: 文件夹名称
+            parent_id: 父文件夹ID
+            **metadata: 额外的元数据（如 x-org-id, x-user-id 等）
+            
+        Returns:
+            创建的文件夹信息
+        """
+        from ...rpc.gen import folder_service_pb2, folder_service_pb2_grpc
+
+        stub = await self.client.get_stub(folder_service_pb2_grpc.FolderServiceStub)
+
+        request = folder_service_pb2.CreateFolderRequest(
+            folder_name=folder_name,
+            parent_id=parent_id
+        )
+
+        # 构建元数据
+        grpc_metadata = self.client.build_metadata(**metadata)
+
+        response = await stub.CreateFolder(request, metadata=grpc_metadata)
+        return self._convert_folder_info(response)
+
+    async def rename_folder(self, folder_id: str, new_name: str, **metadata) -> FolderInfo:
+        """
+        重命名文件夹
+        
+        Args:
+            folder_id: 文件夹ID
+            new_name: 新名称
+            **metadata: 额外的元数据（如 x-org-id, x-user-id 等）
+        """
+        from ...rpc.gen import folder_service_pb2, folder_service_pb2_grpc
+
+        if not new_name:
+            raise ValidationError("文件夹名称不能为空")
+
+        stub = await self.client.get_stub(folder_service_pb2_grpc.FolderServiceStub)
+
+        request = folder_service_pb2.RenameFolderRequest(
+            folder_id=folder_id,
+            new_name=new_name
+        )
+
+        # 构建元数据
+        grpc_metadata = self.client.build_metadata(**metadata)
+
+        try:
+            response = await stub.RenameFolder(request, metadata=grpc_metadata)
+            return self._convert_folder_info(response)
+        except grpc.RpcError as e:
+            if e.code() == grpc.StatusCode.NOT_FOUND:
+                raise FolderNotFoundError(folder_id)
+            raise
+
+    async def move_folder(self, folder_id: str, new_parent_id: str, **metadata) -> FolderInfo:
+        """
+        移动文件夹
+        
+        Args:
+            folder_id: 文件夹ID
+            new_parent_id: 新父文件夹ID
+            **metadata: 额外的元数据（如 x-org-id, x-user-id 等）
+        """
+        from ...rpc.gen import folder_service_pb2, folder_service_pb2_grpc
+
+        stub = await self.client.get_stub(folder_service_pb2_grpc.FolderServiceStub)
+
+        request = folder_service_pb2.MoveFolderRequest(
+            folder_id=folder_id,
+            new_parent_id=new_parent_id
+        )
+
+        # 构建元数据
+        grpc_metadata = self.client.build_metadata(**metadata)
+
+        try:
+            response = await stub.MoveFolder(request, metadata=grpc_metadata)
+            return self._convert_folder_info(response)
+        except grpc.RpcError as e:
+            if e.code() == grpc.StatusCode.NOT_FOUND:
+                raise FolderNotFoundError(folder_id)
+            raise
+
+    async def delete_folder(self, folder_id: str, **metadata) -> None:
+        """
+        删除文件夹
+        
+        Args:
+            folder_id: 文件夹ID
+            **metadata: 额外的元数据（如 x-org-id, x-user-id 等）
+        """
+        from ...rpc.gen import folder_service_pb2, folder_service_pb2_grpc
+
+        stub = await self.client.get_stub(folder_service_pb2_grpc.FolderServiceStub)
+
+        request = folder_service_pb2.DeleteFolderRequest(folder_id=folder_id)
+
+        # 构建元数据
+        grpc_metadata = self.client.build_metadata(**metadata)
+
+        try:
+            await stub.DeleteFolder(request, metadata=grpc_metadata)
+        except grpc.RpcError as e:
+            if e.code() == grpc.StatusCode.NOT_FOUND:
+                raise FolderNotFoundError(folder_id)
+            raise
+
+    async def list_folders(
+            self,
+            parent_id: Optional[str] = None,
+            folder_name: Optional[str] = None,
+            created_by_role: Optional[str] = None,
+            created_by: Optional[str] = None,
+            **metadata
+    ) -> FolderListResponse:
+        """
+        列出文件夹
+        
+        Args:
+            parent_id: 父文件夹ID
+            folder_name: 文件夹名称
+            created_by_role: 创建者角色
+            created_by: 创建者
+            **metadata: 额外的元数据（如 x-org-id, x-user-id 等）
+            
+        Returns:
+            文件夹列表响应
+        """
+        from ...rpc.gen import folder_service_pb2, folder_service_pb2_grpc
+
+        stub = await self.client.get_stub(folder_service_pb2_grpc.FolderServiceStub)
+
+        request = folder_service_pb2.ListFoldersRequest()
+
+        if parent_id:
+            request.parent_id = parent_id
+        if folder_name:
+            request.folder_name = folder_name
+        if created_by_role:
+            request.created_by_role = created_by_role
+        if created_by:
+            request.created_by = created_by
+
+        # 构建元数据
+        grpc_metadata = self.client.build_metadata(**metadata)
+
+        response = await stub.ListFolders(request, metadata=grpc_metadata)
+
+        folders = [self._convert_folder_info(f) for f in response.items]
+
+        return FolderListResponse(items=folders)
+
+    def _convert_folder_info(self, proto_folder: Any) -> FolderInfo:
+        """转换Proto文件夹信息为模型"""
+        from ...utils.converter import timestamp_to_datetime
+
+        return FolderInfo(
+            id=proto_folder.id,
+            org_id=proto_folder.org_id,
+            user_id=proto_folder.user_id,
+            folder_name=proto_folder.folder_name,
+            parent_id=proto_folder.parent_id,
+            created_by=proto_folder.created_by,
+            created_by_role=proto_folder.created_by_role,
+            created_at=timestamp_to_datetime(proto_folder.created_at),
+            updated_at=timestamp_to_datetime(proto_folder.updated_at)
+        )

file_hub_client/services/folder/sync_folder_service.py

@@ -0,0 +1,205 @@
+"""
+同步文件夹服务
+"""
+import grpc
+from typing import Optional, Any
+
+from ...rpc.sync_client import SyncGrpcClient
+from ...schemas import (
+    FolderInfo,
+    FolderListResponse,
+)
+from ...errors import FolderNotFoundError, ValidationError
+from ...utils.retry import retry_with_backoff
+
+
+class SyncFolderService:
+    """同步文件夹服务"""
+
+    def __init__(self, client: SyncGrpcClient):
+        """
+        初始化文件夹服务
+        
+        Args:
+            client: 同步gRPC客户端
+        """
+        self.client = client
+
+    @retry_with_backoff(max_retries=3)
+    def create_folder(
+            self,
+            folder_name: str,
+            parent_id: str,
+            **metadata
+    ) -> FolderInfo:
+        """
+        创建文件夹
+        
+        Args:
+            folder_name: 文件夹名称
+            parent_id: 父文件夹ID
+            **metadata: 额外的元数据（如 x-org-id, x-user-id 等）
+            
+        Returns:
+            创建的文件夹信息
+        """
+        from ...rpc.gen import folder_service_pb2, folder_service_pb2_grpc
+
+        stub = self.client.get_stub(folder_service_pb2_grpc.FolderServiceStub)
+
+        request = folder_service_pb2.CreateFolderRequest(
+            folder_name=folder_name,
+            parent_id=parent_id
+        )
+
+        # 构建元数据
+        grpc_metadata = self.client.build_metadata(**metadata)
+
+        response = stub.CreateFolder(request, metadata=grpc_metadata)
+        return self._convert_folder_info(response)
+
+    def rename_folder(self, folder_id: str, new_name: str, **metadata) -> FolderInfo:
+        """
+        重命名文件夹
+        
+        Args:
+            folder_id: 文件夹ID
+            new_name: 新名称
+            **metadata: 额外的元数据（如 x-org-id, x-user-id 等）
+        """
+        from ...rpc.gen import folder_service_pb2, folder_service_pb2_grpc
+
+        if not new_name:
+            raise ValidationError("文件夹名称不能为空")
+
+        stub = self.client.get_stub(folder_service_pb2_grpc.FolderServiceStub)
+
+        request = folder_service_pb2.RenameFolderRequest(
+            folder_id=folder_id,
+            new_name=new_name
+        )
+
+        # 构建元数据
+        grpc_metadata = self.client.build_metadata(**metadata)
+
+        try:
+            response = stub.RenameFolder(request, metadata=grpc_metadata)
+            return self._convert_folder_info(response)
+        except grpc.RpcError as e:
+            if e.code() == grpc.StatusCode.NOT_FOUND:
+                raise FolderNotFoundError(folder_id)
+            raise
+
+    def move_folder(self, folder_id: str, new_parent_id: str, **metadata) -> FolderInfo:
+        """
+        移动文件夹
+        
+        Args:
+            folder_id: 文件夹ID
+            new_parent_id: 新父文件夹ID
+            **metadata: 额外的元数据（如 x-org-id, x-user-id 等）
+        """
+        from ...rpc.gen import folder_service_pb2, folder_service_pb2_grpc
+
+        stub = self.client.get_stub(folder_service_pb2_grpc.FolderServiceStub)
+
+        request = folder_service_pb2.MoveFolderRequest(
+            folder_id=folder_id,
+            new_parent_id=new_parent_id
+        )
+
+        # 构建元数据
+        grpc_metadata = self.client.build_metadata(**metadata)
+
+        try:
+            response = stub.MoveFolder(request, metadata=grpc_metadata)
+            return self._convert_folder_info(response)
+        except grpc.RpcError as e:
+            if e.code() == grpc.StatusCode.NOT_FOUND:
+                raise FolderNotFoundError(folder_id)
+            raise
+
+    def delete_folder(self, folder_id: str, **metadata) -> None:
+        """
+        删除文件夹
+        
+        Args:
+            folder_id: 文件夹ID
+            **metadata: 额外的元数据（如 x-org-id, x-user-id 等）
+        """
+        from ...rpc.gen import folder_service_pb2, folder_service_pb2_grpc
+
+        stub = self.client.get_stub(folder_service_pb2_grpc.FolderServiceStub)
+
+        request = folder_service_pb2.DeleteFolderRequest(folder_id=folder_id)
+
+        # 构建元数据
+        grpc_metadata = self.client.build_metadata(**metadata)
+
+        try:
+            stub.DeleteFolder(request, metadata=grpc_metadata)
+        except grpc.RpcError as e:
+            if e.code() == grpc.StatusCode.NOT_FOUND:
+                raise FolderNotFoundError(folder_id)
+            raise
+
+    def list_folders(
+            self,
+            parent_id: Optional[str] = None,
+            folder_name: Optional[str] = None,
+            created_by_role: Optional[str] = None,
+            created_by: Optional[str] = None,
+            **metadata
+    ) -> FolderListResponse:
+        """
+        列出文件夹
+        
+        Args:
+            parent_id: 父文件夹ID
+            folder_name: 文件夹名称
+            created_by_role: 创建者角色
+            created_by: 创建者
+            **metadata: 额外的元数据（如 x-org-id, x-user-id 等）
+            
+        Returns:
+            文件夹列表响应
+        """
+        from ...rpc.gen import folder_service_pb2, folder_service_pb2_grpc
+
+        stub = self.client.get_stub(folder_service_pb2_grpc.FolderServiceStub)
+
+        request = folder_service_pb2.ListFoldersRequest()
+
+        if parent_id:
+            request.parent_id = parent_id
+        if folder_name:
+            request.folder_name = folder_name
+        if created_by_role:
+            request.created_by_role = created_by_role
+        if created_by:
+            request.created_by = created_by
+
+        # 构建元数据
+        grpc_metadata = self.client.build_metadata(**metadata)
+
+        response = stub.ListFolders(request, metadata=grpc_metadata)
+
+        folders = [self._convert_folder_info(f) for f in response.items]
+
+        return FolderListResponse(items=folders)
+
+    def _convert_folder_info(self, proto_folder: Any) -> FolderInfo:
+        """转换Proto文件夹信息为模型"""
+        from ...utils.converter import timestamp_to_datetime
+
+        return FolderInfo(
+            id=proto_folder.id,
+            org_id=proto_folder.org_id,
+            user_id=proto_folder.user_id,
+            folder_name=proto_folder.folder_name,
+            parent_id=proto_folder.parent_id,
+            created_by=proto_folder.created_by,
+            created_by_role=proto_folder.created_by_role,
+            created_at=timestamp_to_datetime(proto_folder.created_at),
+            updated_at=timestamp_to_datetime(proto_folder.updated_at)
+        )