Commit 6cf5ce13 authored by Lluis Gifre Renom's avatar Lluis Gifre Renom
Browse files

Corrected genproto script for l3_centralizedattackdetector.

parent 52b94843
Loading
Loading
Loading
Loading
+6 −2
Original line number Diff line number Diff line
@@ -36,10 +36,14 @@ tee proto/__init__.py << EOF > /dev/null

EOF

python -m grpc_tools.protoc -I../../proto --python_out=proto --grpc_python_out=proto l3_centralizedattackdetector.proto
python -m grpc_tools.protoc -I../../proto --python_out=proto --grpc_python_out=proto context.proto
python -m grpc_tools.protoc -I../../proto --python_out=proto --grpc_python_out=proto l3_attackmitigator.proto
python -m grpc_tools.protoc -I../../proto --python_out=proto --grpc_python_out=proto l3_centralizedattackdetector.proto

rm proto/context_pb2_grpc.py
rm proto/l3_attackmitigator_pb2_grpc.py

sed -i -E 's/(import\ .*)_pb2/from . \1_pb2/g' proto/context_pb2.py
sed -i -E 's/(import\ .*)_pb2/from . \1_pb2/g' proto/l3_centralizedattackdetector_pb2.py
sed -i -E 's/(import\ .*)_pb2/from . \1_pb2/g' proto/l3_centralizedattackdetector_pb2_grpc.py
sed -i -E 's/(import\ .*)_pb2/from . \1_pb2/g' proto/l3_attackmitigator_pb2.py
sed -i -E 's/(import\ .*)_pb2/from . \1_pb2/g' proto/l3_attackmitigator_pb2_grpc.py
+3071 −0

File added.

Preview size limit exceeded, changes collapsed.

+0 −102
Original line number Diff line number Diff line
# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
"""Client and server classes corresponding to protobuf-defined services."""
import grpc

from . import context_pb2 as context__pb2
from . import l3_attackmitigator_pb2 as l3__attackmitigator__pb2


class L3AttackmitigatorStub(object):
    """Missing associated documentation comment in .proto file."""

    def __init__(self, channel):
        """Constructor.

        Args:
            channel: A grpc.Channel.
        """
        self.SendOutput = channel.unary_unary(
                '/L3Attackmitigator/SendOutput',
                request_serializer=l3__attackmitigator__pb2.L3AttackmitigatorOutput.SerializeToString,
                response_deserializer=context__pb2.Empty.FromString,
                )
        self.GetMitigation = channel.unary_unary(
                '/L3Attackmitigator/GetMitigation',
                request_serializer=context__pb2.Empty.SerializeToString,
                response_deserializer=context__pb2.Empty.FromString,
                )


class L3AttackmitigatorServicer(object):
    """Missing associated documentation comment in .proto file."""

    def SendOutput(self, request, context):
        """Sends a greeting
        """
        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
        context.set_details('Method not implemented!')
        raise NotImplementedError('Method not implemented!')

    def GetMitigation(self, request, context):
        """Sends another greeting
        """
        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
        context.set_details('Method not implemented!')
        raise NotImplementedError('Method not implemented!')


def add_L3AttackmitigatorServicer_to_server(servicer, server):
    rpc_method_handlers = {
            'SendOutput': grpc.unary_unary_rpc_method_handler(
                    servicer.SendOutput,
                    request_deserializer=l3__attackmitigator__pb2.L3AttackmitigatorOutput.FromString,
                    response_serializer=context__pb2.Empty.SerializeToString,
            ),
            'GetMitigation': grpc.unary_unary_rpc_method_handler(
                    servicer.GetMitigation,
                    request_deserializer=context__pb2.Empty.FromString,
                    response_serializer=context__pb2.Empty.SerializeToString,
            ),
    }
    generic_handler = grpc.method_handlers_generic_handler(
            'L3Attackmitigator', rpc_method_handlers)
    server.add_generic_rpc_handlers((generic_handler,))


 # This class is part of an EXPERIMENTAL API.
class L3Attackmitigator(object):
    """Missing associated documentation comment in .proto file."""

    @staticmethod
    def SendOutput(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, '/L3Attackmitigator/SendOutput',
            l3__attackmitigator__pb2.L3AttackmitigatorOutput.SerializeToString,
            context__pb2.Empty.FromString,
            options, channel_credentials,
            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

    @staticmethod
    def GetMitigation(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, '/L3Attackmitigator/GetMitigation',
            context__pb2.Empty.SerializeToString,
            context__pb2.Empty.FromString,
            options, channel_credentials,
            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)