pika.py

import pika
import logging

class Producer:

    def __init__(self,host='localhost', exchanges=[], queues=[], conn_params=None):
        self.queueNames = [ x.get('queue') for queue in queues ]
        self.exchangeNames = [ x.get('exchange') for exchange in exchanges ]
        if conn_params:
            self.connection = pika.BlockingConnection(conn_params)
        else:
            self.connection = pika.BlockingConnection(pika.ConnectionParameters(host))
        self.channel = self.connection.channel()
        for exchange in exchanges:
            self.channel.exchange_declare(**exchage)
        for queue in queues:
            self.channel.queue_declare(**queues)

    def bind(queue , exchange, routing_key=""):
        if queue not in self.queueNames:
            logging.FATAL("Failed to find the queue.")
            return
        
        if exchange not in self.exchangeNames:
            logging.FATAL("Failed to find the exchange")
            return
        
        self.channel.queue_bind(queue, exchange, routing_key)

    def close():
        for q in self.queueNames:
            self.channel.queue_delete(q)
        for e in self.exchangeNames:
            self.channel.exchange_delete(e)
        

    def __call__(self):
        return self.channel