# (c) 2014, The MITRE Corporation.  All rights reserved.
# Source code distributed pursuant to license agreement.

class UnknownConnector(Exception):
    """
    Exception for dealing with an unknown connector type.
    """

    def __init__(self, value):
        self.value = value

    def __str__(self):
        return repr(self.value)

class MissingDependency(Exception):
    """
    Exception for dealing with a missing dependency.
    """

    def __init__(self, value):
        self.value = value

    def __str__(self):
        return repr(self.value)

class MissingConfiguration(Exception):
    """
    Exception for dealing with a missing configuration for a connector.
    """

    def __init__(self, value):
        self.value = value

    def __str__(self):
        return repr(self.value)

class Connector(object):
    """
    Service connector to help with distributed services.
    """

    # list of supported connectors
    connectors = ['amqp']
    connection = None

    def __init__(self, connector=None, *args, **kwargs):
        """
        Setup connector if supplied. **kwargs will be passed along as additional
        arguments to the connector.

        :param connector: The connector type to make a connection for.
        :type connector: str
        """

        if connector:
            self.connector(connector, args, **kwargs)

    def connector(self, connector=None, *args, **kwargs):
        """
        Setup connector if supplied. **kwargs will be passed along as arguments
        for your connection.

        :param connector: The connector type to make a connection for.
        :type connector: str
        :raises: UnknownConnector
        """

        # If we don't have a connector, or we received an unknown one, raise an
        # Exception.
        if not connector:
            raise UnknownConnector("Need to supply a connector!")
        if connector not in self.connectors:
            raise UnknownConnector("Connector %s is unknown!" % connector)

        # AMQP connection
        if connector == 'amqp':
            if 'uri' not in kwargs:
                raise MissingConfiguration("Need to provide 'uri'!")
            uri = kwargs['uri']
            del kwargs['uri']
            self.connection = self._amqp_connector(uri=uri, **kwargs)
            self.connection.connect()

    def _amqp_connector(self, uri=None, *args, **kwargs):
        """
        AMQP connector. Creates a connection to the AMQP server and returns it.
        Any extra kwargs you wish to pass to Connection() can be passed via
        **kwargs. See Kombu documentation for supported arguments.

        :param uri: The AMQP URI.
        :type uri: str - Ex: amqp://user:pass@host:port//
        :returns: :class:`kombu.Connection`
        """

        try:
            from kombu import Connection
        except:
            raise MissingDependency("Could not find Kombu. Is it installed?")
        if not uri:
            raise MissingConfiguration("Need an AMQP URI to connect to.")
        return Connection(uri, **kwargs)

    def release(self):
        self.connection.release()

@@ -3071,22 +3071,432 @@
 def 
-release(self):
+send_msg(self, msg, routing_key=None):%0A        if not self.connection:%0A            raise MissingConfiguration(%22Missing connection!%22)%0A%0A        from kombu import Exchange, Producer%0A        exch = Exchange('woodchipper', type='topic')%0A        prod = Producer(self.connection, exchange=exch)%0A        prod.publish(msg, routing_key=routing_key)%0A%0A    def release(self):%0A        %22%22%22%0A        Release existing connection.%0A        %22%22%22%0A
 %0A   

Add a send_msg() and docstrings for release().