手机版
你好,游客 登录 注册
背景:
阅读新闻

Python3使用TCP编写一个简易的文件下载器

[日期:2019-05-07] 来源:Linux社区  作者:fighting25 [字体: ]

利用Python3来实现TCP协议,和UDP类似。UDP应用于及时通信,而TCP协议用来传送文件、命令等操作,因为这些数据不允许丢失,否则会造成文件错误或命令混乱。下面代码就是模拟客户端通过命令行操作服务器。客户端输入命令,服务器执行并且返回结果。

TCP(Transmission Control Protocol 传输控制协议):是一种面向连接的、可靠的、基于字节流的传输层通信协议,由IETF的RFC 793定义。

使用TCP编写一个简易的文件下载器要求:需编写文件下载器服务端和文件下载器客户端

服务器端代码:

import socket
import os
import threading


# 处理客户端请求下载文件的操作(从主线程提出来的代码)
def deal_client_request(ip_port, service_client_socket):
    # 连接成功后,输出“客户端连接成功”和客户端的ip和端口
    print("客户端连接成功", ip_port)
    # 接收客户端的请求信息
    file_name = service_client_socket.recv(1024)
    # 解码
    file_name_data = file_name.decode("utf-8")
    # 判断文件是否存在
    if os.path.exists(file_name_data):
        #输出文件字节数
        fsize = os.path.getsize(file_name_data)
        #转化为兆单位
        fmb = fsize/float(1024*1024)
        #要传输的文件信息
        senddata = "文件名:%s  文件大小:%.2fMB"%(file_name_data,fmb)
        #发送和打印文件信息
        service_client_socket.send(senddata.encode("utf-8"))
        print("请求文件名:%s  文件大小:%.2f MB"%(file_name_data,fmb))
        #接受客户是否需要下载
        options = service_client_socket.recv(1024)
        if options.decode("utf-8") == "y":
            # 打开文件
            with open(file_name_data, "rb") as f:
                # 计算总数据包数目
                nums = fsize/1024 
                # 当前传输的数据包数目
                cnum = 0

                while True:
                    file_data = f.read(1024)
                    cnum = cnum + 1
                    jindu = cnum/nums*100

                    print("当前已下载:%.2f%%"%jindu,end = "\r")
                    if file_data:
                        # 只要读取到数据,就向客户端进行发送
                        service_client_socket.send(file_data)
                    # 数据读完,退出循环
                    else:
                        print("请求的文件数据发送完成")
                        break
        else:
            print("下载取消!")
    else:
        print("下载的文件不存在!")
    # 关闭服务当前客户端的套接字
    service_client_socket.close()


if __name__ == '__main__':
    # 把工作目录切换到data目录下
    os.chdir("./data")
    # 创建套接字
    tcp_server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    # 绑定端口号
    tcp_server_socket.bind(("", 3356))
    # 设置监听,将主动套接字变为被动套接字
    tcp_server_socket.listen(128)

    # 循环调用accept,可以支持多个客户端同时连接,和多个客户端同时下载文件
    while True:
        service_client_socket, ip_port = tcp_server_socket.accept()
        # 连接成功后打印套接字号
        #print(id(service_client_socket))

        # 创建子线程
        sub_thread = threading.Thread(target=deal_client_request, args=(ip_port, service_client_socket))
        # 启动子线程
        sub_thread.start()

客户端代码:

# 多任务文件下载器客户端
import socket


if __name__ == '__main__':
    # 创建套接字
    tcp_client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    # 和服务端连接
    server_ip = input("输入服务器IP:")
    tcp_client_socket.connect((server_ip, 3356))
    # 发送下载文件的请求
    file_name = input("请输入要下载的文件名:")
    # 编码
    file_name_data = file_name.encode("utf-8")
    # 发送文件下载请求数据
    tcp_client_socket.send(file_name_data)
    # 接收要下载的文件信息
    file_info = tcp_client_socket.recv(1024)
    # 文件信息解码
    info_decode = file_info.decode("utf-8")
    print(info_decode)
    #获取文件大小
    fileszie = float(info_decode.split(':')[2].split('MB')[0])
    fileszie2 = fileszie*1024
    # 是否下载?输入y 确认 输入q 取消
    opts = input("是否下载?(y 确认 q 取消)")
    if opts == 'q':
        print("下载取消!程序退出")
    else:
        print("正在下载 》》》")
        #向服务器确认正在下载
        tcp_client_socket.send(b'y')

        # 把数据写入到文件里
        with open("./" + file_name, "wb") as file:
            #目前接收到的数据包数目
            cnum = 0

            while True:
                # 循环接收文件数据
                file_data = tcp_client_socket.recv(1024)
                # 接收到数据
                if file_data:
                    # 写入数据
                    file.write(file_data)
                    cnum = cnum+1
                    jindu =cnum/fileszie2*100
                    print("当前已下载:%.2f%%"%jindu,end = "\r")
                # 接收完成
                else:
                    print("下载结束!")
                    break
    # 关闭套接字
    tcp_client_socket.close()

运行窗口如下:

1)服务器端

Python3使用TCP编写一个简易的文件下载器

2)客户端

Python3使用TCP编写一个简易的文件下载器

注意:客户端和服务器端不要运行在IDLE中,直接使用终端运行。

更多Ubuntu相关信息见Ubuntu 专题页面 https://www.linuxidc.com/topicnews.aspx?tid=2

Linux公社的RSS地址https://www.linuxidc.com/rssFeed.aspx

本文永久更新链接地址https://www.linuxidc.com/Linux/2019-05/158530.htm

linux
相关资讯       Python3文件下载器 
本文评论   查看全部评论 (0)
表情: 表情 姓名: 字数

       

评论声明
  • 尊重网上道德,遵守中华人民共和国的各项有关法律法规
  • 承担一切因您的行为而直接或间接导致的民事或刑事法律责任
  • 本站管理人员有权保留或删除其管辖留言中的任意内容
  • 本站有权在网站内转载或引用您的评论
  • 参与本评论即表明您已经阅读并接受上述条款