TCP文件下载案例

服务器

import socket

def send_file_2_client(new_client_socket, client_addr):
    # 1. 接收客户端需要下载的文件名
    file_name = new_client_socket.recv(1024).decode("utf-8")
    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_name:
        new_client_socket.send(file_content)


def main():
    tcp_server_socket = socket.socket(socket.AF_INET, socket.AF_STREAM)
    tcp_server_socket.binid(("", 7890))
    tcp_server_socket.listen(128)
    while True:
        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()

客户端

import socket

def main():
    # 1. 创建套接字
    tcp_socket = socket.socket(socket.AF_INET, socket.AF_STREAM)
    # 2. 获取服务器ip和port
    dest_ip = input("请输入下载服务器ip: ")
    dest_port = int(input(“请输入下载服务器port: ”))
    # 3. 连接服务器
    tcp_socket.connect((dest_ip, dest_port))
    # 4. 获取下载的文件名
    download_file_name = input(“请输入要下载的文件名: ”)
    # 5. 将文件名发送到服务器
    tcp_socket.send(download_file_name,encode("utf-8"))
    # 6. 接收文件内容
    recv_data = tcp_socket.recv(1024)
    # 7. 保存数据
    with open("new_"+download_file_name, "wb") as f:
        f.write(recv_data)
    # 8. 关闭套接字
    tcp_socket.close()

if __name__ == “__main__”：
    main()