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

file_hub_client/rpc/gen/folder_service_pb2_grpc.py

@@ -0,0 +1,269 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+import warnings
+
+import file_hub_client.rpc.gen.folder_service_pb2 as folder__service__pb2
+
+GRPC_GENERATED_VERSION = '1.67.1'
+GRPC_VERSION = grpc.__version__
+_version_not_supported = False
+
+try:
+    from grpc._utilities import first_version_is_lower
+    _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
+except ImportError:
+    _version_not_supported = True
+
+if _version_not_supported:
+    raise RuntimeError(
+        f'The grpc package installed is at version {GRPC_VERSION},'
+        + f' but the generated code in folder_service_pb2_grpc.py depends on'
+        + f' grpcio>={GRPC_GENERATED_VERSION}.'
+        + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
+        + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
+    )
+
+
+class FolderServiceStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.CreateFolder = channel.unary_unary(
+                '/folder.FolderService/CreateFolder',
+                request_serializer=folder__service__pb2.CreateFolderRequest.SerializeToString,
+                response_deserializer=folder__service__pb2.Folder.FromString,
+                _registered_method=True)
+        self.RenameFolder = channel.unary_unary(
+                '/folder.FolderService/RenameFolder',
+                request_serializer=folder__service__pb2.RenameFolderRequest.SerializeToString,
+                response_deserializer=folder__service__pb2.Folder.FromString,
+                _registered_method=True)
+        self.MoveFolder = channel.unary_unary(
+                '/folder.FolderService/MoveFolder',
+                request_serializer=folder__service__pb2.MoveFolderRequest.SerializeToString,
+                response_deserializer=folder__service__pb2.Folder.FromString,
+                _registered_method=True)
+        self.DeleteFolder = channel.unary_unary(
+                '/folder.FolderService/DeleteFolder',
+                request_serializer=folder__service__pb2.DeleteFolderRequest.SerializeToString,
+                response_deserializer=folder__service__pb2.Empty.FromString,
+                _registered_method=True)
+        self.ListFolders = channel.unary_unary(
+                '/folder.FolderService/ListFolders',
+                request_serializer=folder__service__pb2.ListFoldersRequest.SerializeToString,
+                response_deserializer=folder__service__pb2.FolderListResponse.FromString,
+                _registered_method=True)
+
+
+class FolderServiceServicer(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def CreateFolder(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def RenameFolder(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def MoveFolder(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def DeleteFolder(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def ListFolders(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_FolderServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'CreateFolder': grpc.unary_unary_rpc_method_handler(
+                    servicer.CreateFolder,
+                    request_deserializer=folder__service__pb2.CreateFolderRequest.FromString,
+                    response_serializer=folder__service__pb2.Folder.SerializeToString,
+            ),
+            'RenameFolder': grpc.unary_unary_rpc_method_handler(
+                    servicer.RenameFolder,
+                    request_deserializer=folder__service__pb2.RenameFolderRequest.FromString,
+                    response_serializer=folder__service__pb2.Folder.SerializeToString,
+            ),
+            'MoveFolder': grpc.unary_unary_rpc_method_handler(
+                    servicer.MoveFolder,
+                    request_deserializer=folder__service__pb2.MoveFolderRequest.FromString,
+                    response_serializer=folder__service__pb2.Folder.SerializeToString,
+            ),
+            'DeleteFolder': grpc.unary_unary_rpc_method_handler(
+                    servicer.DeleteFolder,
+                    request_deserializer=folder__service__pb2.DeleteFolderRequest.FromString,
+                    response_serializer=folder__service__pb2.Empty.SerializeToString,
+            ),
+            'ListFolders': grpc.unary_unary_rpc_method_handler(
+                    servicer.ListFolders,
+                    request_deserializer=folder__service__pb2.ListFoldersRequest.FromString,
+                    response_serializer=folder__service__pb2.FolderListResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'folder.FolderService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+    server.add_registered_method_handlers('folder.FolderService', rpc_method_handlers)
+
+
+ # This class is part of an EXPERIMENTAL API.
+class FolderService(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def CreateFolder(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/folder.FolderService/CreateFolder',
+            folder__service__pb2.CreateFolderRequest.SerializeToString,
+            folder__service__pb2.Folder.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def RenameFolder(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/folder.FolderService/RenameFolder',
+            folder__service__pb2.RenameFolderRequest.SerializeToString,
+            folder__service__pb2.Folder.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def MoveFolder(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/folder.FolderService/MoveFolder',
+            folder__service__pb2.MoveFolderRequest.SerializeToString,
+            folder__service__pb2.Folder.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def DeleteFolder(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/folder.FolderService/DeleteFolder',
+            folder__service__pb2.DeleteFolderRequest.SerializeToString,
+            folder__service__pb2.Empty.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def ListFolders(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/folder.FolderService/ListFolders',
+            folder__service__pb2.ListFoldersRequest.SerializeToString,
+            folder__service__pb2.FolderListResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)

file_hub_client/rpc/generate_grpc.py

@@ -0,0 +1,76 @@
+"""
+生成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 = [
+            "python", "-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"]:
+            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() 

file_hub_client/rpc/protos/file_service.proto

@@ -0,0 +1,147 @@
+syntax = "proto3";
+
+package file;
+
+import "google/protobuf/timestamp.proto";
+import "google/protobuf/struct.proto";
+
+// ========= 服务定义 =========
+
+service FileService {
+    rpc UploadFile (UploadFileRequest) returns (UploadFileResponse);
+    rpc GenerateUploadUrl (UploadUrlRequest) returns (UploadUrlResponse);
+    rpc GenerateResumableUploadUrl (UploadUrlRequest) returns (UploadUrlResponse);
+    rpc ConfirmUploadCompleted (UploadCompletedRequest) returns (Empty);
+    rpc GenerateDownloadUrl (DownloadUrlRequest) returns (DownloadUrlResponse);
+    rpc GenerateShareLink (ShareLinkRequest) returns (ShareLinkResponse);
+    rpc VisitFile (FileVisitRequest) returns (Empty);
+    rpc GetFile (GetFileRequest) returns (File);
+    rpc RenameFile (RenameFileRequest) returns (File);
+    rpc DeleteFile (DeleteFileRequest) returns (Empty);
+    rpc ListFiles (ListFilesRequest) returns (FileListResponse);
+}
+
+// ========= 数据结构定义 =========
+
+message File {
+    string id = 1;
+    string folder_id = 2;
+    string file_name = 3;
+    string file_type = 4;
+    google.protobuf.Timestamp created_at = 5;
+    google.protobuf.Timestamp updated_at = 6;
+}
+
+message UploadFile {
+    string id = 1;
+    string folder_id = 2;
+    string file_id = 3;
+    string storage_type = 4;
+    string stored_name = 5;
+    string stored_path = 6;
+    string file_name = 7;
+    int64 file_size = 8;
+    string file_ext = 9;
+    string mime_type = 10;
+    string status = 11;
+    google.protobuf.Timestamp created_at = 12;
+    google.protobuf.Timestamp updated_at = 13;
+}
+
+// ========= 请求结构 =========
+
+message UploadFileRequest {
+    optional string folder_id = 1;
+    string file_name = 2;
+    bytes content = 3;
+    string file_type = 4;
+    string mime_type = 5;
+    optional bool is_temporary = 6;
+    optional int32 expire_seconds = 7;
+}
+
+message UploadUrlRequest {
+    optional string folder_id = 1;
+    string file_name = 2;
+    string file_type = 3;
+    string mime_type = 4;
+    int64 file_size = 5;
+    string file_hash = 6;
+    optional bool is_temporary = 7;
+    optional int32 expire_seconds = 8;
+}
+
+message UploadCompletedRequest {
+    string file_id = 1;
+}
+
+message DownloadUrlRequest {
+    string file_id = 1;
+    optional int32 expire_seconds = 2;
+}
+
+message ShareLinkRequest {
+    string file_id = 1;
+    bool is_public = 2;
+    string access_scope = 3;
+    optional int32 expire_seconds = 4;
+    optional int32 max_access = 5;
+    optional string share_password = 6;
+}
+
+message FileVisitRequest {
+    string file_share_id = 1;
+    string access_type = 2;
+    int32 access_duration = 3;
+    google.protobuf.Struct metadata = 4;
+}
+
+message GetFileRequest {
+    string file_id = 1;
+}
+
+message RenameFileRequest {
+    string file_id = 1;
+    string new_name = 2;
+}
+
+message DeleteFileRequest {
+    string file_id = 1;
+}
+
+message ListFilesRequest {
+    optional string folder_id = 1;
+    optional string file_name = 2;
+    repeated string file_type = 3;
+    optional string created_by_role = 4;
+    optional string created_by = 5;
+    optional int32 page_size = 6;
+    optional int32 page = 7;
+}
+
+// ========= 响应结构 =========
+
+message UploadFileResponse {
+    File file = 1;
+    UploadFile upload_file = 2;
+}
+
+message UploadUrlResponse {
+    File file = 1;
+    UploadFile upload_file = 2;
+    string url = 3;
+}
+
+message DownloadUrlResponse {
+    string url = 1;
+}
+
+message ShareLinkResponse {
+    string file_share_id = 1;
+}
+
+message FileListResponse {
+    repeated File files = 1;
+}
+
+message Empty {}

file_hub_client/rpc/protos/folder_service.proto

@@ -0,0 +1,65 @@
+syntax = "proto3";
+
+package folder;
+
+import "google/protobuf/timestamp.proto";
+
+service FolderService {
+    rpc CreateFolder (CreateFolderRequest) returns (Folder);
+    rpc RenameFolder (RenameFolderRequest) returns (Folder);
+    rpc MoveFolder (MoveFolderRequest) returns (Folder);
+    rpc DeleteFolder (DeleteFolderRequest) returns (Empty);
+    rpc ListFolders (ListFoldersRequest) returns (FolderListResponse);
+}
+
+// ========= 数据结构定义 =========
+
+message Folder {
+    string id = 1;
+    string org_id = 2;
+    string user_id = 3;
+    string folder_name = 4;
+    string parent_id = 5;
+    string created_by = 6;
+    string created_by_role = 7;
+    google.protobuf.Timestamp created_at = 8;
+    google.protobuf.Timestamp updated_at = 9;
+}
+
+// ========= 请求结构 =========
+
+message CreateFolderRequest {
+    string folder_name = 1;
+    optional string parent_id = 2;
+}
+
+message RenameFolderRequest {
+    string folder_id = 1;
+    string new_name = 2;
+}
+
+message MoveFolderRequest {
+    string folder_id = 1;
+    string new_parent_id = 2;
+}
+
+message DeleteFolderRequest {
+    string folder_id = 1;
+}
+
+message ListFoldersRequest {
+    optional string parent_id = 1;
+    optional string org_id = 2;
+    optional string user_id = 3;
+    optional string folder_name = 4;
+    optional string created_by_role = 5;
+    optional string created_by = 6;
+}
+
+// ========= 响应结构 =========
+
+message FolderListResponse {
+    repeated Folder folders = 1;
+}
+
+message Empty {}