AMQP Low-Level Client

import os
from amqp import Connection, BasicMessage
import time

# Replace with your AMQP broker URL (e.g., 'amqp://guest:guest@localhost:5672/')
AMQP_URL = os.environ.get('AMQP_BROKER_URL', 'amqp://guest:guest@localhost:5672/')
QUEUE_NAME = 'my_test_queue'

def publish_message(body, connection_url):
    with Connection(connection_url) as connection:
        channel = connection.channel()
        channel.queue_declare(queue=QUEUE_NAME, durable=True)
        message = BasicMessage(body=body.encode('utf-8'))
        channel.basic_publish(message, routing_key=QUEUE_NAME)
        print(f"[x] Sent '{body}'")

def consume_message(connection_url):
    with Connection(connection_url) as connection:
        channel = connection.channel()
        channel.queue_declare(queue=QUEUE_NAME, durable=True)

        def callback(message):
            print(f"[x] Received '{message.body.decode()}'")
            channel.basic_ack(message.delivery_tag)

        channel.basic_consume(queue=QUEUE_NAME, callback=callback)
        print(f' [*] Waiting for messages on {QUEUE_NAME}. To exit press CTRL+C')
        try:
            while True:
                connection.drain_events() # Crucial for processing events and heartbeats
        except KeyboardInterrupt:
            print("Exiting consumer.")

if __name__ == "__main__":
    # Example usage: Publisher
    print("--- Publisher ---")
    publish_message("Hello, AMQP!", AMQP_URL)
    publish_message("Another message.", AMQP_URL)

    # Example usage: Consumer (run in a separate process/thread or after publisher finishes)
    # For demonstration, we'll wait a bit and then consume
    print("\n--- Consumer ---")
    time.sleep(2) # Give publisher time to send messages
    consume_message(AMQP_URL)