python--简单的文件断点续传实例

一、程序说明

1、文件上传目标路径：home/file

2、目标文件：putfile.png

3、服务端代码：put_server.py

4、客户端代码：put_client.py

二、各部分代码

1、服务端代码：put_server.py

 #!/usr/bin/env python
 # -*- coding:utf-8 -*-
 """
 实现文件断点续传的服务器端
 """

 import socket
 import os

 BASE_DIR = os.path.dirname(os.path.dirname(__file__))

 home = os.path.join(BASE_DIR, "home/file")
 sk = socket.socket()
 sk.bind(('127.0.0.1', 8001))
 sk.listen(5)

 while True:
     print("Waiting....")
     conn, addr = sk.accept()
     conn.sendall(bytes('欢迎登录', encoding='utf-8'))
     flag = True
     while flag:
         client_bytes = conn.recv(1024)   #接收客户端发送过来的内容
         client_str = str(client_bytes, encoding='utf-8')  #将内容转换成字符串

         # 将客户端发送过来的内容以"|"拆分为:命名方法，文件名，文件大小，目标路径
         func, file_name, file_byte_size, targe＿path = client_str.split('|', 3)
         file_byte_size = int(file_byte_size)
         path = os.path.join(home, file_name)
         has_received = 0

         #首先判断该路径下是否已存在文件
         if os.path.exists(path):
             conn.sendall(bytes("", encoding='utf-8'))  #发送通知客户端，该文件已存在
             is_continue = str(conn.recv(1024), encoding='utf-8')  #等待客户端选择回复
             if is_continue == "":
                 has_file_size = os.stat(path).st_size
                 conn.sendall(bytes(str(has_file_size), encoding='utf-8'))  #将已接收的文件大小给客户端
                 has_received += has_file_size
                 f = open(path, 'ab')
             else:
                 f = open(path, 'wb')
         else:
             conn.sendall(bytes("", encoding='utf-8'))
             f = open(path, 'wb')

         while has_received < file_byte_size:
             try:
                 data = conn.recv(1024)
                 if not data:
                     raise Exception
             except Exception:
                 flag = False
                 break
             f.write(data)
             has_received += len(data)
         print("文件已接收完成！")
         f.close()

2、客户端代码：put_client.py

 #!/usr/bin/env python
 # -*- coding:utf-8 -*-
 """
 实现文件断点续传的客户端
 """

 import socket
 import sys
 import re
 import os
 FILE_DIR = os.path.dirname(__file__)

 ck = socket.socket()
 ck.connect(('127.0.0.1', 8001))
 print(str(ck.recv(1024), encoding='utf-8'))

 #定义一个函数，计算进度条
 def bar(num = 1, sum = 100):
     rate = float(num) / float(sum)
     rate_num = int(rate * 100)
     temp = '\r%d %%' % (rate_num)
     sys.stdout.write(temp)

 while True:
     inp = input('请输入（内容格式：post|文件路径 目标路径）: \n >>> ').strip()  #输入内容格式：命令|文件路径 目标路径
     func, file_path =inp.split("|", 1)  #将输入的内容拆分为两部分，方法名和路径
     local_path, target_path = re.split("\s*", file_path, 1) #再将路径部分，通过正则表达式。以空格拆分为：文件路径和上传的目标路径
     file_byte_size = os.stat(local_path).st_size  #获取文件的大小
     file_name = os.path.basename(local_path)   #设置文件名

     post_info = "post|%s|%s|%s" % (file_name, file_byte_size, target_path)  #将发送的内容进行拼接
     ck.sendall(bytes(post_info, encoding='utf-8'))  #向服务器端发送内容

     result_exist = str(ck.recv(1024), encoding='utf-8')
     has_sent = 0
     if result_exist == "":
         inp = input("文件已存在，是否续传？Y/N:").strip()
         if inp.upper() == 'Y':
             ck.sendall(bytes("", encoding='utf-8'))
             result_continue_pos = str(ck.recv(1024), encoding='utf-8')  #已经传输了多少的文件内容
             print(result_continue_pos)
             has_sent = int(result_continue_pos)

         else:
             ck.sendall(bytes("", encoding='utf-8'))  #发送2005代表不续传

     file_obj = open(local_path, 'rb')  #对文件进行读操作
     file_obj.seek(has_sent)  #调整指针

     while has_sent < file_byte_size:
         data = file_obj.read(1024)
         ck.sendall(data)
         has_sent += len(data)
         bar(has_sent, file_byte_size)  #进度条

     file_obj.close()
     print("文件上传成功！")