如何将rabbitMQ消息保存到csv文件？

import pika
import csv

# RabbitMQ 连接信息
hostname = 'localhost'
port = 5672
username = 'guest'
password = 'guest'
queue_name = 'my_queue'

# 连接 RabbitMQ
connection = pika.BlockingConnection(pika.ConnectionParameters(host=hostname, port=port, credentials=pika.PlainCredentials(username, password)))
channel = connection.channel()

# 声明队列
channel.queue_declare(queue=queue_name)

# 创建 CSV 文件
csv_file = open('messages.csv', 'w', newline='')
csv_writer = csv.writer(csv_file)
csv_writer.writerow(['Message'])

# 从队列中获取消息并写入 CSV 文件
def callback(ch, method, properties, body):
    csv_writer.writerow([body.decode()])

channel.basic_consume(queue=queue_name, on_message_callback=callback, auto_ack=True)

# 开始消费消息
channel.start_consuming()

# 关闭连接和文件
connection.close()
csv_file.close()