def teardown(self): registry._disabled_content_types.clear() registry._set_default_serializer('json') try: registry.unregister('auth') except SerializerNotInstalled: pass
import logging from kombu import Connection, Exchange, Queue from kombu.serialization import registry from datawire.core import app from datawire.util import JSONEncoder, queue_loads log = logging.getLogger(__name__) registry.unregister('json') registry.register('json', JSONEncoder().encode, queue_loads, 'application/json') exchange = Exchange(app.config.get('INSTANCE', 'dwre'), 'topic', durable=True) inbound_queue = Queue('inbound', exchange=exchange, routing_key='inbound.#') matching_queue = Queue('matching', exchange=exchange, routing_key='matching.#') entity_queue = Queue('entity', exchange=exchange, routing_key='entity.#') indexing_queue = Queue('indexing', exchange=exchange, routing_key='indexing.#') def connect(): return Connection(app.config.get('AMQP_QUEUE_URI')) def make_consumer(connection, queue, callback): consumer = connection.Consumer([queue], callbacks=[callback]) consumer.consume() log.info("Listening on: %s", queue)