import sys
from logging import getLogger
from logging import NullHandler

import grpc

import python_liftbridge.api_pb2_grpc

logger = getLogger(__name__)
logger.addHandler(NullHandler())


class BaseClient(object):

    """
        Connect creates a Client connection for the given Liftbridge cluster.
    """

    def __init__(self, ip_address='127.0.0.1:9292', timeout=5, tls_cert=None):
        self.ip_address = ip_address
        self.timeout = timeout
        if tls_cert:
            logger.debug(
                'Creating a secure channel with address: %s' %
                self.ip_address,
            )
            self.stub = self._secure_connect(tls_cert)
        else:
            logger.debug(
                'Creating an insecure channel with address: %s' % self.ip_address,
            )
            self.channel = grpc.insecure_channel(self.ip_address)
            self.stub = self._insecure_connect()

    def _insecure_connect(self):
        try:
            grpc.channel_ready_future(
                self.channel,
            ).result(timeout=self.timeout)
        except grpc.FutureTimeoutError:
            sys.exit('Error connecting to server')
        else:
            return python_liftbridge.api_pb2_grpc.APIStub(self.channel)

    def _secure_connect(self, secure_file):
        # TODO
        pass

    def close(self):
        logger.debug('Closing channel')
        self.channel.close()

    def __repr__(self):
        return str(self.__dict__)

import sys
from logging import getLogger
from logging import NullHandler

import grpc

import python_liftbridge.api_pb2_grpc

logger = getLogger(__name__)
logger.addHandler(NullHandler())


class BaseClient(object):

    """
        Connect creates a Client connection for the given Liftbridge cluster.
    """

    def __init__(self, ip_address='127.0.0.1:9292', timeout=5, tls_cert=None):
        self.ip_address = ip_address
        self.timeout = timeout
        if tls_cert:
            logger.debug(
                'Creating a secure channel with address: %s' %
                self.ip_address,
            )
            self.stub = self._secure_connect(tls_cert)
        else:
            logger.debug(
                'Creating an insecure channel with address: %s' % self.ip_address,
            )
            self.channel = grpc.insecure_channel(self.ip_address)
            self.stub = self._insecure_connect()

    def _insecure_connect(self):
        try:
            grpc.channel_ready_future(
                self.channel,
            ).result(timeout=self.timeout)
        except grpc.FutureTimeoutError:
            sys.exit('Error connecting to server')
        else:
            return python_liftbridge.api_pb2_grpc.APIStub(self.channel)

    def _secure_connect(self, secure_file):
        # TODO
        pass

    def close(self):
        logger.debug('Closing channel')
        self.channel.close()

    def __repr__(self):
        return str(self.__dict__)

Connect creates a Client connection for the given Liftbridge cluster. # TODO