grpcio_server.py

from typing import Dict, TYPE_CHECKING
from abc import ABC, abstractmethod

if TYPE_CHECKING:
    import grpc


class ServicerBase(ABC):
    """
    Base class for async grpcio servers.
    """

    @property
    @abstractmethod
    def __rpc_methods__(self) -> Dict[str, "grpc.RpcMethodHandler"]:
        ...

    @property
    @abstractmethod
    def __proto_path__(self) -> str:
        ...


def register_servicers(server: "grpc.aio.Server", *servicers: ServicerBase):
    from grpc import method_handlers_generic_handler

    server.add_generic_rpc_handlers(
        tuple(
            method_handlers_generic_handler(
                servicer.__proto_path__, servicer.__rpc_handlers__
            )
            for servicer in servicers
        )
    )