使用context manager創建thrift client的統一入口

"""Utilities for building the thrift transport client."""
import contextlib

from thrift.protocol import TBinaryProtocol
from thrift.transport import TSocket, TTransport


@contextlib.contextmanager
def client_transport_maker(host, port, rpc_type):
  """Context manager to build the thrift transport client."""
  transport = TSocket.TSocket(host=host, port=port)
  tframed_transport = TTransport.TFramedTransport(transport)
  protocol = TBinaryProtocol.TBinaryProtocol(tframed_transport)
  client = rpc_type.Client(protocol)
  try:
    transport.open()
    yield client
  finally:
    transport.close()