RabbitMQ ping

ping.py

import rabbitpy

def send():
    with rabbitpy.Connection() as conn:
        with conn.channel() as chan:
            ex = rabbitpy.DirectExchange(chan, "ping_ex")
            ex.declare()
            while True:
                m = rabbitpy.Message(chan, "ping")
                m.publish(ex, "ping_ex")
                yield


def collect():
    with rabbitpy.Connection() as conn:
        with conn.channel() as chan:
            ex = rabbitpy.DirectExchange(chan, "ping_ex")
            ex.declare()
            q = rabbitpy.Queue(chan)
            q.declare()
            q.bind(ex, "ping_ex")
            for m in q:
                m.ack()
                yield m.body

if __name__ == "__main__":
    import sys
    import time
    if len(sys.argv) == 2 and sys.argv[1] == "send":
        for x in send():
            time.sleep(1)
    else:
        for x in collect():
            print(x.decode("utf8"))