Move request chunk register message to download method.
This commit is contained in:
parent
99bd980f10
commit
f0740851d5
1 changed files with 6 additions and 5 deletions
|
@ -121,6 +121,12 @@ class Peer(MessageServer):
|
|||
dest_file.seek(number * (512 * 1024), 0)
|
||||
dest_file.write(pybase64.b64decode(raw_data.encode('utf-8'), validate=True))
|
||||
dest_file.flush()
|
||||
# send request chunk register to server
|
||||
self._write_message(self._server_sock, {
|
||||
'type': MessageType.REQUEST_CHUNK_REGISTER,
|
||||
'filename': file,
|
||||
'chunknum': number
|
||||
})
|
||||
progress(i + 1, totalchunknum)
|
||||
logger.debug('Got {}\'s chunk # {}'.format(file, number))
|
||||
|
||||
|
@ -178,11 +184,6 @@ class Peer(MessageServer):
|
|||
'chunknum': message['chunknum'],
|
||||
'data': pybase64.b64encode(raw_data).decode('utf-8')
|
||||
})
|
||||
self._write_message(self._server_sock, {
|
||||
'type': MessageType.REQUEST_CHUNK_REGISTER,
|
||||
'filename': message['filename'],
|
||||
'chunknum': message['chunknum']
|
||||
})
|
||||
elif message['type'] == MessageType.PEER_REPLY_CHUNK:
|
||||
self._download_results[message['filename']].put((message['chunknum'], message['data']))
|
||||
else:
|
||||
|
|
Loading…
Reference in a new issue