tamar-file-hub-client 0.0.5__tar.gz → 0.0.6__tar.gz

{tamar_file_hub_client-0.0.5 → tamar_file_hub_client-0.0.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tamar-file-hub-client
-Version: 0.0.5
+Version: 0.0.6
 Summary: A Python SDK for gRPC-based file management system
 Home-page: https://github.com/Tamar-Edge-AI/file-hub-client
 Author: Oscar Ou

{tamar_file_hub_client-0.0.5 → tamar_file_hub_client-0.0.6}/file_hub_client/client.py

@@ -23,6 +23,8 @@ from .services import (
 )
 from .schemas.context import UserContext, RequestContext
 
+MAX_MESSAGE_LENGTH = 2 ** 31 - 1  # 对于32位系统
+
 
 class AsyncTamarFileHubClient:
     """异步文件管理系统客户端"""
@@ -41,6 +43,7 @@ class AsyncTamarFileHubClient:
             request_context: Optional[RequestContext] = None,
             enable_logging: bool = True,
             log_level: str = "INFO",
+            options: Optional[list] = None,
     ):
         """
         初始化客户端
@@ -56,6 +59,13 @@ class AsyncTamarFileHubClient:
             default_metadata: 默认的元数据（如 org_id, user_id 等）
             user_context: 用户上下文
             request_context: 请求上下文
+            enable_logging: 是否启用日志
+            log_level: 日志级别
+            options: gRPC 通道选项列表，例如：
+                [
+                    ('grpc.max_send_message_length', MAX_MESSAGE_LENGTH),
+                    ('grpc.max_receive_message_length', MAX_MESSAGE_LENGTH),
+                ]
             
         环境变量：
             FILE_HUB_HOST: gRPC 服务器地址
@@ -86,11 +96,19 @@ class AsyncTamarFileHubClient:
         if default_metadata is None:
             default_metadata = {}
 
+        # 设置默认的 gRPC 选项（增加消息大小限制）
+        if options is None:
+            options = [
+                ('grpc.max_send_message_length', MAX_MESSAGE_LENGTH),
+                ('grpc.max_receive_message_length', MAX_MESSAGE_LENGTH),
+            ]
+
         self._client = AsyncGrpcClient(
             self._host,
             self._port,
             self._secure,
             credentials,
+            options=options,
             retry_count=self._retry_count,
             retry_delay=self._retry_delay,
             default_metadata=default_metadata,
@@ -243,6 +261,13 @@ class TamarFileHubClient:
             default_metadata: 默认的元数据（如 org_id, user_id 等）
             user_context: 用户上下文
             request_context: 请求上下文
+            enable_logging: 是否启用日志
+            log_level: 日志级别
+            options: gRPC 通道选项列表，例如：
+                [
+                    ('grpc.max_send_message_length', 100 * 1024 * 1024),  # 100MB
+                    ('grpc.max_receive_message_length', 100 * 1024 * 1024),  # 100MB
+                ]
             
         环境变量：
             FILE_HUB_HOST: gRPC 服务器地址
@@ -273,11 +298,19 @@ class TamarFileHubClient:
         if default_metadata is None:
             default_metadata = {}
 
+        # 设置默认的 gRPC 选项（增加消息大小限制）
+        if options is None:
+            options = [
+                ('grpc.max_send_message_length', 100 * 1024 * 1024),  # 100MB
+                ('grpc.max_receive_message_length', 100 * 1024 * 1024),  # 100MB
+            ]
+
         self._client = SyncGrpcClient(
             self._host,
             self._port,
             self._secure,
             credentials,
+            options=options,
             retry_count=self._retry_count,
             retry_delay=self._retry_delay,
             default_metadata=default_metadata,

{tamar_file_hub_client-0.0.5 → tamar_file_hub_client-0.0.6}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="tamar-file-hub-client",
-    version="0.0.5",
+    version="0.0.6",
     description="A Python SDK for gRPC-based file management system",
     long_description=open("README.md", encoding="utf-8").read(),
     long_description_content_type="text/markdown",

{tamar_file_hub_client-0.0.5 → tamar_file_hub_client-0.0.6}/tamar_file_hub_client.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tamar-file-hub-client
-Version: 0.0.5
+Version: 0.0.6
 Summary: A Python SDK for gRPC-based file management system
 Home-page: https://github.com/Tamar-Edge-AI/file-hub-client
 Author: Oscar Ou

{tamar_file_hub_client-0.0.5 → tamar_file_hub_client-0.0.6}/tamar_file_hub_client.egg-info/SOURCES.txt

@@ -12,7 +12,6 @@ file_hub_client/errors/__init__.py
 file_hub_client/errors/exceptions.py
 file_hub_client/rpc/__init__.py
 file_hub_client/rpc/async_client.py
-file_hub_client/rpc/generate_grpc.py
 file_hub_client/rpc/interceptors.py
 file_hub_client/rpc/sync_client.py
 file_hub_client/rpc/gen/__init__.py

tamar_file_hub_client-0.0.5/file_hub_client/rpc/generate_grpc.py

@@ -1,76 +0,0 @@
-"""
-生成gRPC Python代码的脚本
-"""
-import os
-import subprocess
-from pathlib import Path
-
-
-def generate_grpc_code():
-    """生成gRPC Python代码"""
-    # 获取当前脚本所在目录
-    current_dir = Path(__file__).parent
-    proto_dir = current_dir / "protos"
-    gen_dir = current_dir / "gen"
-    
-    # 创建生成目录
-    gen_dir.mkdir(exist_ok=True)
-    
-    # 创建__init__.py文件
-    (gen_dir / "__init__.py").touch()
-    
-    # 获取所有proto文件
-    proto_files = list(proto_dir.glob("*.proto"))
-    
-    for proto_file in proto_files:
-        print(f"生成 {proto_file.name} 的Python代码...")
-        
-        # 生成gRPC代码
-        cmd = [
-            "python3", "-m", "grpc_tools.protoc",
-            f"-I{proto_dir}",
-            f"--python_out={gen_dir}",
-            f"--grpc_python_out={gen_dir}",
-            str(proto_file)
-        ]
-        
-        result = subprocess.run(cmd, capture_output=True, text=True)
-        
-        if result.returncode != 0:
-            print(f"错误: {result.stderr}")
-            raise RuntimeError(f"生成gRPC代码失败: {result.stderr}")
-        else:
-            print(f"成功生成 {proto_file.name}")
-    
-    # 修复导入路径
-    fix_imports(gen_dir)
-    
-    print("所有gRPC代码生成完成！")
-
-
-def fix_imports(gen_dir):
-    """修复生成的Python文件中的导入路径"""
-    for py_file in gen_dir.glob("*_pb2*.py"):
-        with open(py_file, "r", encoding="utf-8") as f:
-            content = f.read()
-        
-        # 修复导入路径 - 现在有多个proto文件
-        for proto_name in ["file_service", "folder_service", "taple_service"]:
-            content = content.replace(f"import {proto_name}_pb2", f"from . import {proto_name}_pb2")
-        
-        with open(py_file, "w", encoding="utf-8") as f:
-            f.write(content)
-
-
-def main():
-    """命令行入口点"""
-    try:
-        generate_grpc_code()
-    except Exception as e:
-        print(f"生成失败: {e}")
-        return 1
-    return 0
-
-
-if __name__ == "__main__":
-    generate_grpc_code()