Fix the import for aio gRPC

pull/22662/head
Lidi Zheng 5 years ago
parent faf946931d
commit b8691f9669
  1. 3
      src/python/grpcio_reflection/grpc_reflection/v1alpha/reflection.py

@ -15,6 +15,7 @@
import sys import sys
import grpc import grpc
from grpc.experimental import aio as grpc_aio
from grpc_reflection.v1alpha import reflection_pb2 as _reflection_pb2 from grpc_reflection.v1alpha import reflection_pb2 as _reflection_pb2
from grpc_reflection.v1alpha import reflection_pb2_grpc as _reflection_pb2_grpc from grpc_reflection.v1alpha import reflection_pb2_grpc as _reflection_pb2_grpc
@ -66,7 +67,7 @@ if sys.version_info[0] >= 3 and sys.version_info[1] >= 6:
server: grpc.Server to which reflection service will be added. server: grpc.Server to which reflection service will be added.
pool: DescriptorPool object to use (descriptor_pool.Default() if None). pool: DescriptorPool object to use (descriptor_pool.Default() if None).
""" """
if isinstance(server, grpc.experimental.aio.Server): if isinstance(server, grpc_aio.Server):
_reflection_pb2_grpc.add_ServerReflectionServicer_to_server( _reflection_pb2_grpc.add_ServerReflectionServicer_to_server(
aio.ReflectionServicer(service_names, pool=pool), server) aio.ReflectionServicer(service_names, pool=pool), server)
else: else:

Loading…
Cancel
Save