Python实现文件下载

在Python中要实现文件下载, 首先需要导入socket

客户端

• 客户端需要做的事情有:
  1 创建套接字
  2 连接到服务器
  3 告诉服务器想要下载的文件名
  4 接收从服务器返回的文件数据
  5 将文件写入到本地
  6 关闭套接字

import socket

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

    # 链接服务器
    tcp_client.connect(('10.100.58.8', 7890))

    # 发送给服务器需要下载的文件名
    download_name = input('请输入您要下载的文件名:')
    tcp_client.send(download_name.encode('utf-8'))

    # 接收服务器返回的数据， 并写入文件
    res = tcp_client.recv(1024)
    if res:
        with open('[new]'+download_name, 'wb') as f:
            f.write(res)

    tcp_client.close()


if __name__ == '__main__':
    main()

服务器端

• 服务器端需要做的事情有:
  1 创建套接字
  2 绑定服务器ip和端口
  3 监听客户端
  4 等待客户端的请求
  5 接收客户端请求下载的文件名并读取相应的文件
  6 将获取到的文件内容发送给客户端
  7 关闭套接字

import socket

def main():
    # 创建套接字
    tcp_server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    # 绑定服务器地址
    tcp_server.bind(('10.100.58.8', 7890))
    # 监听客户端
    tcp_server.listen(120)
    print('waiting......')
    # 等待客户端的链接
    client_sock, client_addr = tcp_server.accept()
    print(str(client_addr))
    # 接收来自客户端的数据请求
    res = client_sock.recv(1024).decode('utf-8')
    print('您要下载的文件是: %s' % res)
    try:
        # 打开并读取文件数据
        f = open(res, 'rb')
        content = f.read()
        f.close()
    except Exception as ret:
        print('no such file')
    # 将文件内容发送给客户端
    client_sock.send(content)
    client_sock.close()
    tcp_server.close()


if __name__ == '__main__':
    main()