Python-网络编程：TCP3 案例（文件下载器-简单版）

描述：

        

客户端：

        思路：

1. 向服务端请求某个文件（eg：1.txt）（将文件名发过去）
2. 接收数据
3. 如果有数据，就新建一个文件（eg：new-1.txt），将接收到的数据写进文件
4. 关闭套接字

       代码：

import socket

def main():
    # 1.创建套接字
    tcp_socket_client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

    # 2.获取服务器的ip port
    dest_ip = input("请输入下载服务器的ip：")
    dest_port = input("请输入下载服务器的port：")

    # 3.连接服务器
    tcp_socket_client.connect((dest_ip, dest_port))

    # 4.获取下载的文件名字
    filename = input("请输入要下载的文件名字：")

    # 5.将文件名发送到服务器
    tcp_socket_client.send(filename.encode('utf-8'))

    # 6.接收文件中的数据
    recv_data = tcp_socket_client.recv(1024*1024) # 1M

    # 7.保存接收到的数据到一个文件中
    if recv_data:
        with open("new-"+filename, "wb") as ref: # wb:2进制模式写
            ref.write(recv_data)

    # 8.关闭套接字
    tcp_socket_client.close()

服务端：

       思路：

1. 接收到一个客户端的连接请求，收到需要下载的文件名（eg：1.txt）
2. 找到这个文件（eg：1.txt）
3. 读取文件内容，并将文件内容发送回客户端
4. 关闭套接字

import socket

def send_file_to_client(new_socket):
    # 1.接收所需文件名
    filename = new_socket.recv(1024).decode('utf-8')
    print("需要下载的文件：", filename)

    # 2.模拟打开的文件内容 
    # # 此处模拟成功。
    # new_socket.send("This is content".encode('utf-8'))
    # # with的前提是这个文件能打开，所以此处用with不合适
    # with open(filename, 'rb') as ref:
    #     msg = ref.read
    #     print(msg)
    filecontent = None
    try:
        f = open(filename, 'rb')
        filecontent = f.read()
        f.close()
    except:
        print("没有要下载的文件", filename)

    # 3.发送数据到客户端
    if filecontent:
        new_socket.send(filecontent)


def main():
    # 1.创建监听套接字
    tcp_socket_server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

    # 2.绑定信息
    server_ip = "127.0.0.1"
    server_port = 7890
    tcp_socket_server.bind((server_ip, server_port))

    # 3.设为监听模式
    tcp_socket_server.listen(128)

    while True:
        # 4.accept
        print("等待连接：")
        new_socket, client_addr = tcp_socket_server.accept()
        print("连接客户端-", client_addr)

        # 5.调用发送文件函数，完成客户端服务
        send_file_to_client(new_socket)

        # 6.关闭套接字
        new_socket.close()
    tcp_socket_server.close()


if __name__ == "__main__":
    main()