Python学习---基于TCP协议的文件下载器案例

客户端：

# 1·导入模块
import socket

# 2·创建套接字
tcp_client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 3·建立和服务器的连接
address = ("192.168.52.2", 10000)
tcp_client_socket.connect(address)
# 4﹑接收用户输入的文件名
file_name = input("请输入要下载的文件名：\n")
# 5·发送文件名到服务器端
tcp_client_socket.send(file_name.encode())
# 6·创建文件﹐准备接收服务端返回的文件数据
with open(r"E:\\python_learn\\python_studypro\\data\\" + file_name, "wb") as file:
    while True:
        file_data = tcp_client_socket.recv(1024)
        if file_data:
            file.write(file_data)
        else:
            break
# 7·保存文件数据
# 8、关闭套接字
tcp_client_socket.close()

 服务端：文件存在于当前目录下

# 1、导入模块
import socket

# 2、创建套接字
tcp_server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 3、绑定端口号
address = ("", 10000)
tcp_server_socket.bind(address)
# 4、设置监听，设置套接字由主动为被动
tcp_server_socket.listen(128)
# 5、接收客户端连接
while True:
    new_client_socket, ip_port = tcp_server_socket.accept()
    print(f"接收到来自{ip_port}的连接请求，已连接……………………")
    # 6、接收客户端发送的文件名
    recv_data = new_client_socket.recv(1024)
    file_name = recv_data.decode()
    print(f"客户端要求下载:{file_name}")

    # 7、根据文件名读取文件内容
    try:
        with open(file_name, "rb") as file:
            while True:
                file_data = file.read(1024)
                if file_data:
                    new_client_socket.send(file_data)
                else:
                    break
    except Exception as e:
        print(f"文件{file_name}不存在，下载失败！！！")
    else:
        print(f"文件下载成功！！！")
    # 8、把读取的内容发送给客户端
    # 9、关闭当前和客户端的连接
    new_client_socket.close()
# 10、关闭服务器,一般不关闭，下面这行代码不执行也可以
tcp_server_socket.close()