masscollaborationlabs/p2pfs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use urlretrive style reporthook.

Browse source
This commit is contained in:
Yuxin Wang 2018-10-03 22:59:55 -04:00
parent e801517fda
commit 0c8db5cf68
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
p2pfs/core/peer.py
View file

@ -93,7 +93,7 @@ class Peer(MessageServer):
self._file_list = self._file_list_result.get()
return self._file_list
def download(self, file, destination, progress=None):
def download(self, file, destination, reporthook=None):
with self._file_list_lock:
if self._file_list is None or file not in self._file_list.keys():
return False, 'Requested file {} does not exist, try list_file?'.format(file)
@ -146,8 +146,8 @@ class Peer(MessageServer):
'filename': file,
'chunknum': number
})
if progress is not None:
progress(i + 1, totalchunknum)
if reporthook:
reporthook(i + 1, Peer._CHUNK_SIZE, fileinfo['size'])
logger.debug('Got {}\'s chunk # {}'.format(file, number))
# change the temp file into the actual file