tcp案例之文件下载器

文件下载器客户端

import socket

def main():

    # 1.创建一个tcp socket

    tcp_client_socket=socket.socket(socket.AF_INET,socket.SOCK_STREAM)

    #2. 连接服务端

    tcp_client_socket.connect(('127.0.0.1',9999))

    #3. 获取用户输入的文件名

    file_name=input("请输入要下载的文件名:")

    # 4. 把要下载的文件名发给服务端

    tcp_client_socket.send(file_name.encode())

    #5. 等待接收服务端回传的文件内容

    recv_data=tcp_client_socket.recv(1024*1024) # 一次最多接收1M字节

    print(recv_data)

    #6. 假如内容不为空，则写入到文件中

    if recv_data:

        with open("/home/python/Desktop/"+file_name,"wb") as f:

            f.write(recv_data)

    #7. 关闭tcp socket

    tcp_client_socket.close()

if __name__ == '__main__':

    main()

文件下载器服务端

import socket

def get_file_content(file_name):

    """通过文件名取得文件内容"""

    try:

        # 打开文件，读取数据   www.png

        file=open(file_name,"rb")

        content=file.read()

        file.close()

        return content

    except:

        print("%s下载的文件找不到"%file_name)

def main():

    # 1. 创建服务端的socket，用来做监听的

    server_socket=socket.socket(socket.AF_INET,socket.SOCK_STREAM)

    # 复用端口

    server_socket.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR,True)

    # 2. 绑定端口

    server_socket.bind(('',9999))

    # 3. 开启监听 listen

    server_socket.listen(5)

    # 4. 循环接收客户端的下载请求

    while True:

        new_sock,client_addr=server_socket.accept()

        print("有新的下载请求,来自-->",client_addr)

        #5. 获取客户端发送过来的要下载的文件名

        recv_data=new_sock.recv(1024) # 取得要下载的文件名

        # 把字节串转为字符串

        file_name=recv_data.decode()

        # 6.通过函数取得要下载的文件的内容

        file_content=get_file_content(file_name)

        # 7. 把内容回传给客户端

        if file_content:

            new_sock.send(file_content)

        # 8. 关闭与客户端对接的socket

        new_sock.close()

    #9. 关闭服务端server_socket,不再接收新的下载文件的请求

    server_socket.close()

if __name__ == '__main__':

    main()