Python网络编程——使用TCP方式传输文件

TCP文件下载器

客户端
需求：输入要下载的文件名，从服务器端将文件拷贝到本地

步骤：
1.创建TCP套接字，绑定端口
2.连接服务端
3.输入要下载的文件名
4.将文件名编码，并发送到服务端
5.接收服务端返回的数据
6.关闭套接字

代码

# 导入模块
import socket

# 创建套接字
tcp_client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 绑定端口
tcp_client_socket.bind(("", 8080))
# 连接IP地址和端口
tcp_client_socket.connect(("192.168.43.216", 8080))
file_name = input("请输入要下载的文件：\n")
# 文件名编码
tcp_client_socket.send(file_name.encode())
try:
    # 文件传输
    with open("C:/Users/Administrator/Desktop/" + file_name, "wb") as file:
        while True:
            # 接收数据
            file_data = tcp_client_socket.recv(1024)
            # 数据长度不为0写入文件
            if file_data:
                file.write(file_data)
            # 数据长度为0表示下载完成
            else:
                break
# 下载出现异常时捕获异常
except Exception as e:
    print("下载异常", e)
# 无异常则下载成功
else:
    print(file_name, "下载成功")
# 关闭客户端
tcp_client_socket.close()

服务端
需求：接收客户端发送的文件名，将文件传输给客户端

步骤：
1.创建TCP套接字，绑定端口
2.设置监听
3.接收客户端传输的文件名信息并解码
4.将文件发送到客户端
5.关闭客户端套接字（断开连接）

代码

# 导入模块
import socket
# 创建tcp服务端socket
tcp_server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 绑定端口
tcp_server_socket.bind(("", 8080))
# 设置监听，把服务端socket由主动套接字改成被动套接字，只能接收客户端的连接请求
tcp_server_socket.listen(128)
while True:
    # 接收客户端信息
    client_socket, client_ip = tcp_server_socket.accept()
    print("客户端：", client_ip, "连接")
    # 接收下载信息
    file_name_data = client_socket.recv(1024)
    # 解码下载信息
    file_name = file_name_data.decode()
    try:
        # 数据传输
        with open("/home/python/Desktop/" + file_name, "rb") as file:
            while True:
                # 读取文件数据
                file_data = file.read(1024)
                # 数据长度不为0表示还有数据没有写入
                if file_data:
                    client_socket.send(file_data)
                # 数据为0表示传输完成
                else:
                    print(file_name, "传输成功")
                    break
    except Exception as e:
        print("传输异常：", e)
    # 关闭客户端连接
    client_socket.close()

测试
客户端请求

服务器响应

客户端接收