chore: grpc client implemented

Author: Javier-Godon

Makefile

@@ -0,0 +1,13 @@
+PROTO_DIR := proto
+OUT_DIR := app/grpc_client_impl/proto
+PROTOC_CMD := python -m grpc_tools.protoc
+
+.PHONY: all clean
+
+all:
+	$(PROTOC_CMD) -I$(PROTO_DIR) --proto_path=$(PROTO_DIR) \
+		--python_out=$(OUT_DIR) --pyi_out=$(OUT_DIR) --grpc_python_out=$(OUT_DIR) \
+		$(PROTO_DIR)/*.proto
+
+clean:
+	rm -rf $(OUT_DIR)/*.py $(OUT_DIR)/*.pyi

app/grpc_client_impl/__init__.py

@@ -0,0 +1,31 @@
+import grpc
+
+import app.grpc_client_impl.proto.get_cpu_system_usage_pb2 as grpc_system_proto_contracts
+import app.grpc_client_impl.proto.get_cpu_system_usage_pb2_grpc as grpc_system_proto_client
+import app.grpc_client_impl.proto.get_cpu_user_usage_pb2 as grpc_user_proto_contracts
+import app.grpc_client_impl.proto.get_cpu_user_usage_pb2_grpc as grpc_user_proto_client
+
+
+class GRPCClient:
+    def __init__(self, channel: grpc.Channel):
+        self.channel = channel
+        self.system_stub = grpc_system_proto_client.GetCpuSystemUsageServiceStub(channel)
+        self.user_stub = grpc_user_proto_client.GetCpuUserUsageServiceStub(channel)
+
+    def get_cpu_system_usage(self, date_from, date_to) -> grpc_system_proto_contracts.GetCpuSystemUsageResponse:
+        request = grpc_system_proto_contracts.GetCpuSystemUsageRequest(date_from=date_from, date_to=date_to)
+        return self.system_stub.GetCpuSystemUsage(request)
+
+    def get_cpu_user_usage(self, date_from, date_to) -> grpc_user_proto_contracts.GetCpuUserUsageResponse:
+        request = grpc_user_proto_contracts.GetCpuUserUsageRequest(date_from=date_from, date_to=date_to)
+        return self.user_stub.GetCpuUserUsage(request)
+
+class GRPCClientSingleton:
+    _instance = None
+
+    @classmethod
+    def get_instance(cls):
+        if cls._instance is None:
+            channel = grpc.insecure_channel("localhost:50051")
+            cls._instance = GRPCClient(channel)
+        return cls._instance

app/grpc_client_impl/grpc_client.py

@@ -0,0 +1,32 @@
+from google.protobuf.internal import containers as _containers
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class GetCpuSystemUsageRequest(_message.Message):
+    __slots__ = ("date_from", "date_to")
+    DATE_FROM_FIELD_NUMBER: _ClassVar[int]
+    DATE_TO_FIELD_NUMBER: _ClassVar[int]
+    date_from: int
+    date_to: int
+    def __init__(self, date_from: _Optional[int] = ..., date_to: _Optional[int] = ...) -> None: ...
+
+class CpuUsage(_message.Message):
+    __slots__ = ("cpu", "avg_usage", "max_usage", "min_usage")
+    CPU_FIELD_NUMBER: _ClassVar[int]
+    AVG_USAGE_FIELD_NUMBER: _ClassVar[int]
+    MAX_USAGE_FIELD_NUMBER: _ClassVar[int]
+    MIN_USAGE_FIELD_NUMBER: _ClassVar[int]
+    cpu: str
+    avg_usage: float
+    max_usage: float
+    min_usage: float
+    def __init__(self, cpu: _Optional[str] = ..., avg_usage: _Optional[float] = ..., max_usage: _Optional[float] = ..., min_usage: _Optional[float] = ...) -> None: ...
+
+class GetCpuSystemUsageResponse(_message.Message):
+    __slots__ = ("usages",)
+    USAGES_FIELD_NUMBER: _ClassVar[int]
+    usages: _containers.RepeatedCompositeFieldContainer[CpuUsage]
+    def __init__(self, usages: _Optional[_Iterable[_Union[CpuUsage, _Mapping]]] = ...) -> None: ...

app/grpc_client_impl/proto/__init__.py

@@ -0,0 +1,97 @@
+# 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 app.grpc_client_impl.proto.get_cpu_system_usage_pb2 as get__cpu__system__usage__pb2
+
+GRPC_GENERATED_VERSION = '1.71.0'
+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 get_cpu_system_usage_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 GetCpuSystemUsageServiceStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.GetCpuSystemUsage = channel.unary_unary(
+                '/cpu_system_usage.GetCpuSystemUsageService/GetCpuSystemUsage',
+                request_serializer=get__cpu__system__usage__pb2.GetCpuSystemUsageRequest.SerializeToString,
+                response_deserializer=get__cpu__system__usage__pb2.GetCpuSystemUsageResponse.FromString,
+                _registered_method=True)
+
+
+class GetCpuSystemUsageServiceServicer(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def GetCpuSystemUsage(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_GetCpuSystemUsageServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'GetCpuSystemUsage': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetCpuSystemUsage,
+                    request_deserializer=get__cpu__system__usage__pb2.GetCpuSystemUsageRequest.FromString,
+                    response_serializer=get__cpu__system__usage__pb2.GetCpuSystemUsageResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'cpu_system_usage.GetCpuSystemUsageService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+    server.add_registered_method_handlers('cpu_system_usage.GetCpuSystemUsageService', rpc_method_handlers)
+
+
+ # This class is part of an EXPERIMENTAL API.
+class GetCpuSystemUsageService(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def GetCpuSystemUsage(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,
+            '/cpu_system_usage.GetCpuSystemUsageService/GetCpuSystemUsage',
+            get__cpu__system__usage__pb2.GetCpuSystemUsageRequest.SerializeToString,
+            get__cpu__system__usage__pb2.GetCpuSystemUsageResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)

app/grpc_client_impl/proto/get_cpu_system_usage_pb2.py

@@ -0,0 +1,32 @@
+from google.protobuf.internal import containers as _containers
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class GetCpuUserUsageRequest(_message.Message):
+    __slots__ = ("date_from", "date_to")
+    DATE_FROM_FIELD_NUMBER: _ClassVar[int]
+    DATE_TO_FIELD_NUMBER: _ClassVar[int]
+    date_from: int
+    date_to: int
+    def __init__(self, date_from: _Optional[int] = ..., date_to: _Optional[int] = ...) -> None: ...
+
+class CpuUsage(_message.Message):
+    __slots__ = ("cpu", "avg_usage", "max_usage", "min_usage")
+    CPU_FIELD_NUMBER: _ClassVar[int]
+    AVG_USAGE_FIELD_NUMBER: _ClassVar[int]
+    MAX_USAGE_FIELD_NUMBER: _ClassVar[int]
+    MIN_USAGE_FIELD_NUMBER: _ClassVar[int]
+    cpu: str
+    avg_usage: float
+    max_usage: float
+    min_usage: float
+    def __init__(self, cpu: _Optional[str] = ..., avg_usage: _Optional[float] = ..., max_usage: _Optional[float] = ..., min_usage: _Optional[float] = ...) -> None: ...
+
+class GetCpuUserUsageResponse(_message.Message):
+    __slots__ = ("usages",)
+    USAGES_FIELD_NUMBER: _ClassVar[int]
+    usages: _containers.RepeatedCompositeFieldContainer[CpuUsage]
+    def __init__(self, usages: _Optional[_Iterable[_Union[CpuUsage, _Mapping]]] = ...) -> None: ...

app/grpc_client_impl/proto/get_cpu_system_usage_pb2.pyi

@@ -0,0 +1,97 @@
+# 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 app.grpc_client_impl.proto.get_cpu_user_usage_pb2 as get__cpu__user__usage__pb2
+
+GRPC_GENERATED_VERSION = '1.71.0'
+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 get_cpu_user_usage_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 GetCpuUserUsageServiceStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.GetCpuUserUsage = channel.unary_unary(
+                '/cpu_user_usage.GetCpuUserUsageService/GetCpuUserUsage',
+                request_serializer=get__cpu__user__usage__pb2.GetCpuUserUsageRequest.SerializeToString,
+                response_deserializer=get__cpu__user__usage__pb2.GetCpuUserUsageResponse.FromString,
+                _registered_method=True)
+
+
+class GetCpuUserUsageServiceServicer(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def GetCpuUserUsage(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_GetCpuUserUsageServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'GetCpuUserUsage': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetCpuUserUsage,
+                    request_deserializer=get__cpu__user__usage__pb2.GetCpuUserUsageRequest.FromString,
+                    response_serializer=get__cpu__user__usage__pb2.GetCpuUserUsageResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'cpu_user_usage.GetCpuUserUsageService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+    server.add_registered_method_handlers('cpu_user_usage.GetCpuUserUsageService', rpc_method_handlers)
+
+
+ # This class is part of an EXPERIMENTAL API.
+class GetCpuUserUsageService(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def GetCpuUserUsage(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,
+            '/cpu_user_usage.GetCpuUserUsageService/GetCpuUserUsage',
+            get__cpu__user__usage__pb2.GetCpuUserUsageRequest.SerializeToString,
+            get__cpu__user__usage__pb2.GetCpuUserUsageResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)