import os
import math
import hashlib
def get_file_md5(self, file_path):
 """獲取文件的md5值"""
 with open(file_path, 'rb') as f:
data = f.read()
return hashlib.md5(data).hexdigest()
def get_filename(self, filepath):
 """獲取文件原始名稱"""
 # 文件名帶后綴
 filename_with_suffix = os.path.basename(filepath)
 # 文件名
 filename = filename_with_suffix.split('.')[0]
 # 后綴名
 suffix = filename_with_suffix.split('.')[-1]
 return filename_with_suffix, filename, suffix
def get_chunk_info(self, file_path):
 """獲取分段信息"""
 # 獲取文件總大小(字節(jié))
 file_total_size = os.path.getsize(file_path)
 print(file_total_size)
 # 分段總數(shù)
 total_chunks_num = math.ceil(file_total_size / self.chunk_size)
 # 文件名（帶后綴）
 filename = self.get_filename(file_path)[0]
 # 文件的md5值
 file_md5 = self.get_file_md5(file_path)
 return file_total_size, total_chunks_num, filename, file_md5

import requests
def do_chunk_and_upload(self, file_path):
​ """將文件分段處理，并上傳"""
 file_total_size, total_chunks_num, filename, file_md5 = self.get_chunk_info(file_path)
 # 遍歷
 for index in range(total_chunks_num):
  print('第{}次文件上傳'.format(index + 1))
  if index + 1 == total_chunks_num:
partSize = file_total_size % chunk_size
  else:
partSize = chunk_size
  # 文件偏移量
  offset = index * chunk_size
  # 生成分片id,從1開(kāi)始
  chunk_id = index + 1
  print('開(kāi)始準(zhǔn)備上傳文件')
  print("分片id:", chunk_id, "文件偏移量：", offset, ",當(dāng)前分片大小:", partSize, )
  # 分段上傳文件
  self.__upload(offset, chunk_id, file_path, file_md5, filename, partSize, total_chunks_num)
def __upload(self, offset, chunk_id, file_path, file_md5, filename, partSize, total):
 """分次上傳文件"""
 url = 'http://**/file/brust/upload'
 params = {'chunk': chunk_id,
 'fileMD5': file_md5,
 'fileName': filename,
 'partSize': partSize,
 'total': total
 }
 # 根據(jù)文件路徑及偏移量，讀取文件二進(jìn)制數(shù)據(jù)
 current_file = open(file_path, 'rb')
 current_file.seek(offset)
 files = {'file': current_file.read(partSize)}
 resp = requests.post(url, params=params, files=files).text
 print(resp)

def merge_file(self, filepath):
  """合并"""
  url = 'http://**/file/brust/merge'
  file_total_size, total_chunks_num, filename, file_md5 = self.get_chunk_info(filepath)
​  payload = json.dumps(
{
 "fileMD5": file_md5,
 "chunkTotal": total_chunks_num,
 "fileName": filename
}
  )
  print(payload)
  headers = {
"Content-Type": "application/json"
  }
  resp = requests.post(url, headers=headers, data=payload).text
  print(resp)

# fileupload.py
...
if __name__ == '__main__':
 filepath = sys.argv[1]
 # 每一段切片的大小(MB)
 chunk_size = 2 * 1024 * 1024
 fileApi = FileApi(chunk_size)
 # 分段上傳
 fileApi.do_chunk_and_upload(filepath)
 # 合并
 fileApi.merge_file(filepath)

# cmd.bat
@echo off
set filepath=%1
python  C:\Users\xingag\Desktop\rpc_demo\fileupload.py %*