socket之文件下载器

文件下载器

先写客户端再写服务端

1.tcp下载器客户端

import socket


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

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

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

    # 获取下载的文件名称
    download_file_name = input("请输入要下载的文件名称：")

    # 将文件名称发送到服务器
    tcp_socket.send(download_file_name.encode("gbk"))

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

    if recv_data:
        # 保存接收到的数据到新文件
        with open("[新]"+download_file_name.wb) as f:
            f.write(recv_data)

    # 关闭套接字
    tcp_socket.close()


if __name__ == '__main__':
    main()

2.tcp下载器服务端

import socket


def send_file_2_client(new_client_socket,client_addr):
    # 1.接收客户端需要下载的文件名
    file_name = new_client_socket.recv(1024).decode("gbk")
    print("客户端(%s)需要下载的文件是：%s" %(str(client_addr),file_name))

    file_content = None
    # 2.打开这个文件，读取数据
    try:
        f = open(file_name,"rb")
        file_content = f.read()
        f.close()
    except Exception as ret:
        print("没有要下载的文件（%s）" % file_name)

    # 3.发送文件数据给客户端
    if file_content:
        new_client_socket.send(file_content)


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

    # 绑定本地信息
    tcp_server_socket.bind("",7890)

    # 让默认套接字由主动变为被动listen
    tcp_server_socket.listen(128)

    while True:
        # 等待客户端的连接 accept
        new_client_socket ,client_addr = tcp_server_socket.accept()

        # 调用发送文件函数，完成客户端的服务
        send_file_2_client(new_client_socket,client_addr)

        # 关闭套接字
        new_client_socket.close()

    tcp_server_socket.close()


if __name__ == "__main__":
    main()