python编写网速测试脚本

要编写网速测试脚本，可以使用 Python 自带的 `socket` 模块进行网络通信，再结合 `time` 模块计算传输时间和速率。下面是一个简单的示例代码，用于测试网络下载速率：

import socket
import time

# 测试下载速率
url = "http://example.com/test.zip" # 下载文件的 URL
buffer_size = 1024 * 1024 # 缓冲区大小，这里设置为 1MB
start_time = time.time()
with socket.create_connection(("example.com", 80)) as sock:
    # 发送 HTTP 请求
    sock.sendall(f"GET {url} HTTP/1.1\r\nHost: example.com\r\n\r\n".encode())
    # 读取 HTTP 响应头
    header = b""
    while b"\r\n\r\n" not in header:
        header += sock.recv(buffer_size)
    # 读取 HTTP 响应体
    data = b""
    while True:
        chunk = sock.recv(buffer_size)
        if not chunk:
            break
        data += chunk
end_time = time.time()
download_speed = len(data) / (end_time - start_time) / 1024 / 1024 # 计算下载速率
print("下载速率：%.2f MB/s" % download_speed)

以上代码中，使用 `socket.create_connection()` 创建 TCP 连接，发送 HTTP 请求，读取响应头和响应体。需要注意的是，下载速率的计算需要除以传输时间，并将字节数转换为 MB，以得到 MB/s 的速率单位。