背景和目的:
利用python request 编写脚本测试公司系统的文件上传接口。前端读取文件的大小然后文件分片传给后端,后端将每一片数据重新组合成文件。大概的过程是:前端将整个文件的md5、size(大小)、name(文件名)、ext(文件后缀)、totalchunk(分片总数)与分片文件的md5、chunk(分片数据),chunkindex(当前分片文件的下标)等传给后台,后台取得这些数据后,通过chunkindex将每一片数据重组,重组完后,进行md5校验,判断文件上传是否成功。我只需要去调用后台的接口,然后判断文件是否上传成功,并且上传没有错误,其他的文件校验就不用去深究。
开发前端使用的是vue,后台使用的是php,要利用python实现对这一接口的调用,那么就先要将前端数据给模拟出来,然后循环去调用接口,将文件分片上传,我的思路大概如下:
1.获取整个文件的大小、名字、后缀、分片总数,定义每片文件的大小:
def __init__(self,data): dat = json.loads(data) self.path = dat['path'] # 获取文件路径 self.CHUNK_SIZE = 1024*1024*2 # 定义每片文件的大小 self.size = os.path.getsize(dat['path']) # 获取文件的大小 self.totalchunk = math.ceil(self.size / self.CHUNK_SIZE) # 获取文件的分片总数 self.ext = os.path.basename(dat['path']).split('.').pop() # 获取文件的后缀 self.name = os.path.basename(dat['path']) # 获取文件的名字
2.获取文件的md5,查看了开发那边的md5算法,利用python实现过程如下:
# 使用hashlib库的md5方法获取指定文件的md5 def getmd5(self,path): m = hashlib.md5() with open(path, 'rb') as f: for line in f: m.update(line) md5code = m.hexdigest() return md5code # 开发那边对md5的算法进行了优化,当文件的大小小于1M时,直接通过getmd5方法去获取文件的md5值; # 当文件大于1M时,通过截取整个文件中的某几个片段,然后拼接成一个文件,再去获取其md5值,最后删除这个文件 def md5(self,path): if self.size < 1024 * 1024: return self.getmd5(path) f = open(path, 'rb') f.seek(0, 0) data = f.read(2012) f.seek(int(self.size / 2) - 1999, 0) data += f.read(1999) f.seek(-2010, 2) data += f.read(2010) f.close() path = 'D:/copy_' + str(os.path.basename(path)) f = open(path, 'wb') f.write(data) f.close() val = self.getmd5(path) os.remove(path) return val
3.调用文件上传的接口
def uploading(self, chunkIndex): MD5 = self.md5(self.path) # 整个文件的md5 start = (chunkIndex - 1) * self.CHUNK_SIZE # 截取文件的起始位置 end = min(self.size, start + self.CHUNK_SIZE)# 截取文件的结束位置 f = open(self.path, 'rb') f.seek(start) data = f.read(end) # 待分片上传的数据 f.close() path1 = 'D:/copy_' + str(os.path.basename(self.path)) # 将该数据保存在本地 f = open(path1, 'wb') f.write(data) f.close() chunk_md5 = self.md5(path1) # 读取分片上传数据的md5 # 将所有的数据储存在files字典当中,利用requests的files传输数据 # 使用requests files类型时,要像下面一样构建参数,不然会有错误 files={ 'chunk':('blob',data,'application/octet-stream'), 'name':(None,self.name), 'ext':(None,self.ext), 'index':(None,chunkIndex), 'total': (None,self.totalchunk), 'size': (None,self.size), 'chunk_md5': (None,chunk_md5), 'md5': (None,MD5), } # 使用requests发送接口请求 res = self.request.send('post', 'https://10.104.17.222/bank/elements/uploading', verify=False, files=files) os.remove(path1) # 删除存在本地的分片文件 return res.json()
4.循环调用文件上传的接口
chunkIndex = 1 while chunkIndex <= totalchunk: res2 = upload.uploading(chunkIndex) chunkIndex += 1
5.从服务器上去读取通过接口上传的文件的md5值,判断是否正确
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。
《魔兽世界》大逃杀!60人新游玩模式《强袭风暴》3月21日上线
暴雪近日发布了《魔兽世界》10.2.6 更新内容,新游玩模式《强袭风暴》即将于3月21 日在亚服上线,届时玩家将前往阿拉希高地展开一场 60 人大逃杀对战。
艾泽拉斯的冒险者已经征服了艾泽拉斯的大地及遥远的彼岸。他们在对抗世界上最致命的敌人时展现出过人的手腕,并且成功阻止终结宇宙等级的威胁。当他们在为即将于《魔兽世界》资料片《地心之战》中来袭的萨拉塔斯势力做战斗准备时,他们还需要在熟悉的阿拉希高地面对一个全新的敌人──那就是彼此。在《巨龙崛起》10.2.6 更新的《强袭风暴》中,玩家将会进入一个全新的海盗主题大逃杀式限时活动,其中包含极高的风险和史诗级的奖励。
《强袭风暴》不是普通的战场,作为一个独立于主游戏之外的活动,玩家可以用大逃杀的风格来体验《魔兽世界》,不分职业、不分装备(除了你在赛局中捡到的),光是技巧和战略的强弱之分就能决定出谁才是能坚持到最后的赢家。本次活动将会开放单人和双人模式,玩家在加入海盗主题的预赛大厅区域前,可以从强袭风暴角色画面新增好友。游玩游戏将可以累计名望轨迹,《巨龙崛起》和《魔兽世界:巫妖王之怒 经典版》的玩家都可以获得奖励。