{每日一题}:tcp协议实现简单的文件下载器(单任务版)

文件下载器客户端

　　这个版本的只是为了方便回顾一下TCP客服端,服务端的创建流程,缺点就是  服务器一次只能让一个人访问下载,过两个写个使用面向对象写一个多线程版的强化一下.

from socket import *



def mian():
    # 1. 创建套接字
    tcp_client_scoket = socket(AF_INET,SOCK_STREAM)

    # 2.连接本地服务器
    tcp_client_scoket.connect(('192.168.0.117',8889))

    # 3.输入要下载的文件名
    file_name = input('请输入要下载的文件名：')
    tcp_client_scoket.send(file_name.encode())

    # 4.接收传过来的文件内容
    data = tcp_client_scoket.recv(1024*1024*1024)
    print(data)

    # 5.写入本地
    if data:
        with open('/home/python/Desktop/' + file_name,'wb') as f:
            f.write(data)

    tcp_client_scoket.close()

if __name__ == '__main__':
    mian()

文件下载器服务端

from socket import *


def get_file_name(file_name):
    """获取文件内容"""
    try:
        with open('/home/python/Desktop/test/'+file_name,'rb') as f:
            content = f.read()
            return content
    except:
        print('无此文件')  # 会返回一个空值

def main():
    # 1.创建服务器的tcp套接字
    tcp_serve_socket = socket(AF_INET,SOCK_STREAM)

    # 2.端口复用
    tcp_serve_socket.setsockopt(SOL_SOCKET,SO_REUSEADDR,True)

    # 3.绑定服务器端口
    tcp_serve_socket.bind(('',8889))

    # 4.开始监听
    tcp_serve_socket.listen(5)

    while True:
        new_socket,client_addr = tcp_serve_socket.accept()
        print("来自%s的下载请求" % str(client_addr))

        # 5.接收来自服务端的文件名
        file_data = new_socket.recv(1024)
        file_name = file_data.decode()

        # 6.查找是否有此文件并写出
        content = get_file_name(file_name)

        # 7. 发送文件内容
        if content:
            new_socket.send(content)
            print('发送成功')

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

    tcp_serve_socket.close()

if __name__ == '__main__':
    main()

如果你和我有共同爱好,我们可以加个好友一起交流!