flwr 1.21.0__py3-none-any.whl → 1.23.0__py3-none-any.whl

flwr/proto/fleet_pb2_grpc.py

@@ -18,15 +18,25 @@ class FleetStub(object):
         Args:
             channel: A grpc.Channel.
         """
-        self.CreateNode = channel.unary_unary(
-                '/flwr.proto.Fleet/CreateNode',
-                request_serializer=flwr_dot_proto_dot_fleet__pb2.CreateNodeRequest.SerializeToString,
-                response_deserializer=flwr_dot_proto_dot_fleet__pb2.CreateNodeResponse.FromString,
+        self.RegisterNode = channel.unary_unary(
+                '/flwr.proto.Fleet/RegisterNode',
+                request_serializer=flwr_dot_proto_dot_fleet__pb2.RegisterNodeFleetRequest.SerializeToString,
+                response_deserializer=flwr_dot_proto_dot_fleet__pb2.RegisterNodeFleetResponse.FromString,
                 )
-        self.DeleteNode = channel.unary_unary(
-                '/flwr.proto.Fleet/DeleteNode',
-                request_serializer=flwr_dot_proto_dot_fleet__pb2.DeleteNodeRequest.SerializeToString,
-                response_deserializer=flwr_dot_proto_dot_fleet__pb2.DeleteNodeResponse.FromString,
+        self.ActivateNode = channel.unary_unary(
+                '/flwr.proto.Fleet/ActivateNode',
+                request_serializer=flwr_dot_proto_dot_fleet__pb2.ActivateNodeRequest.SerializeToString,
+                response_deserializer=flwr_dot_proto_dot_fleet__pb2.ActivateNodeResponse.FromString,
+                )
+        self.DeactivateNode = channel.unary_unary(
+                '/flwr.proto.Fleet/DeactivateNode',
+                request_serializer=flwr_dot_proto_dot_fleet__pb2.DeactivateNodeRequest.SerializeToString,
+                response_deserializer=flwr_dot_proto_dot_fleet__pb2.DeactivateNodeResponse.FromString,
+                )
+        self.UnregisterNode = channel.unary_unary(
+                '/flwr.proto.Fleet/UnregisterNode',
+                request_serializer=flwr_dot_proto_dot_fleet__pb2.UnregisterNodeFleetRequest.SerializeToString,
+                response_deserializer=flwr_dot_proto_dot_fleet__pb2.UnregisterNodeFleetResponse.FromString,
                 )
         self.SendNodeHeartbeat = channel.unary_unary(
                 '/flwr.proto.Fleet/SendNodeHeartbeat',
@@ -73,14 +83,30 @@ class FleetStub(object):
 class FleetServicer(object):
     """Missing associated documentation comment in .proto file."""
 
-    def CreateNode(self, request, context):
-        """Missing associated documentation comment in .proto file."""
+    def RegisterNode(self, request, context):
+        """Register Node
+        """
         context.set_code(grpc.StatusCode.UNIMPLEMENTED)
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
-    def DeleteNode(self, request, context):
-        """Missing associated documentation comment in .proto file."""
+    def ActivateNode(self, request, context):
+        """Activate Node
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def DeactivateNode(self, request, context):
+        """Deactivate Node
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def UnregisterNode(self, request, context):
+        """Unregister Node
+        """
         context.set_code(grpc.StatusCode.UNIMPLEMENTED)
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
@@ -146,15 +172,25 @@ class FleetServicer(object):
 
 def add_FleetServicer_to_server(servicer, server):
     rpc_method_handlers = {
-            'CreateNode': grpc.unary_unary_rpc_method_handler(
-                    servicer.CreateNode,
-                    request_deserializer=flwr_dot_proto_dot_fleet__pb2.CreateNodeRequest.FromString,
-                    response_serializer=flwr_dot_proto_dot_fleet__pb2.CreateNodeResponse.SerializeToString,
+            'RegisterNode': grpc.unary_unary_rpc_method_handler(
+                    servicer.RegisterNode,
+                    request_deserializer=flwr_dot_proto_dot_fleet__pb2.RegisterNodeFleetRequest.FromString,
+                    response_serializer=flwr_dot_proto_dot_fleet__pb2.RegisterNodeFleetResponse.SerializeToString,
+            ),
+            'ActivateNode': grpc.unary_unary_rpc_method_handler(
+                    servicer.ActivateNode,
+                    request_deserializer=flwr_dot_proto_dot_fleet__pb2.ActivateNodeRequest.FromString,
+                    response_serializer=flwr_dot_proto_dot_fleet__pb2.ActivateNodeResponse.SerializeToString,
             ),
-            'DeleteNode': grpc.unary_unary_rpc_method_handler(
-                    servicer.DeleteNode,
-                    request_deserializer=flwr_dot_proto_dot_fleet__pb2.DeleteNodeRequest.FromString,
-                    response_serializer=flwr_dot_proto_dot_fleet__pb2.DeleteNodeResponse.SerializeToString,
+            'DeactivateNode': grpc.unary_unary_rpc_method_handler(
+                    servicer.DeactivateNode,
+                    request_deserializer=flwr_dot_proto_dot_fleet__pb2.DeactivateNodeRequest.FromString,
+                    response_serializer=flwr_dot_proto_dot_fleet__pb2.DeactivateNodeResponse.SerializeToString,
+            ),
+            'UnregisterNode': grpc.unary_unary_rpc_method_handler(
+                    servicer.UnregisterNode,
+                    request_deserializer=flwr_dot_proto_dot_fleet__pb2.UnregisterNodeFleetRequest.FromString,
+                    response_serializer=flwr_dot_proto_dot_fleet__pb2.UnregisterNodeFleetResponse.SerializeToString,
             ),
             'SendNodeHeartbeat': grpc.unary_unary_rpc_method_handler(
                     servicer.SendNodeHeartbeat,
@@ -207,7 +243,41 @@ class Fleet(object):
     """Missing associated documentation comment in .proto file."""
 
     @staticmethod
-    def CreateNode(request,
+    def RegisterNode(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, '/flwr.proto.Fleet/RegisterNode',
+            flwr_dot_proto_dot_fleet__pb2.RegisterNodeFleetRequest.SerializeToString,
+            flwr_dot_proto_dot_fleet__pb2.RegisterNodeFleetResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def ActivateNode(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, '/flwr.proto.Fleet/ActivateNode',
+            flwr_dot_proto_dot_fleet__pb2.ActivateNodeRequest.SerializeToString,
+            flwr_dot_proto_dot_fleet__pb2.ActivateNodeResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def DeactivateNode(request,
             target,
             options=(),
             channel_credentials=None,
@@ -217,14 +287,14 @@ class Fleet(object):
             wait_for_ready=None,
             timeout=None,
             metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/flwr.proto.Fleet/CreateNode',
-            flwr_dot_proto_dot_fleet__pb2.CreateNodeRequest.SerializeToString,
-            flwr_dot_proto_dot_fleet__pb2.CreateNodeResponse.FromString,
+        return grpc.experimental.unary_unary(request, target, '/flwr.proto.Fleet/DeactivateNode',
+            flwr_dot_proto_dot_fleet__pb2.DeactivateNodeRequest.SerializeToString,
+            flwr_dot_proto_dot_fleet__pb2.DeactivateNodeResponse.FromString,
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
 
     @staticmethod
-    def DeleteNode(request,
+    def UnregisterNode(request,
             target,
             options=(),
             channel_credentials=None,
@@ -234,9 +304,9 @@ class Fleet(object):
             wait_for_ready=None,
             timeout=None,
             metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/flwr.proto.Fleet/DeleteNode',
-            flwr_dot_proto_dot_fleet__pb2.DeleteNodeRequest.SerializeToString,
-            flwr_dot_proto_dot_fleet__pb2.DeleteNodeResponse.FromString,
+        return grpc.experimental.unary_unary(request, target, '/flwr.proto.Fleet/UnregisterNode',
+            flwr_dot_proto_dot_fleet__pb2.UnregisterNodeFleetRequest.SerializeToString,
+            flwr_dot_proto_dot_fleet__pb2.UnregisterNodeFleetResponse.FromString,
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
 

flwr/proto/fleet_pb2_grpc.pyi

@@ -12,13 +12,25 @@ import grpc
 
 class FleetStub:
     def __init__(self, channel: grpc.Channel) -> None: ...
-    CreateNode: grpc.UnaryUnaryMultiCallable[
-        flwr.proto.fleet_pb2.CreateNodeRequest,
-        flwr.proto.fleet_pb2.CreateNodeResponse]
-
-    DeleteNode: grpc.UnaryUnaryMultiCallable[
-        flwr.proto.fleet_pb2.DeleteNodeRequest,
-        flwr.proto.fleet_pb2.DeleteNodeResponse]
+    RegisterNode: grpc.UnaryUnaryMultiCallable[
+        flwr.proto.fleet_pb2.RegisterNodeFleetRequest,
+        flwr.proto.fleet_pb2.RegisterNodeFleetResponse]
+    """Register Node"""
+
+    ActivateNode: grpc.UnaryUnaryMultiCallable[
+        flwr.proto.fleet_pb2.ActivateNodeRequest,
+        flwr.proto.fleet_pb2.ActivateNodeResponse]
+    """Activate Node"""
+
+    DeactivateNode: grpc.UnaryUnaryMultiCallable[
+        flwr.proto.fleet_pb2.DeactivateNodeRequest,
+        flwr.proto.fleet_pb2.DeactivateNodeResponse]
+    """Deactivate Node"""
+
+    UnregisterNode: grpc.UnaryUnaryMultiCallable[
+        flwr.proto.fleet_pb2.UnregisterNodeFleetRequest,
+        flwr.proto.fleet_pb2.UnregisterNodeFleetResponse]
+    """Unregister Node"""
 
     SendNodeHeartbeat: grpc.UnaryUnaryMultiCallable[
         flwr.proto.heartbeat_pb2.SendNodeHeartbeatRequest,
@@ -67,16 +79,36 @@ class FleetStub:
 
 class FleetServicer(metaclass=abc.ABCMeta):
     @abc.abstractmethod
-    def CreateNode(self,
-        request: flwr.proto.fleet_pb2.CreateNodeRequest,
+    def RegisterNode(self,
+        request: flwr.proto.fleet_pb2.RegisterNodeFleetRequest,
+        context: grpc.ServicerContext,
+    ) -> flwr.proto.fleet_pb2.RegisterNodeFleetResponse:
+        """Register Node"""
+        pass
+
+    @abc.abstractmethod
+    def ActivateNode(self,
+        request: flwr.proto.fleet_pb2.ActivateNodeRequest,
         context: grpc.ServicerContext,
-    ) -> flwr.proto.fleet_pb2.CreateNodeResponse: ...
+    ) -> flwr.proto.fleet_pb2.ActivateNodeResponse:
+        """Activate Node"""
+        pass
 
     @abc.abstractmethod
-    def DeleteNode(self,
-        request: flwr.proto.fleet_pb2.DeleteNodeRequest,
+    def DeactivateNode(self,
+        request: flwr.proto.fleet_pb2.DeactivateNodeRequest,
         context: grpc.ServicerContext,
-    ) -> flwr.proto.fleet_pb2.DeleteNodeResponse: ...
+    ) -> flwr.proto.fleet_pb2.DeactivateNodeResponse:
+        """Deactivate Node"""
+        pass
+
+    @abc.abstractmethod
+    def UnregisterNode(self,
+        request: flwr.proto.fleet_pb2.UnregisterNodeFleetRequest,
+        context: grpc.ServicerContext,
+    ) -> flwr.proto.fleet_pb2.UnregisterNodeFleetResponse:
+        """Unregister Node"""
+        pass
 
     @abc.abstractmethod
     def SendNodeHeartbeat(self,

flwr/proto/node_pb2.py

@@ -14,7 +14,7 @@ _sym_db = _symbol_database.Default()
 
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x15\x66lwr/proto/node.proto\x12\nflwr.proto\"\x17\n\x04Node\x12\x0f\n\x07node_id\x18\x01 \x01(\x04\x62\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x15\x66lwr/proto/node.proto\x12\nflwr.proto\"\x17\n\x04Node\x12\x0f\n\x07node_id\x18\x01 \x01(\x04\"\xd3\x02\n\x08NodeInfo\x12\x0f\n\x07node_id\x18\x01 \x01(\x04\x12\x11\n\towner_aid\x18\x02 \x01(\t\x12\x0e\n\x06status\x18\x03 \x01(\t\x12\x15\n\rregistered_at\x18\x04 \x01(\t\x12\x1e\n\x11last_activated_at\x18\x05 \x01(\tH\x00\x88\x01\x01\x12 \n\x13last_deactivated_at\x18\x06 \x01(\tH\x01\x88\x01\x01\x12\x1c\n\x0funregistered_at\x18\x07 \x01(\tH\x02\x88\x01\x01\x12\x19\n\x0conline_until\x18\x08 \x01(\x01H\x03\x88\x01\x01\x12\x1a\n\x12heartbeat_interval\x18\t \x01(\x01\x12\x12\n\npublic_key\x18\n \x01(\x0c\x42\x14\n\x12_last_activated_atB\x16\n\x14_last_deactivated_atB\x12\n\x10_unregistered_atB\x0f\n\r_online_untilb\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -23,4 +23,6 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   DESCRIPTOR._options = None
   _globals['_NODE']._serialized_start=37
   _globals['_NODE']._serialized_end=60
+  _globals['_NODEINFO']._serialized_start=63
+  _globals['_NODEINFO']._serialized_end=402
 # @@protoc_insertion_point(module_scope)

flwr/proto/node_pb2.pyi

@@ -5,6 +5,7 @@ isort:skip_file
 import builtins
 import google.protobuf.descriptor
 import google.protobuf.message
+import typing
 import typing_extensions
 
 DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
@@ -19,3 +20,50 @@ class Node(google.protobuf.message.Message):
         ) -> None: ...
     def ClearField(self, field_name: typing_extensions.Literal["node_id",b"node_id"]) -> None: ...
 global___Node = Node
+
+class NodeInfo(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+    NODE_ID_FIELD_NUMBER: builtins.int
+    OWNER_AID_FIELD_NUMBER: builtins.int
+    STATUS_FIELD_NUMBER: builtins.int
+    REGISTERED_AT_FIELD_NUMBER: builtins.int
+    LAST_ACTIVATED_AT_FIELD_NUMBER: builtins.int
+    LAST_DEACTIVATED_AT_FIELD_NUMBER: builtins.int
+    UNREGISTERED_AT_FIELD_NUMBER: builtins.int
+    ONLINE_UNTIL_FIELD_NUMBER: builtins.int
+    HEARTBEAT_INTERVAL_FIELD_NUMBER: builtins.int
+    PUBLIC_KEY_FIELD_NUMBER: builtins.int
+    node_id: builtins.int
+    owner_aid: typing.Text
+    status: typing.Text
+    registered_at: typing.Text
+    last_activated_at: typing.Text
+    last_deactivated_at: typing.Text
+    unregistered_at: typing.Text
+    online_until: builtins.float
+    heartbeat_interval: builtins.float
+    public_key: builtins.bytes
+    def __init__(self,
+        *,
+        node_id: builtins.int = ...,
+        owner_aid: typing.Text = ...,
+        status: typing.Text = ...,
+        registered_at: typing.Text = ...,
+        last_activated_at: typing.Optional[typing.Text] = ...,
+        last_deactivated_at: typing.Optional[typing.Text] = ...,
+        unregistered_at: typing.Optional[typing.Text] = ...,
+        online_until: typing.Optional[builtins.float] = ...,
+        heartbeat_interval: builtins.float = ...,
+        public_key: builtins.bytes = ...,
+        ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["_last_activated_at",b"_last_activated_at","_last_deactivated_at",b"_last_deactivated_at","_online_until",b"_online_until","_unregistered_at",b"_unregistered_at","last_activated_at",b"last_activated_at","last_deactivated_at",b"last_deactivated_at","online_until",b"online_until","unregistered_at",b"unregistered_at"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["_last_activated_at",b"_last_activated_at","_last_deactivated_at",b"_last_deactivated_at","_online_until",b"_online_until","_unregistered_at",b"_unregistered_at","heartbeat_interval",b"heartbeat_interval","last_activated_at",b"last_activated_at","last_deactivated_at",b"last_deactivated_at","node_id",b"node_id","online_until",b"online_until","owner_aid",b"owner_aid","public_key",b"public_key","registered_at",b"registered_at","status",b"status","unregistered_at",b"unregistered_at"]) -> None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["_last_activated_at",b"_last_activated_at"]) -> typing.Optional[typing_extensions.Literal["last_activated_at"]]: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["_last_deactivated_at",b"_last_deactivated_at"]) -> typing.Optional[typing_extensions.Literal["last_deactivated_at"]]: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["_online_until",b"_online_until"]) -> typing.Optional[typing_extensions.Literal["online_until"]]: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["_unregistered_at",b"_unregistered_at"]) -> typing.Optional[typing_extensions.Literal["unregistered_at"]]: ...
+global___NodeInfo = NodeInfo