tcp案例：文件下载

tcp案例

客户端

import socket


def main():

    # 1.创建套接字
    tcp_client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

    # 2.获取服务器的ip,port
    client_port = int(input('请输入您要链接的服务器的端口号：'))

    # 3.链接服务器
    tcp_client_socket.connect(('192.168.10.4', client_port))

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

    # 5.将文件名字发送到服务器
    tcp_client_socket.send(download_file_name.encode('gbk'))

    # 6.接收文件中的数据
    recv_data = tcp_client_socket.recv(1024)  # 1024---->1k   1024*1024---->1M  1024*1024*1024---->1G

    # 如果接收到数据就创建文件，否则不创建
    if recv_data:
        # 7.保存接收到的数据到一个文件中
        with open('[新]'+download_file_name, 'wb') as f:
            f.write(recv_data)

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


if __name__ == '__main__':
    main()
    

服务端

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():
    # 1.创建套接字
    tcp_sever_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

    # 2.绑定本地信息
    tcp_sever_socket.bind(('', 7788))

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

    # 4.等待客户端的链接
    new_client_socket, client_addr = tcp_sever_socket.accept()

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

    # 6.关闭套接字
    new_client_socket.close()
    tcp_sever_socket.close()


if __name__ == '__main__':
    main()