add wrappers, simple test is working, but problem to got structure back

Author: slampy97

tests/coordination/coordination_client.py

@@ -0,0 +1,33 @@
+import ydb
+from ydb._grpc.v4.protos import ydb_coordination_pb2
+
+
+def test_coordination_nodes(driver_sync: ydb.Driver):
+    client = driver_sync.coordination_client
+    node_path = "/local/test_node"
+
+    try:
+        drop_res = client.delete_node(node_path)
+        print(f"Node deleted (pre-clean), operation id: {drop_res.operation.id}")
+    except ydb.SchemeError:
+        pass
+
+    create_res = client.create_node(node_path)
+    assert create_res.operation.id is not None
+    print(f"Node created, operation id: {create_res.operation.id}")
+
+    describe_res = client.describe_node(node_path)
+    assert describe_res.operation.id is not None
+    print(f"Node described, operation id: {describe_res.operation.id}")
+
+    describe_result_proto = ydb_coordination_pb2.DescribeNodeResult()
+    describe_res.operation.result.Unpack(describe_result_proto)
+
+    print(f"Node path: {describe_result_proto.path}")
+    if describe_result_proto.HasField("config"):
+        print(f"Node config: {describe_result_proto.config}")
+
+    # --- Удаляем узел ---
+    drop_res = client.delete_node(node_path)
+    assert drop_res.operation.id is not None
+    print(f"Node deleted, operation id: {drop_res.operation.id}")

ydb/_apis.py

@@ -11,6 +11,7 @@
         ydb_operation_v1_pb2_grpc,
         ydb_topic_v1_pb2_grpc,
         ydb_query_v1_pb2_grpc,
+        ydb_coordination_v1_pb2_grpc,
     )
 
     from ._grpc.v4.protos import (
@@ -22,6 +23,7 @@
         ydb_operation_pb2,
         ydb_common_pb2,
         ydb_query_pb2,
+        ydb_coordination_pb2,
     )
 
 else:
@@ -33,6 +35,7 @@
         ydb_operation_v1_pb2_grpc,
         ydb_topic_v1_pb2_grpc,
         ydb_query_v1_pb2_grpc,
+        ydb_coordination_v1_pb2_grpc,
     )
 
     from ._grpc.common.protos import (
@@ -44,6 +47,7 @@
         ydb_operation_pb2,
         ydb_common_pb2,
         ydb_query_pb2,
+        ydb_coordination_pb2,
     )
 
 
@@ -56,6 +60,7 @@
 ydb_discovery = ydb_discovery_pb2
 ydb_operation = ydb_operation_pb2
 ydb_query = ydb_query_pb2
+ydb_coordination = ydb_coordination_pb2
 
 
 class CmsService(object):
@@ -134,3 +139,19 @@ class QueryService(object):
     ExecuteQuery = "ExecuteQuery"
     ExecuteScript = "ExecuteScript"
     FetchScriptResults = "FetchScriptResults"
+
+class CoordinationService(object):
+    Stub = ydb_coordination_v1_pb2_grpc.CoordinationServiceStub
+
+    Session = "Session"
+    CreateNode = "CreateNode"
+    AlterNode = "AlterNode"
+    DropNode = "DropNode"
+    DescribeNode = "DescribeNode"
+
+    Request = ydb_coordination.CreateNodeRequest
+    Response = ydb_coordination.CreateNodeResponse
+    DescribeRequest = ydb_coordination.DescribeNodeRequest
+    DescribeResponse = ydb_coordination.DescribeNodeResponse
+    DropRequest = ydb_coordination.DropNodeRequest
+    DropResponse = ydb_coordination.DropNodeResponse

ydb/_grpc/grpcwrapper/ydb_coordination.py

@@ -0,0 +1,93 @@
+import typing
+from dataclasses import dataclass
+
+import ydb
+
+if typing.TYPE_CHECKING:
+    from ..v4.protos import ydb_coordination_pb2
+else:
+    from ..common.protos import ydb_coordination_pb2
+
+from .common_utils import IToProto, IFromProto, ServerStatus
+from . import ydb_coordination_public_types as public_types
+
+
+# -------------------- Requests --------------------
+
+@dataclass
+class CreateNodeRequest(IToProto):
+    path: str
+    config: typing.Optional[public_types.NodeConfig] = None
+    operation_params: typing.Any = None
+
+    def to_proto(self) -> ydb_coordination_pb2.CreateNodeRequest:
+        cfg_proto = self.config.to_proto() if self.config else None
+        return ydb_coordination_pb2.CreateNodeRequest(
+            path=self.path,
+            config=cfg_proto,
+            operation_params=self.operation_params,
+        )
+
+
+@dataclass
+class DescribeNodeRequest(IToProto):
+    path: str
+    operation_params: typing.Any = None
+
+    def to_proto(self) -> ydb_coordination_pb2.DescribeNodeRequest:
+        return ydb_coordination_pb2.DescribeNodeRequest(
+            path=self.path,
+            operation_params=self.operation_params,
+        )
+
+
+@dataclass
+class DropNodeRequest(IToProto):
+    path: str
+    operation_params: typing.Any = None
+
+    def to_proto(self) -> ydb_coordination_pb2.DropNodeRequest:
+        return ydb_coordination_pb2.DropNodeRequest(
+            path=self.path,
+            operation_params=self.operation_params,
+        )
+
+
+@dataclass
+class CreateNodeResponse(IFromProto):
+    operation : ydb.Operation
+    OPERATION_FIELD_NUMBER : int
+
+    @staticmethod
+    def from_proto(msg: ydb_coordination_pb2.CreateNodeResponse) -> "CreateNodeResponse":
+        return CreateNodeResponse(
+            operation=msg.operation,
+            OPERATION_FIELD_NUMBER=msg.OPERATION_FIELD_NUMBER
+        )
+
+
+@dataclass
+class DescribeNodeResponse(IFromProto):
+    operation : ydb.Operation
+    OPERATION_FIELD_NUMBER : int
+
+    @staticmethod
+    def from_proto(msg: "ydb_coordination_pb2.DescribeNodeResponse") -> "DescribeNodeResponse":
+        return DescribeNodeResponse(
+            operation=msg.operation,
+            OPERATION_FIELD_NUMBER=msg.OPERATION_FIELD_NUMBER
+        )
+
+
+@dataclass
+class DropNodeResponse(IFromProto):
+    operation : ydb.Operation
+    OPERATION_FIELD_NUMBER : int
+
+    @staticmethod
+    def from_proto(msg: ydb_coordination_pb2.DropNodeResponse) -> "DropNodeResponse":
+        return DropNodeResponse(
+            operation=msg.operation,
+            OPERATION_FIELD_NUMBER=msg.OPERATION_FIELD_NUMBER
+        )
+

ydb/_grpc/grpcwrapper/ydb_coordination_public_types.py

@@ -0,0 +1,47 @@
+from dataclasses import dataclass
+from enum import Enum
+import typing
+
+if typing.TYPE_CHECKING:
+    from ..v4.protos import ydb_coordination_pb2
+else:
+    from ..common.protos import ydb_coordination_pb2
+
+class ConsistencyMode(Enum):
+    STRICT = 0
+    RELAXED = 1
+
+class RateLimiterCountersMode(Enum):
+    NONE = 0
+    BASIC = 1
+    FULL = 2
+
+@dataclass
+class NodeConfig:
+    attach_consistency_mode: ConsistencyMode
+    path: str
+    rate_limiter_counters_mode: RateLimiterCountersMode
+    read_consistency_mode: ConsistencyMode
+    self_check_period_millis: int
+    session_grace_period_millis: int
+
+    @staticmethod
+    def from_proto(msg: ydb_coordination_pb2.Config) -> "NodeConfig":
+        return NodeConfig(
+            attach_consistency_mode=ConsistencyMode(msg.attach_consistency_mode),
+            path=msg.path,
+            rate_limiter_counters_mode=RateLimiterCountersMode(msg.rate_limiter_counters_mode),
+            read_consistency_mode=ConsistencyMode(msg.read_consistency_mode),
+            self_check_period_millis=msg.self_check_period_millis,
+            session_grace_period_millis=msg.session_grace_period_millis,
+        )
+
+    def to_proto(self) -> ydb_coordination_pb2.Config:
+        return ydb_coordination_pb2.Config(
+            attach_consistency_mode=self.attach_consistency_mode.value,
+            path=self.path,
+            rate_limiter_counters_mode=self.rate_limiter_counters_mode.value,
+            read_consistency_mode=self.read_consistency_mode.value,
+            self_check_period_millis=self.self_check_period_millis,
+            session_grace_period_millis=self.session_grace_period_millis,
+        )