snacks 0.0.2

Wrapper to pika for an easier to use RabbitMq interface.

Snacks

Snacks is a wrapper around pika to provide a convenient interface to publish/subscribe to queues in RabbitMq.

Example

from pika import BlockingConnection, PlainCredentials
from pika.adapters.blocking_connection import BlockingChannel

from snacks.rabbit import Rabbit
from snacks.rabbit_config import RabbitConfig

# Initial setup of configuration and Rabbit class.
config = RabbitConfig(
    host='localhost',
    port=5672,
    exchange_name='',
    credentials=PlainCredentials('snacks', 'snacks')
)
rabbit = Rabbit(config)
# Setup queues to use.
queue = 'snacks'
key = 'snackey'
mq_conn = BlockingConnection(rabbit.config.params)
channel: BlockingChannel = mq_conn.channel()
channel.exchange_declare(
    exchange=rabbit.config.exchange_name,
    exchange_type='topic',
    durable=True
)
channel.queue_declare(queue=queue, durable=True)
channel.queue_bind(
    exchange=rabbit.config.exchange_name,
    queue=queue,
    routing_key=key
)


@rabbit.listener([queue])
def listen(event: str) -> None:
    print(f'Received message: {event}')


if __name__ == '__main__':
    rabbit.publish('To a python, rabbits and pikas are snacks.', key)
    try:
        while True:
            pass
    except KeyboardInterrupt:
        print('Exiting...')

Config

The RabbitConfig object can easily read from yaml or ini config files.

RabbitConfig.from_yaml_file(Path.cwd().joinpath('rabbit.yaml'))
RabbitConfig.from_ini_file(Path.cwd().joinpath('rabbit.ini'))

Config ini file:

[rabbit]
host = localhost
user = snacks
pass = snacks
port = 5672
exchange_name = default

Config yaml file:

host: localhost
user: snacks
pass: snacks
port: 5672
exchange_name: default