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

file_hub_client/rpc/async_client.py

@@ -0,0 +1,312 @@
+"""
+异步gRPC客户端
+"""
+from enum import Enum
+
+import grpc
+import asyncio
+import uuid
+import platform
+import socket
+from typing import Optional, Dict, List, Tuple
+
+from ..enums import Role
+from ..errors import ConnectionError
+from ..schemas.context import UserContext, RequestContext, FullContext
+
+
+class AsyncGrpcClient:
+    """异步gRPC客户端基类"""
+
+    def __init__(
+            self,
+            host: str = "localhost",
+            port: int = 50051,
+            secure: bool = False,
+            credentials: Optional[dict] = None,
+            options: Optional[list] = None,
+            retry_count: int = 3,
+            retry_delay: float = 1.0,
+            default_metadata: Optional[Dict[str, str]] = None,
+            user_context: Optional[UserContext] = None,
+            request_context: Optional[RequestContext] = None,
+    ):
+        """
+        初始化异步gRPC客户端
+        
+        Args:
+            host: 服务器地址
+            port: 服务器端口
+            secure: 是否使用安全连接（TLS）
+            credentials: 认证凭据字典（如 {'api_key': 'xxx'}）
+            options: gRPC通道选项
+            retry_count: 连接重试次数
+            retry_delay: 重试延迟（秒）
+            default_metadata: 默认的元数据（如 org_id, user_id 等）
+            user_context: 用户上下文
+            request_context: 请求上下文
+        """
+        self.host = host
+        self.port = port
+        self.address = f"{host}:{port}"
+        self.secure = secure
+        self.credentials = credentials
+        self.options = options or []
+        self.retry_count = retry_count
+        self.retry_delay = retry_delay
+        self.default_metadata = default_metadata or {}
+        self._channel: Optional[grpc.aio.Channel] = None
+        self._stubs = {}
+        self._stub_lock = asyncio.Lock()
+
+        # 上下文管理
+        self._user_context = user_context
+        self._request_context = request_context or self._create_default_request_context()
+
+        # 如果提供了user_context，将其添加到default_metadata
+        if self._user_context:
+            self.default_metadata.update(self._user_context.to_metadata())
+
+        # 添加请求上下文到default_metadata
+        self.default_metadata.update(self._request_context.to_metadata())
+
+    def _create_default_request_context(self) -> RequestContext:
+        """创建默认的请求上下文"""
+        # 尝试获取客户端IP
+        client_ip = None
+        try:
+            # 获取本机IP（适用于内网环境）
+            hostname = socket.gethostname()
+            client_ip = socket.gethostbyname(hostname)
+        except:
+            pass
+
+        # 获取客户端信息
+        return RequestContext(
+            client_ip=client_ip,
+            client_type="python-sdk",
+            client_version="1.0.0",  # TODO: 从包版本获取
+            user_agent=f"FileHubClient/1.0.0 Python/{platform.python_version()} {platform.system()}/{platform.release()}"
+        )
+
+    def _create_channel_credentials(self) -> Optional[grpc.ChannelCredentials]:
+        """创建通道凭据"""
+        if not self.secure:
+            return None
+
+        # 使用默认的SSL凭据
+        channel_credentials = grpc.ssl_channel_credentials()
+
+        # 如果有API密钥，创建组合凭据
+        if self.credentials and 'api_key' in self.credentials:
+            # 创建元数据凭据
+            def metadata_callback(context, callback):
+                metadata = [('authorization', f"Bearer {self.credentials['api_key']}")]
+                callback(metadata, None)
+
+            call_credentials = grpc.metadata_call_credentials(metadata_callback)
+            channel_credentials = grpc.composite_channel_credentials(
+                channel_credentials,
+                call_credentials
+            )
+
+        return channel_credentials
+
+    async def connect(self):
+        """连接到gRPC服务器（带重试）"""
+        if self._channel is not None:
+            return
+
+        last_error = None
+        for attempt in range(self.retry_count):
+            try:
+                if attempt > 0:
+                    await asyncio.sleep(self.retry_delay)
+
+                channel_credentials = self._create_channel_credentials()
+
+                if channel_credentials:
+                    self._channel = grpc.aio.secure_channel(
+                        self.address,
+                        channel_credentials,
+                        options=self.options
+                    )
+                else:
+                    self._channel = grpc.aio.insecure_channel(
+                        self.address,
+                        options=self.options
+                    )
+
+                # 连接
+                try:
+                    await asyncio.wait_for(self._channel.channel_ready(), timeout=5.0)
+                except asyncio.TimeoutError:
+                    raise ConnectionError(f"连接超时：{self.address}")
+
+                # 连接成功
+                return
+
+            except Exception as e:
+                last_error = e
+                if attempt < self.retry_count - 1:
+                    print(f"连接失败 (尝试 {attempt + 1}/{self.retry_count}): {str(e)}")
+                    if self._channel:
+                        await self._channel.close()
+                        self._channel = None
+                else:
+                    # 最后一次尝试失败
+                    if self._channel:
+                        await self._channel.close()
+                        self._channel = None
+
+        # 所有重试都失败
+        raise ConnectionError(
+            f"无法连接到gRPC服务器 {self.address} (尝试了 {self.retry_count} 次): {str(last_error)}"
+        )
+
+    async def close(self):
+        """关闭连接"""
+        if self._channel:
+            await self._channel.close()
+            self._channel = None
+            self._stubs.clear()
+
+    async def get_stub(self, stub_class):
+        """
+        获取gRPC stub实例
+        
+        Args:
+            stub_class: Stub类
+            
+        Returns:
+            Stub实例
+        """
+        if not self._channel:
+            raise ConnectionError("未连接到gRPC服务器")
+
+        stub_name = stub_class.__name__
+        async with self._stub_lock:
+            if stub_name not in self._stubs:
+                self._stubs[stub_name] = stub_class(self._channel)
+            return self._stubs[stub_name]
+
+    def build_metadata(self, **kwargs) -> List[Tuple[str, str]]:
+        """
+        构建请求元数据
+        
+        Args:
+            **kwargs: 要覆盖或添加的元数据
+            
+        Returns:
+            元数据列表
+        """
+        metadata = {}
+
+        # 添加默认元数据
+        metadata.update(self.default_metadata)
+
+        # 添加/覆盖传入的元数据
+        metadata.update(kwargs)
+
+        # 如果没有 request_id，自动生成一个
+        if 'x-request-id' not in metadata:
+            metadata['x-request-id'] = (
+                    self._request_context.extra.get("request_id") or str(uuid.uuid4())
+            )
+
+        # 转换为 gRPC 需要的格式
+        result = []
+        for k, v in metadata.items():
+            if v is None:
+                continue
+            if isinstance(v, Enum):
+                v = v.value
+            result.append((k, str(v)))
+
+        return result
+
+    def update_default_metadata(self, **kwargs):
+        """
+        更新默认元数据
+        
+        Args:
+            **kwargs: 要更新的元数据键值对
+        """
+        self.default_metadata.update(kwargs)
+
+    def set_user_context(self, org_id: str, user_id: str, role: Role = Role.ACCOUNT, actor_id: Optional[str] = None):
+        """
+        设置用户上下文信息
+        
+        Args:
+            org_id: 组织ID
+            user_id: 用户ID
+            role: 用户角色（默认为 ACCOUNT）
+            actor_id: 操作者ID（如果不同于 user_id）
+        """
+        self._user_context = UserContext(
+            org_id=org_id,
+            user_id=user_id,
+            role=role,
+            actor_id=actor_id
+        )
+        # 更新到默认元数据
+        self.update_default_metadata(**self._user_context.to_metadata())
+
+    def get_user_context(self) -> Optional[UserContext]:
+        """获取当前用户上下文"""
+        return self._user_context
+
+    def clear_user_context(self):
+        """清除用户上下文信息"""
+        self._user_context = None
+        for key in ['x-org-id', 'x-user-id', 'x-role', 'x-actor-id']:
+            self.default_metadata.pop(key, None)
+
+    def set_request_context(self, request_context: RequestContext):
+        """设置请求上下文"""
+        self._request_context = request_context
+        # 更新到默认元数据
+        self.update_default_metadata(**self._request_context.to_metadata())
+
+    def get_request_context(self) -> RequestContext:
+        """获取当前请求上下文"""
+        return self._request_context
+
+    def update_request_context(self, **kwargs):
+        """
+        更新请求上下文的部分字段
+        
+        Args:
+            **kwargs: 要更新的字段
+        """
+        if kwargs.get('client_ip'):
+            self._request_context.client_ip = kwargs['client_ip']
+        if kwargs.get('client_version'):
+            self._request_context.client_version = kwargs['client_version']
+        if kwargs.get('client_type'):
+            self._request_context.client_type = kwargs['client_type']
+        if kwargs.get('user_agent'):
+            self._request_context.user_agent = kwargs['user_agent']
+
+        # 处理extra字段
+        extra = kwargs.get('extra')
+        if extra and isinstance(extra, dict):
+            self._request_context.extra.update(extra)
+
+        # 更新到默认元数据
+        self.update_default_metadata(**self._request_context.to_metadata())
+
+    def get_full_context(self) -> FullContext:
+        """获取完整的上下文信息"""
+        return FullContext(
+            user_context=self._user_context,
+            request_context=self._request_context
+        )
+
+    async def __aenter__(self) -> "AsyncGrpcClient":
+        await self.connect()
+        return self
+
+    async def __aexit__(self, exc_type, exc_val, exc_tb):
+        await self.close()

file_hub_client/rpc/gen/__init__.py

@@ -0,0 +1 @@
+# Auto-generated init file for gRPC package

file_hub_client/rpc/gen/file_service_pb2.py

@@ -0,0 +1,74 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: file_service.proto
+# Protobuf Python Version: 5.27.2
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    5,
+    27,
+    2,
+    '',
+    'file_service.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
+from google.protobuf import struct_pb2 as google_dot_protobuf_dot_struct__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x12\x66ile_service.proto\x12\x04\x66ile\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x1cgoogle/protobuf/struct.proto\"\xab\x01\n\x04\x46ile\x12\n\n\x02id\x18\x01 \x01(\t\x12\x11\n\tfolder_id\x18\x02 \x01(\t\x12\x11\n\tfile_name\x18\x03 \x01(\t\x12\x11\n\tfile_type\x18\x04 \x01(\t\x12.\n\ncreated_at\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12.\n\nupdated_at\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\"\xb7\x02\n\nUploadFile\x12\n\n\x02id\x18\x01 \x01(\t\x12\x11\n\tfolder_id\x18\x02 \x01(\t\x12\x0f\n\x07\x66ile_id\x18\x03 \x01(\t\x12\x14\n\x0cstorage_type\x18\x04 \x01(\t\x12\x13\n\x0bstored_name\x18\x05 \x01(\t\x12\x13\n\x0bstored_path\x18\x06 \x01(\t\x12\x11\n\tfile_name\x18\x07 \x01(\t\x12\x11\n\tfile_size\x18\x08 \x01(\x03\x12\x10\n\x08\x66ile_ext\x18\t \x01(\t\x12\x11\n\tmime_type\x18\n \x01(\t\x12\x0e\n\x06status\x18\x0b \x01(\t\x12.\n\ncreated_at\x18\x0c \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12.\n\nupdated_at\x18\r \x01(\x0b\x32\x1a.google.protobuf.Timestamp\"\xdf\x01\n\x11UploadFileRequest\x12\x16\n\tfolder_id\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x11\n\tfile_name\x18\x02 \x01(\t\x12\x0f\n\x07\x63ontent\x18\x03 \x01(\x0c\x12\x11\n\tfile_type\x18\x04 \x01(\t\x12\x11\n\tmime_type\x18\x05 \x01(\t\x12\x19\n\x0cis_temporary\x18\x06 \x01(\x08H\x01\x88\x01\x01\x12\x1b\n\x0e\x65xpire_seconds\x18\x07 \x01(\x05H\x02\x88\x01\x01\x42\x0c\n\n_folder_idB\x0f\n\r_is_temporaryB\x11\n\x0f_expire_seconds\"\xf3\x01\n\x10UploadUrlRequest\x12\x16\n\tfolder_id\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x11\n\tfile_name\x18\x02 \x01(\t\x12\x11\n\tfile_type\x18\x03 \x01(\t\x12\x11\n\tmime_type\x18\x04 \x01(\t\x12\x11\n\tfile_size\x18\x05 \x01(\x03\x12\x11\n\tfile_hash\x18\x06 \x01(\t\x12\x19\n\x0cis_temporary\x18\x07 \x01(\x08H\x01\x88\x01\x01\x12\x1b\n\x0e\x65xpire_seconds\x18\x08 \x01(\x05H\x02\x88\x01\x01\x42\x0c\n\n_folder_idB\x0f\n\r_is_temporaryB\x11\n\x0f_expire_seconds\")\n\x16UploadCompletedRequest\x12\x0f\n\x07\x66ile_id\x18\x01 \x01(\t\"U\n\x12\x44ownloadUrlRequest\x12\x0f\n\x07\x66ile_id\x18\x01 \x01(\t\x12\x1b\n\x0e\x65xpire_seconds\x18\x02 \x01(\x05H\x00\x88\x01\x01\x42\x11\n\x0f_expire_seconds\"\xd4\x01\n\x10ShareLinkRequest\x12\x0f\n\x07\x66ile_id\x18\x01 \x01(\t\x12\x11\n\tis_public\x18\x02 \x01(\x08\x12\x14\n\x0c\x61\x63\x63\x65ss_scope\x18\x03 \x01(\t\x12\x1b\n\x0e\x65xpire_seconds\x18\x04 \x01(\x05H\x00\x88\x01\x01\x12\x17\n\nmax_access\x18\x05 \x01(\x05H\x01\x88\x01\x01\x12\x1b\n\x0eshare_password\x18\x06 \x01(\tH\x02\x88\x01\x01\x42\x11\n\x0f_expire_secondsB\r\n\x0b_max_accessB\x11\n\x0f_share_password\"\x82\x01\n\x10\x46ileVisitRequest\x12\x15\n\rfile_share_id\x18\x01 \x01(\t\x12\x13\n\x0b\x61\x63\x63\x65ss_type\x18\x02 \x01(\t\x12\x17\n\x0f\x61\x63\x63\x65ss_duration\x18\x03 \x01(\x05\x12)\n\x08metadata\x18\x04 \x01(\x0b\x32\x17.google.protobuf.Struct\"!\n\x0eGetFileRequest\x12\x0f\n\x07\x66ile_id\x18\x01 \x01(\t\"6\n\x11RenameFileRequest\x12\x0f\n\x07\x66ile_id\x18\x01 \x01(\t\x12\x10\n\x08new_name\x18\x02 \x01(\t\"$\n\x11\x44\x65leteFileRequest\x12\x0f\n\x07\x66ile_id\x18\x01 \x01(\t\"\x8d\x02\n\x10ListFilesRequest\x12\x16\n\tfolder_id\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x16\n\tfile_name\x18\x02 \x01(\tH\x01\x88\x01\x01\x12\x11\n\tfile_type\x18\x03 \x03(\t\x12\x1c\n\x0f\x63reated_by_role\x18\x04 \x01(\tH\x02\x88\x01\x01\x12\x17\n\ncreated_by\x18\x05 \x01(\tH\x03\x88\x01\x01\x12\x16\n\tpage_size\x18\x06 \x01(\x05H\x04\x88\x01\x01\x12\x11\n\x04page\x18\x07 \x01(\x05H\x05\x88\x01\x01\x42\x0c\n\n_folder_idB\x0c\n\n_file_nameB\x12\n\x10_created_by_roleB\r\n\x0b_created_byB\x0c\n\n_page_sizeB\x07\n\x05_page\"U\n\x12UploadFileResponse\x12\x18\n\x04\x66ile\x18\x01 \x01(\x0b\x32\n.file.File\x12%\n\x0bupload_file\x18\x02 \x01(\x0b\x32\x10.file.UploadFile\"a\n\x11UploadUrlResponse\x12\x18\n\x04\x66ile\x18\x01 \x01(\x0b\x32\n.file.File\x12%\n\x0bupload_file\x18\x02 \x01(\x0b\x32\x10.file.UploadFile\x12\x0b\n\x03url\x18\x03 \x01(\t\"\"\n\x13\x44ownloadUrlResponse\x12\x0b\n\x03url\x18\x01 \x01(\t\"*\n\x11ShareLinkResponse\x12\x15\n\rfile_share_id\x18\x01 \x01(\t\"-\n\x10\x46ileListResponse\x12\x19\n\x05\x66iles\x18\x01 \x03(\x0b\x32\n.file.File\"\x07\n\x05\x45mpty2\xbd\x05\n\x0b\x46ileService\x12?\n\nUploadFile\x12\x17.file.UploadFileRequest\x1a\x18.file.UploadFileResponse\x12\x44\n\x11GenerateUploadUrl\x12\x16.file.UploadUrlRequest\x1a\x17.file.UploadUrlResponse\x12M\n\x1aGenerateResumableUploadUrl\x12\x16.file.UploadUrlRequest\x1a\x17.file.UploadUrlResponse\x12\x43\n\x16\x43onfirmUploadCompleted\x12\x1c.file.UploadCompletedRequest\x1a\x0b.file.Empty\x12J\n\x13GenerateDownloadUrl\x12\x18.file.DownloadUrlRequest\x1a\x19.file.DownloadUrlResponse\x12\x44\n\x11GenerateShareLink\x12\x16.file.ShareLinkRequest\x1a\x17.file.ShareLinkResponse\x12\x30\n\tVisitFile\x12\x16.file.FileVisitRequest\x1a\x0b.file.Empty\x12+\n\x07GetFile\x12\x14.file.GetFileRequest\x1a\n.file.File\x12\x31\n\nRenameFile\x12\x17.file.RenameFileRequest\x1a\n.file.File\x12\x32\n\nDeleteFile\x12\x17.file.DeleteFileRequest\x1a\x0b.file.Empty\x12;\n\tListFiles\x12\x16.file.ListFilesRequest\x1a\x16.file.FileListResponseb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'file_service_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
+  _globals['_FILE']._serialized_start=92
+  _globals['_FILE']._serialized_end=263
+  _globals['_UPLOADFILE']._serialized_start=266
+  _globals['_UPLOADFILE']._serialized_end=577
+  _globals['_UPLOADFILEREQUEST']._serialized_start=580
+  _globals['_UPLOADFILEREQUEST']._serialized_end=803
+  _globals['_UPLOADURLREQUEST']._serialized_start=806
+  _globals['_UPLOADURLREQUEST']._serialized_end=1049
+  _globals['_UPLOADCOMPLETEDREQUEST']._serialized_start=1051
+  _globals['_UPLOADCOMPLETEDREQUEST']._serialized_end=1092
+  _globals['_DOWNLOADURLREQUEST']._serialized_start=1094
+  _globals['_DOWNLOADURLREQUEST']._serialized_end=1179
+  _globals['_SHARELINKREQUEST']._serialized_start=1182
+  _globals['_SHARELINKREQUEST']._serialized_end=1394
+  _globals['_FILEVISITREQUEST']._serialized_start=1397
+  _globals['_FILEVISITREQUEST']._serialized_end=1527
+  _globals['_GETFILEREQUEST']._serialized_start=1529
+  _globals['_GETFILEREQUEST']._serialized_end=1562
+  _globals['_RENAMEFILEREQUEST']._serialized_start=1564
+  _globals['_RENAMEFILEREQUEST']._serialized_end=1618
+  _globals['_DELETEFILEREQUEST']._serialized_start=1620
+  _globals['_DELETEFILEREQUEST']._serialized_end=1656
+  _globals['_LISTFILESREQUEST']._serialized_start=1659
+  _globals['_LISTFILESREQUEST']._serialized_end=1928
+  _globals['_UPLOADFILERESPONSE']._serialized_start=1930
+  _globals['_UPLOADFILERESPONSE']._serialized_end=2015
+  _globals['_UPLOADURLRESPONSE']._serialized_start=2017
+  _globals['_UPLOADURLRESPONSE']._serialized_end=2114
+  _globals['_DOWNLOADURLRESPONSE']._serialized_start=2116
+  _globals['_DOWNLOADURLRESPONSE']._serialized_end=2150
+  _globals['_SHARELINKRESPONSE']._serialized_start=2152
+  _globals['_SHARELINKRESPONSE']._serialized_end=2194
+  _globals['_FILELISTRESPONSE']._serialized_start=2196
+  _globals['_FILELISTRESPONSE']._serialized_end=2241
+  _globals['_EMPTY']._serialized_start=2243
+  _globals['_EMPTY']._serialized_end=2250
+  _globals['_FILESERVICE']._serialized_start=2253
+  _globals['_FILESERVICE']._serialized_end=2954
+# @@protoc_insertion_point(module_scope)