async def _store(self,
magnet: str,
content: StreamReader,
chunk_size=1024):
"""Check and store content file.
:param magnet:
:param content:
:param chunk_size:
:return:
"""
to_path = self.get_absolute_path(magnet)
tmp_content_path = ''.join(
[str(to_path),
'tmp.%s' % random.randint(10000, 99999)])
check = keccak.new(digest_bytes=32)
try:
with open(tmp_content_path, 'wb') as fd:
async for chunk, _ in content.iter_chunks():
fd.write(chunk)
check.update(data=chunk)
checksum = check.hexdigest()
if checksum != magnet:
log.error(
"Downloaded content file %s checksum %s didn't match",
magnet, checksum)
raise InvalidChecksum(magnet, checksum)
shutil.move(tmp_content_path, to_path)
return to_path
finally:
os.unlink(tmp_content_path)
async def store(self, magnet: str, content: StreamReader, chunk_size=1024):
"""Check and store content file.
:param magnet:
:param content:
:param chunk_size:
:return:
"""
to_path = self.get_absolute_path(magnet)
# content_path = Path(to_path) / magnet_path(magnet)
tmp_content_path = ''.join([str(to_path), 'tmp.%s' % random.randint(10000, 99999)])
check = keccak.new(digest_bytes=32)
try:
Path(tmp_content_path).parent.mkdir(parents=True, exist_ok=True)
with open(tmp_content_path, 'wb') as fd:
async for chunk, _ in content.iter_chunks():
fd.write(chunk)
check.update(data=chunk)
checksum = check.hexdigest()
if checksum != magnet:
self.log.error("Downloaded content file %s checksum %s didn't match", magnet, checksum)
raise InvalidChecksum(magnet, checksum)
shutil.move(tmp_content_path, to_path)
finally:
try:
os.unlink(tmp_content_path)
except FileNotFoundError:
# it is ok, file was moved
pass
