消息推送系统

import pika

def send_message(message):

connection = pika.BlockingConnection(pika.ConnectionParameters('localhost'))

channel = connection.channel()

channel.queue_declare(queue='notification')

channel.basic_publish(exchange='',

routing_key='notification',

body=message)

print(" [x] Sent %r" % message)

connection.close()

if __name__ == '__main__':

send_message("这是一条测试通知")