在python中，如何从一个套接字接收数据并将其发送到另一个套接字？

import socket

def transfer_data(source_socket, destination_socket):
    while True:
        data = source_socket.recv(1024)  # 从源套接字接收数据
        if not data:
            break
        destination_socket.sendall(data)  # 将数据发送到目标套接字

# 创建源套接字
source_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
source_socket.bind(('localhost', 8000))
source_socket.listen(1)
source_conn, source_addr = source_socket.accept()

# 创建目标套接字
destination_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
destination_socket.connect(('localhost', 9000))

# 调用数据传输函数
transfer_data(source_conn, destination_socket)

# 关闭套接字连接
source_conn.close()
destination_socket.close()